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

# Создание экземпляра бота
chatbot = ChatBot('WikiBot')

# Обучение бота
trainer = ListTrainer(chatbot)

# Загрузка данных из файла dataset.txt
with open('dataset.txt', 'r', encoding='utf-8') as file:
    data = file.readlines()

# Обучение бота данным из файла
trainer.train(data)

# Функция для получения ответа от бота
def get_response(query):
    response = chatbot.get_response(query)
    return str(response)

# Функция для обработки ввода пользователя
def chat_with_bot(user_input):
    return get_response(user_input)

# Создание Gradio интерфейса
interface = gr.Interface(
    fn=chat_with_bot,
    inputs="text",
    outputs="text",
    title="WikiBot",
    description="Введите ваш вопрос и получите ответ от бота, обученного на данных из Википедии."
)

# Запуск интерфейса
interface.launch()