Update app.py

app.py

@@ -352,13 +352,21 @@ if "bot_start_time_utc" not in st.session_state: st.session_state.bot_start_time
 if "processed_message_sids" not in st.session_state: st.session_state.processed_message_sids = set()
 if "manual_chat_history" not in st.session_state: st.session_state.manual_chat_history = []
 
+# --- Helper: Simple Intent Classifier ---
 # --- Helper: Simple Intent Classifier ---
 def simple_intent_classifier(query):
     query_lower = query.lower()
-    if any(k in query_lower for k in ["order", "status", "track", "delivery"]): return "ORDER_STATUS"
+    if any(k in query_lower for k in ["order", "status", "track", "delivery"]):
+        # Attempt to extract an order ID more robustly
+        import re
+        match = re.search(r'(ord[a-z0-9]+)', query_lower) # Regex to find 'ord' followed by alphanumeric
+        if match:
+            return "ORDER_STATUS", match.group(1).upper() # Return intent and extracted ID
+        return "ORDER_STATUS", None # Indicate order status intent but no ID found yet
+
     if any(k in query_lower for k in ["product", "item", "buy", "price", "feature", "stock"]): return "PRODUCT_INFO"
     if any(k in query_lower for k in ["return", "policy", "refund", "exchange", "faq", "question", "how to", "support"]): return "GENERAL_POLICY_FAQ"
-    return "UNKNOWN"
+    return "UNKNOWN", None # Return intent and None for ID if unknown
 
 # --- Main Application Controls ---
 col1, col2, col3, col4 = st.columns(4)
@@ -456,22 +464,34 @@ if st.session_state.get("app_started") and st.session_state.get("rag_pipeline_re
         with st.chat_message("user"): st.markdown(user_query_manual)
 
         with st.spinner("Thinking..."):
-            intent = simple_intent_classifier(user_query_manual)
+            intent_result = simple_intent_classifier(user_query_manual) # Get both intent and potential_id
+            intent = intent_result[0]
+            potential_oid_from_intent = intent_result[1] # This is the extracted ID if any
+
             context_for_llm, raw_context_data = "No specific context.", None
 
             if intent == "ORDER_STATUS":
-                words = user_query_manual.split()
-                potential_oid = next((w for w in words if w.upper().startswith("ORD")), None)
-                if potential_oid:
-                    raw_context_data = get_order_details(potential_oid.upper(), st.session_state.customer_orders_data)
+                order_id_to_check = None
+                if potential_oid_from_intent:
+                    order_id_to_check = potential_oid_from_intent
+                else:
+                    # Fallback or additional parsing if needed
+                    # For example, if 'ORD1001' is not found by regex, try splitting words
+                    words = user_query_manual.split()
+                    # Look for something that starts with 'ORD' and is not just 'order' or 'order_id'
+                    order_id_to_check = next((w for w in words if w.upper().startswith("ORD") and len(w) > 3), None) # ensure it's longer than just "ORD"
+
+                if order_id_to_check:
+                    raw_context_data = get_order_details(order_id_to_check.upper(), st.session_state.customer_orders_data)
                     context_for_llm = f"Order Details: {raw_context_data}"
                 else:
-                    context_for_llm = "Please provide an Order ID."
+                    context_for_llm = "Please provide a valid Order ID (e.g., ORD1234)."
                     raw_context_data = {"message": "Order ID needed."}
             elif intent == "PRODUCT_INFO":
                 raw_context_data = get_product_info(user_query_manual, st.session_state.products_data)
                 context_for_llm = f"Product Information: {raw_context_data}"
             elif intent == "GENERAL_POLICY_FAQ" or intent == "UNKNOWN":
+                # ... (rest of your existing logic for these intents) ...
                 if st.session_state.faiss_index_pdfs and st.session_state.embedding_model:
                     k_val = 2 if intent == "GENERAL_POLICY_FAQ" else 1
                     retrieved_chunks = search_faiss_index(st.session_state.faiss_index_pdfs, user_query_manual,
@@ -491,7 +511,6 @@ if st.session_state.get("app_started") and st.session_state.get("rag_pipeline_re
                 st.markdown(llm_response)
                 if raw_context_data:
                     with st.expander("Retrieved Context"):
-                        # This block is the one that needs the fix
                         try:
                             if isinstance(raw_context_data, str) and (raw_context_data.strip().startswith('{') or raw_context_data.strip().startswith('[')):
                                 st.json(json.loads(raw_context_data))