Update app.py

app.py

@@ -67,36 +67,30 @@ with gr.Blocks() as demo:
             gr.Markdown("Powered by **smolagents**")
 
         with gr.Column(scale=3):
-            output = gr.Chatbot(label="Agent", type="messages" )  # 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)]
+            output = gr.Chatbot(label="Agent", type="messages")
 
     def run_query(prompt, file):
         global agent_instance
         agent_instance = build_agent(file_path=file)
         result = agent_instance.run(prompt)
-        
-    # If result is a dict, convert to readable string (optional)
+
+        # Handle different result types to convert to string for chatbot output
         if isinstance(result, dict):
             result = "\n\n".join(f"**{k}**: {v}" for k, v in result.items())
-
-    # Return the conversation in the new format for gr.Chatbot (type="messages")
+        elif isinstance(result, list):
+            if all(isinstance(item, dict) and "name" in item and "starter" in item for item in result):
+                result = "\n\n".join(f"{item['name']}: {item['starter']}" for item in result)
+            else:
+                result = str(result)
+        else:
+            result = str(result)
+
+        # Return as list of message dicts for Gradio chatbot type="messages"
         return [
             {"role": "user", "content": prompt},
             {"role": "assistant", "content": result}
         ]
 
-
     submit.click(fn=run_query, inputs=[prompt, guest_file], outputs=output)
 
 if __name__ == "__main__":