Update app.py

app.py

@@ -1,153 +1,61 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from Gradio_UI import GradioUI
 
-# Tool: Fetch current time in a given timezone
+from Gradio_UI import GradioUI
 
+# Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def get_current_time_in_timezone(timezone: str) -> str:
+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
+    """A tool that does nothing yet 
+    Args:
+        arg1: the first argument
+        arg2: the second argument
     """
-    Fetches the current local time in a specified timezone.
+    return "What magic will you build ?"
 
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that 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.
+        timezone: A string representing a valid timezone (e.g., 'America/New_York').
     """
     try:
+        # Create timezone object
         tz = pytz.timezone(timezone)
+        # Get current time in that timezone
         local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
         return f"The current local time in {timezone} is: {local_time}"
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-# Tool: Fetch real-time weather for a given location
-@tool
-def get_weather(city: str) -> str:
-    """
-    Fetches the current weather for a given city.
-
-    Args:
-        city (str): The name of the city for which to fetch the weather.
-    
-    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:
-        return response.text
-    return "Unable to retrieve weather information."
-
-
-# Tool: Fetch currency exchange rate
-@tool
-def get_exchange_rate(base_currency: str, target_currency: str) -> str:
-    """
-    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:
-        data = response.json()
-        rate = data['rates'].get(target_currency, "N/A")
-        return f"1 {base_currency} = {rate} {target_currency}"
-    return "Exchange rate information not available."
-
-# Tool: Fetch Wikipedia summary
-@tool
-def get_wikipedia_summary(query: str) -> str:
-    """
-    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:
-        return response.json().get("extract", "No summary available.")
-    return "Wikipedia summary not found."
-
-# Tool: Fetch latest news headlines
-@tool
-def get_latest_news() -> str:
-    """
-    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:
-        articles = response.json().get("articles", [])[:5]
-        return "\n".join([f"{i+1}. {article['title']}" for i, article in enumerate(articles)])
-    return "Unable to retrieve news."
-
-# 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.
-
-    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:
-        data = response.json()
-        return f"Stock price of {symbol}: ${data.get('c', 'N/A')}"
-    return "Stock information not available."
 
 final_answer = FinalAnswerTool()
 
-# Hugging Face model setup
+# 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' 
+
 model = HfApiModel(
-    max_tokens=2096,
-    temperature=0.5,
-    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-    custom_role_conversions=None,
+max_tokens=2096,
+temperature=0.5,
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+custom_role_conversions=None,
 )
 
-# Load image generation tool from Hugging Face
+
+# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
-# Creating the AI agent
 agent = CodeAgent(
     model=model,
-    tools=[
-        final_answer,
-        get_current_time_in_timezone,
-        get_weather,
-        get_exchange_rate,
-        get_wikipedia_summary,
-        get_latest_news,
-        get_stock_price,
-        image_generation_tool,
-    ],
+    tools=[final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -157,5 +65,5 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-# Launch Gradio UI for interaction
+
 GradioUI(agent).launch()