Update app.py

app.py

@@ -92,8 +92,24 @@ def image_generation_tool(prompt: str) -> str:
     Args:
         prompt: The text prompt to generate an image from.
     """
-    path = original_image_generation_tool(prompt)
-    return image_path_to_markdown(path)
+    try:
+        # Call the original tool, which returns a PIL Image
+        image = original_image_generation_tool(prompt)
+        
+        # Save image to disk
+        filename = f"generated_image_{hash(prompt)}.png"
+        image.save(filename)
+
+        # Convert to base64 markdown
+        with open(filename, "rb") as f:
+            img_bytes = f.read()
+        img_base64 = base64.b64encode(img_bytes).decode("utf-8")
+
+        return f"Image generated from prompt: '{prompt}'\nSaved as: {filename}\n\n![Generated Image](data:image/png;base64,{img_base64})"
+    
+    except Exception as e:
+        return f"Error generating image: {str(e)}"
+
 
 search_tool = DuckDuckGoSearchTool()