Update app.py

Try DeepSeek solution

app.py

@@ -4,6 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
+from io import BytesIO
+import base64
 from PIL import Image
 
 from Gradio_UI import GradioUI
@@ -51,6 +53,23 @@ 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 generate_image(description: str) -> Image:
+#     """A tool that generates an image from a text description.
+#     Args:
+#         description: A string representing the text description to generate an image of.
+#     """
+#     image_generation_tool = Tool.from_space(
+#     "black-forest-labs/FLUX.1-schnell",
+#     name="image_generator",
+#     description="Generate an image from a prompt"
+# )
+#     # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
+#     image: Image = image_generation_tool(description)
+#     # return f"Image generated for prompt: {description}."
+#     return image
+
+
 @tool
 def generate_image(description: str) -> Image:
     """A tool that generates an image from a text description.
@@ -62,10 +81,19 @@ def generate_image(description: str) -> Image:
     name="image_generator",
     description="Generate an image from a prompt"
 )
-    # image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
-    image: Image = image_generation_tool(description)
-    # return f"Image generated for prompt: {description}."
-    return image
+    # Generate the image
+    pil_image: Image = image_generation_tool(description)
+    
+    # Convert image to bytes
+    buffered = BytesIO()
+    pil_image.save(buffered, format="PNG")
+    buffered.seek(0)
+    
+    # Encode as base64
+    img_base64 = base64.b64encode(buffered.read()).decode()
+    
+    # Return as Markdown image
+    return f"![Generated Image](data:image/png;base64,{img_base64})"