Update app.py

app.py

@@ -1,4 +1,5 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from PIL import Image
 import datetime
 import os
 import io
@@ -11,6 +12,16 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+def image_path_to_markdown(path: str) -> str:
+    try:
+        with open(path, "rb") as f:
+            img_bytes = f.read()
+        img_base64 = base64.b64encode(img_bytes).decode("utf-8")
+        return f"![Generated Image](data:image/png;base64,{img_base64})"
+    except Exception as e:
+        return f"Image saved at: {path} (Failed to embed image: {str(e)})"
+
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
 def my_custom_tool(text:str, size:int = 200)-> str: #it's import to specify the return type
@@ -73,7 +84,13 @@ custom_role_conversions=None,
 
 
 # Import tool from Hub
-image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+original_image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+
+@tool
+def image_generation_tool(prompt: str) -> str:
+    """Generates an image from a prompt and embeds it as base64 in markdown."""
+    path = original_image_generation_tool(prompt)
+    return image_path_to_markdown(path)
 
 search_tool = DuckDuckGoSearchTool()