Update app.py

app.py

@@ -6,6 +6,8 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 import os
+import IPython.display as display
+
 
 from Gradio_UI import GradioUI
 
@@ -27,7 +29,10 @@ def generate_and_show_surfer_image(next_tide: str) -> str:
     prompt = f"Generate a surfer {action}"
     try:
         image_result = image_generation_tool(prompt)
-        return image_result
+        if isinstance(image_result, str) and image_result.startswith("http"):
+            display.display(display.Image(url=image_result))
+        else:
+            display.display(display.Image(image_result))
     except Exception as e:
         return f"Error generating image: {str(e)}"