Update app.py

app.py

@@ -67,12 +67,12 @@ def update_faiss_index():
     return db
 
 
-# Если индекс ещё не создан, обновляем его, иначе загружаем существующий
-# 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):
@@ -80,20 +80,20 @@ 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
-    return "### Hello World"
+    Поиск по индексу: возвращает список из документов
+    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-интерфейс