Update app.py

app.py

@@ -10,8 +10,7 @@ from retriever import load_guest_dataset
 # Constants
 SAMPLE_FILE = "sample_guests.csv"
 
-# Generate sample dataset
-
+# Generate sample dataset if not already present
 def generate_sample_guest_csv():
     if not os.path.exists(SAMPLE_FILE):
         guest_dataset = datasets.load_dataset("agents-course/unit3-invitees", split="train")
@@ -68,12 +67,18 @@ with gr.Blocks() as demo:
             gr.Markdown("Powered by **smolagents**")
 
         with gr.Column(scale=3):
-            output = gr.Chatbot(label="Agent")
+            output = gr.Chatbot(label="Agent")  # default format (list of tuples)
 
     def run_query(prompt, file):
         global agent_instance
         agent_instance = build_agent(file_path=file)
         result = agent_instance.run(prompt)
+
+        # Ensure result is string for chatbot format
+        if isinstance(result, dict):
+            # Convert dict to readable string if agent returns JSON-like structure
+            result = "\n\n".join(f"**{k}**: {v}" for k, v in result.items())
+
         return [("User", prompt), ("Agent", result)]
 
     submit.click(fn=run_query, inputs=[prompt, guest_file], outputs=output)