normalize steel

utils.py

@@ -195,19 +195,24 @@ def debug_search_tables(vector_index, search_term="С-25"):
     
     return matching
 
-from documents_prep import normalize_text
+from documents_prep import normalize_text, normalize_steel_designations
 
 def answer_question(question, query_engine, reranker, current_model, chunks_df=None, rerank_top_k=20):
+    
     normalized_question = normalize_text(question)
+    log_message(f"Normalized question: {normalized_question}")
+    normalized_question_2 = normalize_steel_designations(normalized_question)
+    log_message(f"After steel normalization: {normalized_question_2}")
     
     if query_engine is None:
         return "<div style='background-color: #e53e3e; color: white; padding: 20px; border-radius: 10px;'>Система не инициализирована</div>", "", ""
     
     try:
         start_time = time.time()
-        retrieved_nodes = query_engine.retriever.retrieve(normalized_question)
+        retrieved_nodes = query_engine.retriever.retrieve(normalized_question_2)
         log_message(f"user query: {question}")
         log_message(f"normalized query: {normalized_question}")
+        log_message(f"after steel normalization: {normalized_question_2}")
         
         log_message(f"RETRIEVED: {len(retrieved_nodes)} nodes")