Update app.py

app.py

@@ -8,6 +8,29 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
+
+@tool
+def generate_huggingface_image(prompt: str, model: str = "stabilityai/stable-diffusion-2") -> str:
+    """A tool that generates images using Hugging Face's HfApiModel.
+    
+    Args:
+        prompt: The text description of the image to generate.
+        model: The model to use for image generation (default: stabilityai/stable-diffusion-2).
+    
+    Returns:
+        URL of the generated image.
+    """
+        
+    # Send the prompt to the Hugging Face model
+    response = client.text_to_image(model=model, prompt=prompt)
+    
+    # Save the response image
+    image_path = "generated_image.png"
+    with open(image_path, "wb") as f:
+        f.write(response)
+    
+    return image_path
+
 @tool
 def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool