Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import gradio as gr
+import spaces
 from datasets import load_dataset, concatenate_datasets
 from langchain.docstore.document import Document
 from langchain.vectorstores import FAISS
@@ -67,31 +68,32 @@ def update_faiss_index():
 
 
 # Если индекс ещё не создан, обновляем его, иначе загружаем существующий
-if not os.path.exists(INDEX_PATH):
-    db = update_faiss_index()
-else:
-    db = FAISS.load_local(INDEX_PATH, embeddings, allow_dangerous_deserialization=True)
-    print("Загружен существующий FAISS индекс из:", INDEX_PATH)
-
+# if not os.path.exists(INDEX_PATH):
+#     db = update_faiss_index()
+# else:
+#     db = FAISS.load_local(INDEX_PATH, embeddings, allow_dangerous_deserialization=True)
+#     print("Загружен существующий FAISS индекс из:", INDEX_PATH)
 
+@spaces.GPU
 def retrieve_articles(query):
     """
     Принимает запрос пользователя, ищет в FAISS-индексе топ-3 наиболее релевантных документа
     и возвращает отформатированный результат в Markdown.
     """
     # Поиск по индексу: возвращает список из документов
-    results = db.similarity_search(query, k=3)
-
-    # Форматируем результаты для вывода
-    result_text = ""
-    for doc in results:
-        result_text += (
-            f"### Статья {doc.metadata['section']}: {doc.metadata['section_name']}\n"
-        )
-        result_text += f"**Глава:** {doc.metadata['chapter_name']}\n\n"
-        result_text += f"**Текст статьи:**\n{doc.page_content}\n\n"
-        result_text += "---\n\n"
-    return result_text
+    # results = db.similarity_search(query, k=3)
+
+    # # Форматируем результаты для вывода
+    # result_text = ""
+    # for doc in results:
+    #     result_text += (
+    #         f"### Статья {doc.metadata['section']}: {doc.metadata['section_name']}\n"
+    #     )
+    #     result_text += f"**Глава:** {doc.metadata['chapter_name']}\n\n"
+    #     result_text += f"**Текст статьи:**\n{doc.page_content}\n\n"
+    #     result_text += "---\n\n"
+    # return result_text
+    return "### Hello World"
 
 
 # Создаём Gradio-интерфейс