Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,get_weather,load_tool,tool
 import datetime
 import requests
 import pytz
@@ -10,8 +10,16 @@ from Gradio_UI import GradioUI
 
 @tool
 def get_weather(city: str) -> str:
-    # Your weather logic here
-    return f"Demo response for {city}"
+    """
+    Retrieves the current weather for a given city.
+
+    Args:
+        city (str): Name of the city to get weather for.
+
+    Returns:
+        str: Weather description for the city.
+    """
+    return f"Demo response for {city}
 
 demo = gr.Interface(fn=get_weather, inputs="text", outputs="text")
 demo.launch()
@@ -51,6 +59,17 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
+
+    # registry.py
+from tools import get_weather, get_current_time_in_timezone
+from tools.final_answer import FinalAnswerTool
+from smolagents import load_tool
+
+tool_registry = {
+    "get_weather": {
+        "tool": get_weather,
+        "schema": get_weather.tool_schema(),
+        "enabled": True
     
 agent = CodeAgent(
     model=model,
@@ -63,6 +82,10 @@ agent = CodeAgent(
     description=None,
     prompt_templates=prompt_templates
 )
+     # app.py
+from agent_setup import agent
+from Gradio_UI import GradioUI
 
-
-GradioUI(agent).launch()
+if __name__ == "__main__":
+    print("Launching DTstudios demo agent...")
+    GradioUI(agent).launch()