Update app.py

app.py

@@ -1,6 +1,5 @@
 import streamlit as st
 import pandas as pd
-from datetime import datetime
 import json
 import os
 import logging
@@ -335,6 +334,41 @@ def load_models():
         """)
         raise
 
+def run_bm25_diagnostic(hybrid_search, question, top_k=5):
+    """Отдельная функция для диагностики BM25 с явной передачей hybrid_search"""
+    try:
+        logger.info(f"Запуск диагностики BM25 для вопроса: '{question}'")
+        
+        if not hybrid_search:
+            error_msg = "Гибридный поиск (hybrid_search) не инициализирован"
+            logger.error(error_msg)
+            raise ValueError(error_msg)
+        
+        if not hybrid_search.bm25:
+            error_msg = "BM25 не был создан при инициализации HybridSearch"
+            logger.error(error_msg)
+            raise ValueError(error_msg)
+        
+        # Запуск поиска с подробным логированием
+        logger.info(f"Поиск BM25 с top_k={top_k}")
+        results = hybrid_search.search(question, top_k=top_k)
+        
+        if not results:
+            logger.info("BM25 вернул 0 результатов (возможно, низкие оценки)")
+        else:
+            logger.info(f"Найдено результатов BM25: {len(results)}")
+            for i, res in enumerate(results, 1):
+                logger.info(
+                    f"Результат #{i}: Оценка={res['score']:.2f}, "
+                    f"Тип={res.get('type', 'unknown')}, "
+                    f"Текст={res['text'][:50]}..."
+                )
+        
+        return results
+    except Exception as e:
+        logger.error(f"Ошибка в run_bm25_diagnostic: {str(e)}", exc_info=True)
+        raise
+
 # Подключение к SQLite базе
 def get_db_connection(db_path):
     try:
@@ -628,7 +662,7 @@ def get_answer(question):
         return answer
     
     # 5. Ответ по умолчанию
-    answer = "К сожалению, не удалось найти точный ответ в базе знаний. Попробуйте переформулировать вопрос."
+    answer = "К сожалению, не удалось найти точный ответ. Попробуйте переформулировать вопрос."
     save_log(question, answer)
     return answer
 
@@ -710,18 +744,48 @@ if st.sidebar.button("Запустить диагностику BM25"):
         st.write(f"'{test_question}'")
         
         try:
-            results = hybrid_search.search(test_question, top_k=5) if hybrid_search else []
+            # Логирование начала диагностики
+            logger.info(f"Запущена диагностика BM25 для вопроса: '{test_question}'")
+            
+            # Проверка инициализации
+            if not hybrid_search:
+                error_msg = "HybridSearch не инициализирован!"
+                logger.error(error_msg)
+                st.error(error_msg)
+                st.stop()
+            
+            if not hybrid_search.bm25:
+                error_msg = "BM25 не был инициализирован в HybridSearch!"
+                logger.error(error_msg)
+                st.error(error_msg)
+                st.stop()
+            
+            # Выполнение поиска
+            results = hybrid_search.search(test_question, top_k=5)
+            logger.info(f"Найдено результатов BM25: {len(results)}")
             
             if not results:
-                st.write("Не найдено результатов или BM25 не инициализирован")
+                msg = "BM25 не вернул результатов (возможно, низкие оценки совпадения)"
+                logger.warning(msg)
+                st.warning(msg)
             else:
+                st.success(f"Найдено результатов: {len(results)}")
                 for i, res in enumerate(results, 1):
-                    st.write(f"#### Результат {i} (оценка: {res['score']:.2f})")
+                    logger.info(
+                        f"Результат {i}: Оценка={res['score']:.2f}, "
+                        f"Тип={res.get('type', 'unknown')}, "
+                        f"Текст={res['text'][:50]}..."
+                    )
+                    
+                    st.markdown(f"#### Результат {i} (оценка: {res['score']:.2f})")
                     st.write(f"**Текст:** {res['text']}")
-                    st.write(f"**Источник:** {res['source']}")
+                    st.write(f"**Источник:** {res.get('source', 'нет данных')}")
                     st.write("---")
+                    
         except Exception as e:
-            st.error(f"Ошибка при диагностике BM25: {e}")
+            error_msg = f"Критическая ошибка при диагностике BM25: {str(e)}"
+            logger.exception(error_msg)
+            st.error(error_msg)
 
 # Проверка содержимого SQLite базы
 if st.sidebar.button("Проверить SQLite на наличие термина"):