Update app.py

app.py

@@ -9,7 +9,14 @@ from Gradio_UI import GradioUI
 # Tool to fetch weather data (No API Key Required, Uses wttr.in)
 @tool
 def get_weather(city: str) -> str:
-    """Fetches current weather for a given city. Uses wttr.in (No API key required)."""
+    """Fetches current weather for a given city.
+    
+    Args:
+        city (str): The name of the city to fetch weather for.
+
+    Returns:
+        str: The weather information including temperature, humidity, and wind speed.
+    """
     try:
         url = f"https://wttr.in/{city}?format=%C+%t+%h+%w"
         response = requests.get(url)
@@ -22,7 +29,14 @@ def get_weather(city: str) -> str:
 # Tool to fetch AQI (Air Quality Index) - No API Key Needed (Uses waqi.info with a demo token)
 @tool
 def get_air_quality(city: str) -> str:
-    """Fetches air quality index (AQI) for a city using waqi.info."""
+    """Fetches air quality index (AQI) for a city using waqi.info.
+    
+    Args:
+        city (str): The name of the city to fetch AQI for.
+
+    Returns:
+        str: The AQI value and air quality status.
+    """
     try:
         url = f"https://api.waqi.info/feed/{city}/?token=demo"  # Demo token (no key needed)
         response = requests.get(url).json()
@@ -35,7 +49,14 @@ def get_air_quality(city: str) -> str:
 # Tool to get local time
 @tool
 def get_local_time(city: str) -> str:
-    """Fetches the current local time for a city."""
+    """Fetches the current local time for a city.
+    
+    Args:
+        city (str): The name of the city to fetch the local time for.
+
+    Returns:
+        str: The current local time in YYYY-MM-DD HH:MM:SS format.
+    """
     try:
         tz = pytz.timezone(city.replace(" ", "_"))  # Handle spaces in city names
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -46,7 +67,14 @@ def get_local_time(city: str) -> str:
 # Tool to fetch top 5 restaurants using DuckDuckGo Search
 @tool
 def get_top_restaurants(city: str) -> str:
-    """Finds top 5 restaurants in a city using DuckDuckGo search."""
+    """Finds top 5 restaurants in a city using DuckDuckGo search.
+    
+    Args:
+        city (str): The name of the city to find restaurants in.
+
+    Returns:
+        str: A list of the top 5 restaurants with their names and links.
+    """
     try:
         search_tool = DuckDuckGoSearchTool()
         results = search_tool.run(f"Top 5 restaurants in {city}")
@@ -87,3 +115,4 @@ agent = CodeAgent(
 
 # Launch Gradio UI
 GradioUI(agent).launch()
+