Return file path instead of PIL Image for grounding visualization

app.py

@@ -66,7 +66,6 @@ agent = Agent(
 print(f"Tools loaded: {len(tools)}")
 
 import glob
-from PIL import Image
 
 def chat(message, history):
     config = {"configurable": {"thread_id": "default"}}
@@ -95,10 +94,9 @@ def chat(message, history):
         # Get the most recent visualization
         viz_files.sort(key=os.path.getmtime, reverse=True)
         latest_viz = viz_files[0]
-        try:
-            viz_image = Image.open(latest_viz)
-        except:
-            pass
+        # Return the file path directly - Gradio can handle it
+        if os.path.exists(latest_viz):
+            viz_image = latest_viz
 
     return assistant_message, viz_image