Update app.py

app.py

@@ -144,7 +144,6 @@ def send_twilio_message(client, conversation_sid, body):
 def get_latest_whatsapp_conversation_sid(client):
     try:
         conversations = client.conversations.v1.conversations.list(limit=20)
-        # Filter conversations created after app start time
         filtered = [
             c for c in conversations
             if c.date_created and c.date_created > APP_START_TIME
@@ -157,6 +156,35 @@ def get_latest_whatsapp_conversation_sid(client):
         print("Error fetching valid conversation SID:", e)
     return None
 
+# ---------------- Load Orders ----------------
+def load_orders():
+    orders_path = "docs/orders.json"
+    try:
+        with open(orders_path, "r", encoding="utf-8") as f:
+            return json.load(f)
+    except Exception as e:
+        print(f"Error loading orders: {e}")
+        return {}
+
+def extract_order_id(text):
+    pattern = r"\bORD\d{3,}\b"
+    match = re.search(pattern, text, re.IGNORECASE)
+    if match:
+        return match.group(0).upper()
+    return None
+
+def format_order_response(order_id, order_data):
+    if not order_data:
+        return f"Sorry, I could not find details for order ID {order_id}."
+    details = [
+        f"Order ID: {order_id}",
+        f"Customer Name: {order_data.get('customer_name', 'N/A')}",
+        f"Address: {order_data.get('address', 'N/A')}",
+        f"Items: {', '.join(order_data.get('items', []))}",
+        f"Status: {order_data.get('status', 'N/A')}",
+    ]
+    return "\n".join(details)
+
 # ---------------- Knowledge Base Setup ----------------
 def setup_knowledge_base():
     folder = "docs"
@@ -169,6 +197,9 @@ def setup_knowledge_base():
         elif f.endswith(".docx"):
             text += clean_extracted_text(extract_text_from_docx(path)) + "\n"
         elif f.endswith(".json"):
+            # Skip orders.json here to avoid mixing with KB text
+            if f == "orders.json":
+                continue
             text += load_json_data(path) + "\n"
         elif f.endswith(".csv"):
             with open(path, newline='', encoding='utf-8') as csvfile:
@@ -185,7 +216,9 @@ def process_messages_loop():
     embeddings = embed_model.encode(text_chunks)
     index = faiss.IndexFlatL2(embeddings.shape[1])
     index.add(embeddings)
-
+    
+    orders = load_orders()  # Load orders once at start
+    
     seen_sids = set()
 
     while True:
@@ -198,8 +231,14 @@ def process_messages_loop():
         if message and message["sid"] not in seen_sids:
             seen_sids.add(message["sid"])
             question = message["body"]
-            chunks = retrieve_chunks(question, index, embed_model, text_chunks)
-            answer = generate_answer_with_groq(question, "\n\n".join(chunks))
+
+            order_id = extract_order_id(question)
+            if order_id and order_id in orders:
+                answer = format_order_response(order_id, orders[order_id])
+            else:
+                chunks = retrieve_chunks(question, index, embed_model, text_chunks)
+                answer = generate_answer_with_groq(question, "\n\n".join(chunks))
+
             send_twilio_message(twilio_client, conversation_sid, answer)
 
         time.sleep(5)
@@ -208,6 +247,6 @@ def process_messages_loop():
 st.title("ToyShop WhatsApp Assistant (Groq + Twilio)")
 
 if st.button("Start WhatsApp Bot"):
-    thread = threading.Thread(target=process_messages_loop)
+    thread = threading.Thread(target=process_messages_loop, daemon=True)
     thread.start()
-    st.success("WhatsApp assistant started and monitoring for new messages.")
+    st.success("WhatsApp assistant started and monitoring for new messages.")