import gradio as gr
from chatterbot import ChatBot
from chatterbot.trainers import ListTrainer
from chatterbot.trainers import ChatterBotCorpusTrainer
import wikipedia

chatbot = ChatBot("Chatpot",storage_adapter='chatterbot.storage.SQLStorageAdapter',database_uri = 'sqlite:///database.sqlite3',
                 logic_adapters=[  {
            "import_path": "chatterbot.logic.BestMatch"
        }
    ])




#file1 = open("dialogs.txt", "r")
#training_data = file1.read().splitlines()


#trainer = ListTrainer(chatbot)
#trainer.train(training_data)

trainer = ChatterBotCorpusTrainer(chatbot)
trainer.train("chatterbot.corpus.english","chatterbot.corpus.english.ai")

#print(training_data)



def greet(name):
    text = chatbot.get_response(name)
    
    if text == "wiki":
        result = wikipedia.summary(name, sentences =10)
        return result
    return str(text)

iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch()