Update app.py

app.py

@@ -4,7 +4,10 @@ import gradio as gr
 import time
 import re
 import os
+from dotenv import load_dotenv
+load_dotenv()
 
+# API_BASE_URL = os.getenv("API_BASE_URL")
 API_BASE_URL = "https://sale-agent-m179.onrender.com"
 
 
@@ -57,14 +60,15 @@ def respond_to_chat(message, history, state, selected_model, debug_mode):
             j = resp.json()
             response = j.get("response", "")
             specs_advantages = j.get("specs_advantages")
+            solution_packages = j.get("solution_packages")
             raw_documents = j.get("raw_documents")
             outputs = j.get("outputs")
         else:
             response = f"Error: API status {resp.status_code}"
-            specs_advantages, raw_documents, outputs = None, None, None
+            specs_advantages, solution_packages, raw_documents, outputs = None, None, None, None
     except Exception as e:
         response = f"Error calling API: {e}"
-        specs_advantages, raw_documents, outputs = None, None, None
+        specs_advantages, solution_packages, raw_documents, outputs = None, None, None, None
 
     end = time.perf_counter()
 
@@ -74,6 +78,8 @@ def respond_to_chat(message, history, state, selected_model, debug_mode):
     # specs_advantages is now always a dict for both products and solutions
     if isinstance(specs_advantages, dict):
         state["specs_advantages"] = specs_advantages
+    if solution_packages:
+        state["solution_packages"] = solution_packages
     state["raw_documents"] = raw_documents
     state["outputs"] = outputs
 
@@ -236,22 +242,10 @@ def show_advantages(state, history=None):
 
 def show_solution_packages(state, history=None):
     """Show solution packages in a structured format"""
-    specs_map = state.get("specs_advantages", {})
+    solution_packages = state.get("solution_packages", [])
     if history is None:
         history = []
 
-    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"})
-        return history
-
-    # Filter only solution packages
-    solution_packages = {}
-    for key, data in specs_map.items():
-        if data.get("model") == "Gói giải pháp":
-            solution_packages[key] = data
-
     if not solution_packages:
         # Fix: Use proper ChatInterface message format
         history.append(
@@ -261,9 +255,9 @@ def show_solution_packages(state, history=None):
     # Build markdown content for each package
     markdown_content = "## 📦 Gói sản phẩm\n\n"
 
-    for key, data in solution_packages.items():
-        spec_content = data.get("specification", "")
-        markdown_content += spec_content + "\n\n"
+    # solution_packages is a list
+    for package in solution_packages:
+        markdown_content += f"{package}\n\n"
 
     # Fix: Use proper ChatInterface message format
     history.append({"role": "assistant", "content": markdown_content})
@@ -452,4 +446,4 @@ with gr.Blocks(fill_height=True, css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()