Update app.py

app.py

@@ -7,31 +7,48 @@ 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 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
-    """A tool that does nothing yet 
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    return "What magic will you build ?"
-
-@tool
-def get_current_time_in_timezone(timezone: str) -> str:
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
+import requests
+
+class FinalAnswerTool:
+    def __call__(self, result: str):
+        return f"\n Final Answer:\n{result}"
+
+class ToolCallingAgent:
+    def __init__(self, tools, final_tool, model, max_steps=10):
+        self.tools = tools
+        self.final_tool = final_tool
+        self.model = model
+        self.max_steps = max_steps
+
+    def run(self, input_text):
+        print(f"\n Agent received: {input_text}")
+        for tool in self.tools:
+            try:
+                result = tool(input_text)
+                if result:
+                    final_output = self.final_tool(str(result))
+                    print(final_output)
+                    return final_output  # Stop after first useful tool response
+            except Exception as e:
+                print(f"\n[{tool.__class__.__name__} Error]: {str(e)}")
+
+# Initialize all tools
+weather_tool = WeatherTool(api_key="YOUR_API_KEY_HERE")
+visit_webpage = VisitWebpageTool()
+duckduckgo_tool = CustomDuckDuckGoSearchTool()
+final_answer = FinalAnswerTool()
+model = DummyModel()
+
+# Initialize the agent with the final answer tool
+web_agent = ToolCallingAgent(
+    tools=[duckduckgo_tool, visit_webpage, weather_tool],
+    final_tool=final_answer,
+    model=model,
+    max_steps=10
+)
+
+# Run with a query
+web_agent.run("Tokyo")  # Works for weather, search, or URL
 
 
 final_answer = FinalAnswerTool()