Add grounding visualization image display in chat interface

app.py

@@ -65,6 +65,8 @@ agent = Agent(
 
 print(f"Tools loaded: {len(tools)}")
 
+import glob
+
 def chat(message, history):
     config = {"configurable": {"thread_id": "default"}}
 
@@ -81,7 +83,21 @@ def chat(message, history):
         {"messages": [("user", message)]},
         config=config
     )
-    return response["messages"][-1].content
+
+    # Extract text response
+    assistant_message = response["messages"][-1].content
+
+    # Check for grounding visualization images
+    viz_files = glob.glob("temp/grounding_*.png")
+    if viz_files:
+        # Get the most recent visualization
+        viz_files.sort(key=os.path.getmtime, reverse=True)
+        latest_viz = viz_files[0]
+
+        # Append image reference that Gradio can render
+        assistant_message += f"\n\n**Grounding Visualization:**\n\n<img src='/file={latest_viz}' style='max-width: 600px;'/>"
+
+    return assistant_message
 
 demo = gr.ChatInterface(
     fn=chat,