Update app.py

app.py

@@ -85,6 +85,29 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+@tool
+def generate_image(prompt: str) -> str:
+    """
+    A tool that generates an image from a text prompt using Replicate's Stable Diffusion API.
+    
+    Args:
+        prompt: A detailed text description of the image to generate.
+
+    Returns:
+        A URL of the generated image.
+    """
+    try:
+        api_url = "https://replicate.com/api/models/stability-ai/stable-diffusion"
+        response = requests.post(api_url, json={"input": {"prompt": prompt}})
+
+        if response.status_code == 200:
+            return response.json().get("output", ["Image generation failed."])[0]
+        else:
+            return f"Error: {response.status_code} - {response.text}"
+
+    except Exception as e:
+        return f"Failed to generate image: {str(e)}"
+
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -105,7 +128,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,enhanced_translate,image_generation_tool,search_tool ], ## add your tools here (don't remove final answer)
+    tools=[final_answer,enhanced_translate,generate_image,search_tool ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,