Update app.py

app.py

@@ -6,12 +6,16 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, t
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
-# 1. Tool: Get Current Time in Any Timezone
+# ---------------------------- TOOLS ----------------------------
+
+# 1. Get Current Time in Any Timezone
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """Fetches the current local time in a specified timezone.
     Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
+        timezone: A valid timezone string (e.g., 'America/New_York').
+    Returns:
+        A string with the current local time in the requested timezone.
     """
     try:
         tz = pytz.timezone(timezone)
@@ -20,14 +24,16 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# 2. Tool: Get Current Weather Data (Requires OpenWeatherMap API)
+# 2. Get Current Weather Data
 @tool
 def get_weather(city: str) -> str:
     """Fetches current weather information for a specified city.
     Args:
         city: The name of the city (e.g., 'New York').
+    Returns:
+        A string with the weather description and temperature.
     """
-    api_key = "ac662b5c2463cfa522e1acd2d4cf5a81" # Replace with your OpenWeather API key
+    api_key = "YOUR_OPENWEATHER_API_KEY"  # Replace with your actual API key
     url = f"http://api.openweathermap.org/data/2.5/weather?q={city}&appid={api_key}&units=metric"
     
     try:
@@ -41,32 +47,36 @@ def get_weather(city: str) -> str:
     except Exception as e:
         return f"Error fetching weather data: {str(e)}"
 
-# 3. Tool: Currency Exchange Converter
+# 3. Currency Exchange Converter
 @tool
 def convert_currency(amount: float, from_currency: str, to_currency: str) -> str:
     """Converts currency from one type to another using real-time exchange rates.
     Args:
         amount: The amount to convert.
-        from_currency: The source currency code (e.g., 'USD').
-        to_currency: The target currency code (e.g., 'INR').
+        from_currency: Source currency code (e.g., 'USD').
+        to_currency: Target currency code (e.g., 'INR').
+    Returns:
+        A string with the converted currency amount.
     """
-    api_key = "90171124ae953d74d1515263"  # Replace with a valid API key from ExchangeRate API
+    api_key = "YOUR_EXCHANGE_RATE_API_KEY"  # Replace with your actual API key
     url = f"https://v6.exchangerate-api.com/v6/{api_key}/latest/{from_currency}"
     
     try:
         response = requests.get(url).json()
         if "conversion_rates" not in response:
-            return f"Error: Unable to fetch exchange rates."
+            return "Error: Unable to fetch exchange rates."
         
-        rate = response["conversion_rates"].get(to_currency, None)
+        rate = response["conversion_rates"].get(to_currency)
         if rate is None:
-            return f"Error: Unsupported currency code."
+            return "Error: Unsupported currency code."
         
         converted_amount = round(amount * rate, 2)
         return f"{amount} {from_currency} is equal to {converted_amount} {to_currency}."
     except Exception as e:
         return f"Error fetching exchange rates: {str(e)}"
-    
+
+# ---------------------------- MODEL & AGENT SETUP ----------------------------
+
 # Load Hugging Face Text-to-Image Tool
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
@@ -78,14 +88,13 @@ model = HfApiModel(
     max_tokens=2096,
     temperature=0.7,
     model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
-    custom_role_conversions=None,
 )
 
 # Load Prompt Templates
 with open("prompts.yaml", "r") as stream:
     prompt_templates = yaml.safe_load(stream)
 
-# Create Agent with Additional Tools
+# Create AI Agent with Defined Tools
 agent = CodeAgent(
     model=model,
     tools=[
@@ -98,12 +107,10 @@ agent = CodeAgent(
     ],
     max_steps=6,
     verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
     prompt_templates=prompt_templates,
 )
 
+# ---------------------------- LAUNCH APPLICATION ----------------------------
+
 # Launch Gradio UI
-GradioUI(agent).launch()
+GradioUI(agent).launch()