Update app.py

app.py

@@ -55,6 +55,10 @@ agent = CodeAgent(
 
 
 
+# Gradio Gallery and visualization output
+visual_output = gr.Gallery(label="Generated Visualizations", columns=3, height=400)
+
+# Fix in the `run_agent` function to handle agent results correctly
 def run_agent(_):
     if df_global is None:
         return "Please upload a file first.", []
@@ -74,10 +78,10 @@ def run_agent(_):
        - Insightful relationships between key columns.
        - At least 3 visualizations showing important trends.
     4. Derive at least 3 actionable real-world insights.
-    5. Save all visualizations to ./figures/ directory.
-    Return a JSON object with keys:
-      - 'insights': clean bullet-point insights.
-      - 'figures': list of file paths of generated visualizations.
+       5. Save all visualizations to ./figures/ directory.
+       Return a JSON object with keys:
+         - 'insights': clean bullet-point insights.
+         - 'figures': list of file paths of generated visualizations.
     """
 
     result = agent.run(
@@ -85,16 +89,16 @@ def run_agent(_):
         additional_args={"source_file": temp_file.name}
     )
 
-    # If the result is text, attempt to parse it as JSON
-    try:
-        result_str = result.strip()
-        result_dict = json.loads(result_str)
+    # Process result as a dictionary if possible
+    if isinstance(result, dict):
+        insights = result.get("insights", "No insights generated.")
+        image_paths = result.get("figures", [])
+    else:
+        insights = "Error: The result is not in the expected format."
+        image_paths = []
+
+    return insights, image_paths
 
-        insights = result_dict.get("insights", "No insights generated.")
-        image_paths = result_dict.get("figures", [])
-        return insights, image_paths
-    except Exception as e:
-        return f"Error parsing agent response: {e}", []