Update app.py

app.py

@@ -1,37 +1,38 @@
-from chatterbot import ChatBot
-from chatterbot.trainers import ListTrainer
+import spacy
 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,
+# Загрузка модели spaCy
+nlp = spacy.load("ru_core_news_md")  # или "ru_core_news_md" для русского языка
+
+# Функция для загрузки статей из файла
+def load_articles(file_path):
+    with open(file_path, "r", encoding="utf-8") as f:
+        articles = f.readlines()
+    return [article.strip() for article in articles]
+
+# Функция для нахождения самой релевантной статьи
+def find_most_relevant_article(query, articles):
+    query_doc = nlp(query)
+    similarities = [(article, query_doc.similarity(nlp(article))) for article in articles]
+    most_relevant_article = max(similarities, key=lambda x: x[1])
+    return most_relevant_article[0]
+
+# Загрузка статей из файла
+file_path = "dataset.txt"
+articles = load_articles(file_path)
+
+# Определение функции для интерфейса Gradio
+def get_relevant_article(query):
+    return find_most_relevant_article(query, articles)
+
+# Создание интерфейса Gradio
+iface = gr.Interface(
+    fn=get_relevant_article,
     inputs="text",
     outputs="text",
-    title="WikiBot",
-    description="Введите ваш вопрос и получите ответ от бота, обученного на данных из Википедии."
+    title="Поиск релевантных статей из Википедии",
+    description="Введите запрос, чтобы найти самую релевантную статью из датасета."
 )
 
 # Запуск интерфейса
-interface.launch()
+iface.launch()