Update utils.py

utils.py

@@ -197,8 +197,8 @@ def debug_search_tables(vector_index, search_term="С-25"):
 
 from documents_prep import normalize_text
 
-# MODIFIED: Update answer_question function
-def answer_question(question, query_engine, reranker, current_model, chunks_df=None):
+# MODIFIED: Update answer_question function signature
+def answer_question(question, query_engine, reranker, current_model, chunks_df=None, rerank_top_k=20):
     # NORMALIZE the question to convert C to С
     normalized_question = normalize_text(question)
     
@@ -226,8 +226,9 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
             log_message(f"  [{i+1}] {doc_id} - Table {table_num}: {table_title[:50]}")
         log_message(f"UNIQUE NODES: {len(unique_retrieved)} nodes")
         
-        # Simple reranking with NORMALIZED question
-        reranked_nodes = rerank_nodes(normalized_question, unique_retrieved, reranker, top_k=20)
+        # Simple reranking with NORMALIZED question and PARAMETERIZED top_k
+        reranked_nodes = rerank_nodes(normalized_question, unique_retrieved, reranker, 
+                                     top_k=rerank_top_k)  # NOW PARAMETERIZED
         
         # Direct query without formatting - use normalized question
         response = query_engine.query(normalized_question)
@@ -243,7 +244,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
         <h3 style='color: #63b3ed; margin-top: 0;'>Ответ (Модель: {current_model}):</h3>
         <div style='line-height: 1.6; font-size: 16px;'>{response.response}</div>
         <div style='margin-top: 15px; padding-top: 10px; border-top: 1px solid #4a5568; font-size: 14px; color: #a0aec0;'>
-        Время обработки: {processing_time:.2f} секунд
+        Время обработки: {processing_time:.2f} секунд 
         </div>
         </div>"""
         log_message(f"Model Answer: {response.response}")