new index

app.py

@@ -4,7 +4,7 @@ import sys
 import logging
 from config import *
 from documents_prep import DocumentsPreparation
-from index_retriever import initialize_models
+import index_retriever
 from chat_handler import ChatHandler
 
 REPO_ID = "MrSimple01/AIEXP_RAG_FILES"
@@ -14,7 +14,6 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 logger = logging.getLogger(__name__)
 
 doc_prep = None
-index_retriever = None
 chat_handler = None
 
 def log_message(message):
@@ -42,7 +41,7 @@ def initialize_system():
             log_message("Не удалось инициализировать модели")
             return False
         
-        chat_handler = ChatHandler(index_retriever)
+        chat_handler = ChatHandler(None)
         
         log_message("Система успешно инициализирована")
         return True
@@ -54,17 +53,57 @@ def initialize_system():
 def handle_question(question):
     if chat_handler is None:
         return "Система не инициализирована", ""
-    return chat_handler.answer_question(question)
+    
+    try:
+        answer = index_retriever.query(question)
+        sources = get_sources_for_question(question)
+        
+        chat_handler.add_to_history(question, answer)
+        
+        return answer, sources
+    except Exception as e:
+        error_msg = f"Ошибка обработки вопроса: {str(e)}"
+        log_message(error_msg)
+        return error_msg, ""
+
+def get_sources_for_question(question):
+    try:
+        nodes = index_retriever.retrieve_nodes(question)
+        if not nodes:
+            return "<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; text-align: center;'>Источники не найдены</div>"
+        
+        sources_html = "<div style='background-color: #2d3748; color: white; padding: 15px; border-radius: 10px;'>"
+        sources_html += "<h3 style='color: #4fd1c7; margin-top: 0;'>📚 Источники:</h3>"
+        
+        for i, node in enumerate(nodes[:5], 1):
+            source_text = node.text[:200] + "..." if len(node.text) > 200 else node.text
+            sources_html += f"<div style='margin: 10px 0; padding: 10px; background-color: #4a5568; border-radius: 5px;'>"
+            sources_html += f"<strong>Источник {i}:</strong><br>"
+            sources_html += f"<small>{source_text}</small>"
+            sources_html += "</div>"
+        
+        sources_html += "</div>"
+        return sources_html
+        
+    except Exception as e:
+        log_message(f"Ошибка получения источников: {str(e)}")
+        return "<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; text-align: center;'>Ошибка загрузки источников</div>"
 
 def handle_model_switch(model_name):
-    if index_retriever is None:
-        return "Система не инициализирована"
-    return index_retriever.switch_model(model_name)
+    try:
+        return index_retriever.switch_model(model_name)
+    except Exception as e:
+        error_msg = f"Ошибка переключения модели: {str(e)}"
+        log_message(error_msg)
+        return f"❌ {error_msg}"
 
 def get_current_model_status():
-    if index_retriever is None:
-        return "Система не инициализирована"
-    return f"Текущая модель: {index_retriever.get_current_model()}"
+    try:
+        if not index_retriever.is_initialized():
+            return "Система не инициализирована"
+        return f"Текущая модель: {index_retriever.get_current_model()}"
+    except Exception as e:
+        return "Ошибка получения статуса модели"
 
 def get_chat_history_html():
     if chat_handler is None:

index_retriever.py

@@ -78,7 +78,7 @@ def initialize_models(documents):
         return False
 
 def create_query_engine():
-    global query_engine, vector_index
+    global query_engine
     
     try:
         log_message(f"Применяется промпт: {PROMPT_SIMPLE_POISK[:100]}...")
@@ -118,8 +118,6 @@ def create_query_engine():
         raise
 
 def query(question):
-    global query_engine, current_model
-    
     if query_engine is None:
         log_message("❌ Query engine не инициализирован")
         return "❌ Система не инициализирована"
@@ -141,7 +139,7 @@ def query(question):
         return f"❌ {error_msg}"
 
 def switch_model(model_name):
-    global current_model, vector_index
+    global current_model
     
     try:
         log_message(f"Переключение на модель: {model_name}")
@@ -163,8 +161,6 @@ def switch_model(model_name):
         return f"❌ {error_msg}"
 
 def rerank_nodes(query_text, nodes, top_k=10):
-    global reranker
-    
     if not nodes or not reranker:
         return nodes[:top_k]
     
@@ -189,8 +185,6 @@ def rerank_nodes(query_text, nodes, top_k=10):
         return nodes[:top_k]
 
 def retrieve_nodes(question):
-    global query_engine
-    
     if query_engine is None:
         return []