Update app.py

app.py

@@ -38,18 +38,14 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 @tool
-def convert_image_to_base64(image: PILImage) -> dict:
+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 {
-        "type": "image",
-        "format": "base64",
-        "data": f"data:image/png;base64,{encoded}"
-    }
+    return f'<img src="data:image/png;base64,{encoded}" alt="Generated Image" style="max-width:100%; height:auto;" />'
 
 
 final_answer = FinalAnswerTool()