Update app.py

app.py

@@ -9,7 +9,15 @@ from Gradio_UI import GradioUI
 # Tool: Fetch current time in a given timezone
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
-    """Fetches the current local time in a specified timezone."""
+    """
+    Fetches the current local time in a specified timezone.
+
+    Args:
+        timezone (str): A string representing a valid timezone (e.g., 'America/New_York').
+    
+    Returns:
+        str: The current time in the given timezone.
+    """
     try:
         tz = pytz.timezone(timezone)
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
@@ -20,7 +28,15 @@ def get_current_time_in_timezone(timezone: str) -> str:
 # Tool: Fetch real-time weather for a given location
 @tool
 def get_weather(city: str) -> str:
-    """Fetches the current weather for a given city."""
+    """
+    Fetches the current weather for a given city.
+
+    Args:
+        city (str): Name of the city.
+    
+    Returns:
+        str: Weather description and temperature.
+    """
     api_url = f"https://wttr.in/{city}?format=%C+%t"
     response = requests.get(api_url)
     if response.status_code == 200:
@@ -30,7 +46,16 @@ def get_weather(city: str) -> str:
 # Tool: Fetch currency exchange rate
 @tool
 def get_exchange_rate(base_currency: str, target_currency: str) -> str:
-    """Fetches the exchange rate between two currencies."""
+    """
+    Fetches the exchange rate between two currencies.
+
+    Args:
+        base_currency (str): The currency to convert from.
+        target_currency (str): The currency to convert to.
+    
+    Returns:
+        str: Exchange rate information.
+    """
     url = f"https://api.exchangerate-api.com/v4/latest/{base_currency}"
     response = requests.get(url)
     if response.status_code == 200:
@@ -42,7 +67,15 @@ def get_exchange_rate(base_currency: str, target_currency: str) -> str:
 # Tool: Fetch Wikipedia summary
 @tool
 def get_wikipedia_summary(query: str) -> str:
-    """Fetches a short summary from Wikipedia."""
+    """
+    Fetches a short summary from Wikipedia.
+
+    Args:
+        query (str): The topic to search on Wikipedia.
+    
+    Returns:
+        str: Summary of the topic.
+    """
     url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{query}"
     response = requests.get(url)
     if response.status_code == 200:
@@ -52,7 +85,12 @@ def get_wikipedia_summary(query: str) -> str:
 # Tool: Fetch latest news headlines
 @tool
 def get_latest_news() -> str:
-    """Fetches the latest news headlines."""
+    """
+    Fetches the latest news headlines.
+    
+    Returns:
+        str: Top 5 latest news headlines.
+    """
     url = "https://newsapi.org/v2/top-headlines?country=us&apiKey=YOUR_NEWSAPI_KEY"
     response = requests.get(url)
     if response.status_code == 200:
@@ -63,7 +101,15 @@ def get_latest_news() -> str:
 # Tool: Fetch stock price for a given company symbol
 @tool
 def get_stock_price(symbol: str) -> str:
-    """Fetches the latest stock price for a given company symbol."""
+    """
+    Fetches the latest stock price for a given company symbol.
+
+    Args:
+        symbol (str): The stock ticker symbol.
+    
+    Returns:
+        str: The latest stock price.
+    """
     url = f"https://finnhub.io/api/v1/quote?symbol={symbol}&token=YOUR_FINNHUB_API_KEY"
     response = requests.get(url)
     if response.status_code == 200:
@@ -110,4 +156,4 @@ agent = CodeAgent(
 )
 
 # Launch Gradio UI for interaction
-GradioUI(agent).launch()
+GradioUI(agent).launch()