Update app.py

app.py

@@ -18,6 +18,44 @@ 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_image(description: str) -> str:
+    """Generates an image based on the provided description using Hugging Face's image generation tool."""
+    try:
+        API_KEY = os.getenv("HUGGINGFACE_API_KEY")
+        if not API_KEY:
+            return "Error: HuggingFace API key is not set."
+        
+        headers = {
+            "Authorization": f"Bearer {API_KEY}",
+            "Content-Type": "application/json",
+        }
+        
+        payload = {
+            "model": "stability-ai/sdxl",
+            "inputs": description,
+        }
+        
+        response = requests.post(
+            "https://api-inference.huggingface.co/pipeline/image-generation",
+            headers=headers,
+            json=payload
+        )
+        
+        if response.status_code == 200:
+            image_data = response.json()
+            if isinstance(image_data, list) and len(image_data) > 0:
+                image_base64 = image_data[0].get("base64")
+                if image_base64:
+                    return f"Generated image (base64): {image_base64}"
+                else:
+                    return "Error: No image data found in the response."
+            else:
+                return "Error: Unexpected response format."
+        else:
+            return f"Error: API request failed with status code {response.status_code}. Details: {response.text}"
+    except Exception as e:
+        return f"Error generating image: {str(e)}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: