Update app.py

app.py

@@ -68,71 +68,73 @@ def semantic_search(query, k=5):
 def handle_query(message, customer_id=""):
     text = message.lower()
 
-    # 1) Find a cuisine
+    # 1. cuisine search
     if "find" in text and "restaurant" in text:
-        for cuisine in df['cuisine_type'].str.lower().unique():
+        for cuisine in df["cuisine_type"].str.lower().unique():
             if cuisine in text:
                 return find_by_cuisine(cuisine).to_html(index=False)
         return semantic_search(message).to_html(index=False)
 
-    # 2) Best places
+    # 2. best-rated query
     if "best" in text:
-        for cuisine in df['cuisine_type'].str.lower().unique():
+        for cuisine in df["cuisine_type"].str.lower().unique():
             if cuisine in text:
                 return best_rated_by_cuisine(cuisine).to_html(index=False)
         return semantic_search(message).to_html(index=False)
 
-    # 3) Cheap recommendations
-    if "cheap" in text or "cheapest" in text or "value" in text:
+    # 3. cheap places
+    if "cheap" in text or "value" in text:
         return cheapest_high_rated().to_html(index=False)
 
-    # 4) Personalized recall
+    # 4. personalized recall
     if "what did i order" in text:
         m = re.search(r"on (\w+)", text)
         if not customer_id:
             return "Please enter customer_id."
         if not m:
-            return "Please specify the day, e.g. 'on Tuesday'"
+            return "Please specify the day (e.g., Tuesday)"
         day = m.group(1)
-        res = personalized_recall(customer_id, day)
-        if res.empty:
-            return "No results found."
-        return res.to_html(index=False)
+        r = personalized_recall(customer_id, day)
+        if r.empty:
+            return "No matching records."
+        return r.to_html(index=False)
 
-    # 5) Fallback semantic
     return semantic_search(message).to_html(index=False)
 
 
 # -------------------------------
-# Chatbot Function (OLD FORMAT)
+# CHATBOT FUNCTION (DICTIONARY FORMAT)
 # -------------------------------
 def chatbot_fn(history, message, customer_id):
     reply_html = handle_query(message, customer_id)
 
-    # OLD GRADIO FORMAT: append tuple
-    history.append((message, "Here are the results below 👇"))
+    # append user message
+    history.append({"role": "user", "content": message})
+
+    # append assistant message
+    history.append({"role": "assistant", "content": "Here are the results 👇"})
 
     return history, "", reply_html
 
 
 # -------------------------------
-# Gradio UI (Compatible with Spaces)
+# INTERFACE
 # -------------------------------
 with gr.Blocks() as demo:
     gr.Markdown("## 🍽️ Restaurant Guide Chatbot")
 
-    chat = gr.Chatbot(label="Chat History")
-    html_output = gr.HTML(label="Search Results")
+    chat = gr.Chatbot(label="Chat History")   # no type arg
+    html_out = gr.HTML(label="Search Results")
 
     with gr.Row():
-        user_msg = gr.Textbox(placeholder="Type your message…")
-        cust_id = gr.Textbox(label="Customer ID (optional)")
-        send = gr.Button("Send")
+        msg = gr.Textbox(placeholder="Ask me anything…")
+        cid = gr.Textbox(label="Customer ID (optional)")
+        btn = gr.Button("Send")
 
-    send.click(
+    btn.click(
         chatbot_fn,
-        inputs=[chat, user_msg, cust_id],
-        outputs=[chat, user_msg, html_output]
+        inputs=[chat, msg, cid],
+        outputs=[chat, msg, html_out]
     )
 
 demo.launch()