Update app.py

app.py

@@ -8,6 +8,9 @@ from tools.visit_webpage import VisitWebpageTool
 import os
 import IPython.display as display
 
+import io
+import base64
+from PIL import Image  # Only needed if 'image' is a PIL Image object
 
 from Gradio_UI import GradioUI
 
@@ -28,7 +31,15 @@ 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
+        # return image_result
+        # Save the image to a byte stream
+        image_byte_arr = io.BytesIO()
+        image.save(image_byte_arr, format='JPEG')
+        image_byte_arr = image_byte_arr.getvalue()
+        base64_image = base64.b64encode(image_byte_arr).decode('utf-8')
+        return f'<img src="data:image/jpeg;base64,{base64_image}"/>'
+
+        
     except Exception as e:
         return f"Error generating image: {str(e)}"