Update app.py

app.py

@@ -4,6 +4,9 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from PIL.Image import Image as PILImage
+from io import BytesIO
+import base64
 
 from Gradio_UI import GradioUI
 
@@ -34,6 +37,16 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def convert_image_to_base64(image: PILImage) -> str:
+    """A tool that you MUST use exactly after the image_generation_tool since it encodes it's output to a printable image
+    Args:
+        image: An image that is the output of the image_generation_tool and is not in printable form
+    """
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    return f"data:image/png;base64,{base64.b64encode(buffered.getvalue()).decode()}"
+
 
 final_answer = FinalAnswerTool()