Calculatot

app.py

@@ -7,26 +7,9 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-@tool
-def my_custom_tool(prompt: str) -> str:
-    """Generates an image using Hugging Face's Stable Diffusion API.
-    Args:
-        prompt: A text description of the image.
-    """
-    API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-2"
-    HEADERS = {"Authorization": "Bearer YOUR_HUGGINGFACE_API_KEY"}
-
-    payload = {"inputs": prompt}
-    response = requests.post(API_URL, headers=HEADERS, json=payload)
-
-    if response.status_code == 200:
-        image_path = "generated_image.png"
-        with open(image_path, "wb") as file:
-            file.write(response.content)
-        return f"Image successfully generated: {image_path}"
-    else:
-        return f"Error: {response.json()}"
-        
+def calculator(a: int, b: int) -> int:
+    """Multiply two integers."""
+    return a * b
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.