new r + stee product expansion

utils.py

@@ -254,7 +254,11 @@ def enhance_query_for_steel_grades(query):
 def answer_question(question, query_engine, reranker, current_model, chunks_df=None, rerank_top_k=20):
     
     normalized_question = normalize_text(question)
-    normalized_question_2, query_changes, change_list = normalize_steel_designations(question)  # FIX: 3 values
+    normalized_question_2, query_changes, change_list = normalize_steel_designations(question)
+    
+    # FIX: Add query enhancement for steel grades
+    enhanced_question = enhance_query_for_steel_grades(normalized_question_2)
+    
     if change_list:
         log_message(f"Query changes: {', '.join(change_list)}")    
     if query_engine is None:
@@ -262,12 +266,13 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
     
     try:
         start_time = time.time()
-        retrieved_nodes = query_engine.retriever.retrieve(normalized_question_2)
+        # FIX: Use enhanced_question instead of normalized_question_2
+        retrieved_nodes = query_engine.retriever.retrieve(enhanced_question)
         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"enhanced query: {enhanced_question}")  # ADD THIS LOG
         log_message(f"Steel grades normalized in query: {query_changes}")
-
         
         log_message(f"RETRIEVED: {len(retrieved_nodes)} nodes")
         
@@ -295,11 +300,11 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
         log_message(f"UNIQUE NODES: {len(unique_retrieved)} nodes")
         
         # Simple reranking with NORMALIZED question and PARAMETERIZED top_k
-        reranked_nodes = rerank_nodes(normalized_question_2, unique_retrieved, reranker, 
+        reranked_nodes = rerank_nodes(enhanced_question, unique_retrieved, reranker, 
                                      top_k=rerank_top_k)  # NOW PARAMETERIZED
         
         # Direct query without formatting - use normalized question
-        response = query_engine.query(normalized_question_2)
+        response = query_engine.query(enhanced_question)
         
         end_time = time.time()
         processing_time = end_time - start_time