uhygfd commited on
Commit
1f4b9ea
·
verified ·
1 Parent(s): 67b409d

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -0
app.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from chatterbot import ChatBot
2
+ from chatterbot.trainers import ListTrainer
3
+ import gradio as gr
4
+
5
+ # Создание экземпляра бота
6
+ chatbot = ChatBot('WikiBot')
7
+
8
+ # Обучение бота
9
+ trainer = ListTrainer(chatbot)
10
+
11
+ # Загрузка данных из файла dataset.txt
12
+ with open('dataset.txt', 'r', encoding='utf-8') as file:
13
+ data = file.readlines()
14
+
15
+ # Обучение бота данным из файла
16
+ trainer.train(data)
17
+
18
+ # Функция для получения ответа от бота
19
+ def get_response(query):
20
+ response = chatbot.get_response(query)
21
+ return str(response)
22
+
23
+ # Функция для обработки ввода пользователя
24
+ def chat_with_bot(user_input):
25
+ return get_response(user_input)
26
+
27
+ # Создание Gradio интерфейса
28
+ interface = gr.Interface(
29
+ fn=chat_with_bot,
30
+ inputs="text",
31
+ outputs="text",
32
+ title="WikiBot",
33
+ description="Введите ваш вопрос и получите ответ от бота, обученного на данных из Википедии."
34
+ )
35
+
36
+ # Запуск интерфейса
37
+ interface.launch()