Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -8,20 +8,18 @@ import os
 from tavily import TavilyClient
 from Gradio_UI import GradioUI
 
-# Below is an example of a tool that does nothing. Amaze us with your creativity !
+# Example tool template - can be modified for your needs
+
 @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
+def my_custom_tool(arg1: str, arg2: int) -> str:  # Return type specification is important
     """A tool that does nothing yet 
     Args:
         arg1: the first argument
         arg2: the second argument
     """
-    return "What magic will you build ?"
-
+    return "What magic will you build?"
 
 
-# Tavily safe search tool 
 @tool
 def tavily_search(query: str) -> str:
     """An enhanced search tool that performs web search using Tavily API with better error handling and formatting.
@@ -35,8 +33,8 @@ def tavily_search(query: str) -> str:
         search_result = client.search(
             query=query, 
             search_depth="advanced",
-            include_domains=["*.gov", "*.edu", "*.org"],  # More relaible sources
-            exclude_domains=["pinterest.com", "reddit.com"]  # Social media excluded
+            include_domains=["*.gov", "*.edu", "*.org"],  # More reliable sources
+            exclude_domains=["pinterest.com", "reddit.com"]  # Exclude social media
         )
         
         summary = f"🔍 Search Results for: {query}\n\n"
@@ -54,10 +52,8 @@ def tavily_search(query: str) -> str:
         return f"Search Error: {str(e)}. Please try rephrasing your query or try again later."
 
 
-
-# Improved time tool
 @tool
-def world_time_tool(location: str) -> str:
+def get_current_time_in_timezone(location: str) -> str:
     """An improved timezone tool with better location handling and formatted output.
     Args:
         location: City name or timezone identifier (e.g., 'Tokyo' or 'Asia/Tokyo')
@@ -65,7 +61,7 @@ def world_time_tool(location: str) -> str:
         str: Formatted time information with additional context
     """
     try:
-        # Timezone mapping for common cities
+        # Common city to timezone mappings
         city_to_timezone = {
             'tokyo': 'Asia/Tokyo',
             'new york': 'America/New_York',
@@ -74,7 +70,7 @@ def world_time_tool(location: str) -> str:
             'istanbul': 'Europe/Istanbul'
         }
         
-        # Clean and check input
+        # Input validation and normalization
         clean_location = location.lower().strip()
         if clean_location in city_to_timezone:
             timezone = city_to_timezone[clean_location]
@@ -84,44 +80,41 @@ def world_time_tool(location: str) -> str:
         tz = pytz.timezone(timezone)
         current_time = datetime.datetime.now(tz)
         
-        # Zengin formatlama
+        # Format the output
         formatted_time = current_time.strftime("%Y-%m-%d %H:%M:%S")
         day_name = current_time.strftime("%A")
         
         return f"""
-            📍 Location: {location.title()}
-            🕒 Current Time: {formatted_time}
-            📅 Day: {day_name}
-        """
+                📍 Location: {location.title()}
+                🕒 Current Time: {formatted_time}
+                📅 Day: {day_name}
+                """
     except pytz.exceptions.UnknownTimeZoneError:
         return f"Error: '{location}' is not a recognized timezone or city. Please try a major city name or standard timezone format."
     except Exception as e:
         return f"Time lookup error: {str(e)}"
 
-
+# Initialize tools
 final_answer = FinalAnswerTool()
-
-# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-
-# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
-model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
+# If the agent does not answer, the model is overloaded, please use another model or 
+# the following Hugging Face Endpoint that also contains qwen2.5 coder:
 
 # Model configuration
+model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.3,  # Lower temperature for more consistent responses
-    model_id=model_id, # 'Qwen/Qwen2.5-Coder-32B-Instruct',
+    model_id=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)
-    
+
+# Initialize agent with tools
 agent = CodeAgent(
     model=model,
     tools=[final_answer, tavily_search, get_current_time_in_timezone, image_generation_tool], # Added tavily_search