Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
+import re
 
-# Загрузка текстов из файлов лора
+# Загружаем текстовые данные из файлов лора
 def load_lore_files():
     lore_data = {}
     for filename in ["vampires.txt"]:
@@ -10,21 +11,29 @@ def load_lore_files():
 
 lore_data = load_lore_files()
 
-# Функция, которая ищет упоминания в лоре
+# Функция для очистки текста от нежелательных символов
+def clean_text(text):
+    # Убираем все символы, кроме букв и пробелов
+    return re.sub(r'[^a-zA-Zа-яА-ЯёЁ\s]', '', text.lower())
+
+# Функция поиска ответа
 def chatbot(message, history, system_message, max_tokens, temperature, top_p):
-    # Простой поиск по всем файлам лора
+    # Приводим запрос к стандартному виду
+    cleaned_message = clean_text(message)
+
     response = ""
     for filename, text in lore_data.items():
-        if message.lower() in text.lower():
+        cleaned_text = clean_text(text)
+        if cleaned_message in cleaned_text:
             response += f"В файле {filename} есть что-то похожее на ваш запрос!\n"
+
     if response == "":
         return "Извините, ничего не нашёл по вашему запросу."
-    
-    # Если есть ответ, добавляем его в историю
+
+    # Добавляем ответ в историю
     history.append((message, response))
     return response, history
 
-
 # Интерфейс чата с Gradio
 demo = gr.ChatInterface(
     chatbot,