Update app.py

app.py

@@ -89,28 +89,30 @@ original_image_generation_tool = load_tool("agents-course/text-to-image", trust_
 @tool
 def image_generation_tool(prompt: str) -> str:
     """Generates an image from a prompt and embeds it as base64 in markdown.
+    
     Args:
         prompt: The text prompt to generate an image from.
     """
     try:
-        # Call the original tool, which returns a PIL Image
+        # Run the tool and get back a PIL Image object
         image = original_image_generation_tool(prompt)
         
-        # Save image to disk
-        filename = f"generated_image_{hash(prompt)}.png"
+        # Save the image to a file
+        filename = f"generated_image_{abs(hash(prompt))}.png"
         image.save(filename)
 
-        # Convert to base64 markdown
+        # Convert image to base64 for embedding
         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})"
-    
+        return f"Image generated for 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()
 
 with open("prompts.yaml", 'r') as stream: