Update app.py

app.py

@@ -1,8 +1,9 @@
-import re
-import time
-import gradio as gr
-import pandas as pd
 import requests
+import pandas as pd
+import gradio as gr
+import time
+import re
+import os
 
 API_BASE_URL = "https://sale-agent-m179.onrender.com"
 
@@ -20,9 +21,8 @@ def clear_chat():
     return [], {}, {}, [], ""
 
 
-def respond_to_chat(message, history, state, selected_model):
+def respond_to_chat(message, history, state, selected_model, debug_mode):
     """Handle chat responses with image support"""
-    # print(f"🔄 === DEBUG STATE ===\n Chat request with model: {selected_model}")
     start = time.perf_counter()
     # Handle multimodal input - message is a dict with 'text' and 'files' keys
     if isinstance(message, dict):
@@ -45,11 +45,11 @@ def respond_to_chat(message, history, state, selected_model):
         if image_path:
             with open(image_path, 'rb') as f:
                 files = {"image": f}
-                data = {"message": user_query}
+                data = {"message": user_query, "debug": debug_mode}
                 resp = requests.post(
                     f"{API_BASE_URL}/chat_with_image", files=files, data=data, timeout=120)
         else:
-            payload = {"message": user_query}
+            payload = {"message": user_query, "debug": debug_mode}
             resp = requests.post(f"{API_BASE_URL}/chat",
                                  json=payload, timeout=120)
 
@@ -91,7 +91,8 @@ def show_specs(state, history=None):
         df = pd.DataFrame({}, columns=columns)
         markdown_table = df.to_markdown(index=False)
         # Fix: Use proper ChatInterface message format
-        history.append({"role": "assistant", "content": f"📄 Thông số kỹ thuật\n{markdown_table}"})
+        history.append(
+            {"role": "assistant", "content": f"📄 Thông số kỹ thuật\n{markdown_table}"})
         return history
 
     # print(specs_map)
@@ -187,7 +188,8 @@ def show_specs(state, history=None):
 
     markdown_table = df.to_markdown(index=False)
     # Fix: Use proper ChatInterface message format
-    history.append({"role": "assistant", "content": f"📄 Thông số kỹ thuật\n{markdown_table}"})
+    history.append(
+        {"role": "assistant", "content": f"📄 Thông số kỹ thuật\n{markdown_table}"})
     return history
 
 
@@ -202,7 +204,8 @@ def show_advantages(state, history=None):
         df = pd.DataFrame([["Không có ưu điểm", ""]], columns=columns)
         markdown_table = df.to_markdown(index=False)
         # Fix: Use proper ChatInterface message format
-        history.append({"role": "assistant", "content": f"💡 Ưu điểm nổi trội\n{markdown_table}"})
+        history.append(
+            {"role": "assistant", "content": f"💡 Ưu điểm nổi trội\n{markdown_table}"})
         return history
 
     for prod_id, data in specs_map.items():
@@ -226,7 +229,8 @@ def show_advantages(state, history=None):
 
     markdown_table = df.to_markdown(index=False)
     # Fix: Use proper ChatInterface message format
-    history.append({"role": "assistant", "content": f"💡 Ưu điểm nổi trội\n{markdown_table}"})
+    history.append(
+        {"role": "assistant", "content": f"💡 Ưu điểm nổi trội\n{markdown_table}"})
     return history
 
 
@@ -238,7 +242,8 @@ def show_solution_packages(state, history=None):
 
     if not specs_map:
         # Fix: Use proper ChatInterface message format
-        history.append({"role": "assistant", "content": "📦 Không có gói sản phẩm nào"})
+        history.append(
+            {"role": "assistant", "content": "📦 Không có gói sản phẩm nào"})
         return history
 
     # Filter only solution packages
@@ -249,7 +254,8 @@ def show_solution_packages(state, history=None):
 
     if not solution_packages:
         # Fix: Use proper ChatInterface message format
-        history.append({"role": "assistant", "content": "📦 Không có gói sản phẩm nào"})
+        history.append(
+            {"role": "assistant", "content": "📦 Không có gói sản phẩm nào"})
         return history
 
     # Build markdown content for each package
@@ -322,7 +328,7 @@ css = """
     .gradio-container {
         position: relative;
     }
-    .model-dropdown {
+    .model-dropdown, .debug-dropdown {
         background: linear-gradient(135deg, #667eea, #764ba2) !important;
         color: white !important;
         border: none !important;
@@ -333,15 +339,15 @@ css = """
         min-height: 32px !important;
         box-shadow: 0 2px 4px rgba(0,0,0,0.15) !important;
     }
-    .model-dropdown select {
+    .model-dropdown select, .debug-dropdown select {
         color: white !important;
         background: transparent !important;
         cursor: pointer !important;
     }
-    .model-dropdown select:focus {
+    .model-dropdown select:focus, .debug-dropdown select:focus {
         outline: none;
     }
-    .model-dropdown select:not(:focus) {
+    .model-dropdown select:not(:focus), .debug-dropdown select:not(:focus) {
         pointer-events: auto;
     }
 """
@@ -367,6 +373,17 @@ with gr.Blocks(fill_height=True, css=css) as demo:
                 container=False
             )
 
+        with gr.Column(scale=1, min_width=120):
+            debug_dropdown = gr.Dropdown(
+                choices=["Normal", "Debug"],
+                value="Normal",
+                label="Mode",
+                show_label=False,
+                elem_classes="debug-dropdown",
+                interactive=True,
+                container=False
+            )
+
         with gr.Column(scale=1, min_width=30):
             clear_btn = gr.Button(
                 "✨ Xoá", variant="secondary", size="sm", elem_classes="custom-clear-btn")
@@ -384,14 +401,13 @@ with gr.Blocks(fill_height=True, css=css) as demo:
         fn=respond_to_chat,
         multimodal=True,
         type="messages",
-        cache_examples=False,
         examples=[
             ["Tìm sản phẩm bình giữ nhiệt dung tích dưới 2 lít."],
             ["Tìm sản phẩm ổ cắm thông minh"],
             ["Tư vấn cho tôi đèn học chống cận cho con gái của tôi học lớp 6"]
         ],
         title="_",
-        additional_inputs=[state, model_dropdown],
+        additional_inputs=[state, model_dropdown, debug_dropdown],
         chatbot=chatbot
     )
 
@@ -407,12 +423,12 @@ with gr.Blocks(fill_height=True, css=css) as demo:
     btn_spec.click(fn=show_specs, inputs=[state, chatbot], outputs=[chatbot])
     btn_adv.click(fn=show_advantages,
                   inputs=[state, chatbot],
-                  outputs=[chatbot]
-                  )
+                  outputs=[chatbot])
 
     # Gói sản phẩm button shows solution packages in structured format
-    btn_pckg.click(fn=show_solution_packages, inputs=[
-                   state, chatbot], outputs=[chatbot])
+    btn_pckg.click(fn=show_solution_packages,
+                   inputs=[state, chatbot],
+                   outputs=[chatbot])
 
     # Model change handler -> call API
     def _on_model_change(model_name):