Update app.py

app.py

@@ -1,4 +1,5 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+import os
 import datetime
 import requests
 import pytz
@@ -7,33 +8,89 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+# Below is an example of a tool that does nothing. Amaze us with your creativity !
+@tool
+def get_coin_price(symbol: str, currency: str = "usd")-> 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 fetches the latest price on a cryptocurrency symbol from Coingecko API.
+    Args:
+        symbol: A string representing a valid symbol as CoinGecko's ID, not ticker (e.g., 'bitcoin' not BTC).
+        currency: A string representing the currency the value is returned in (e.g., 'usd').
+    """
+    # set to lowercase for api compatibility and consistency
+    symbol = symbol.lower()
+    currency = currency.lower()
+    
+    #define the endpoint with return headers - https://docs.coingecko.com/v3.0.1/reference/simple-price
+    url = "https://api.coingecko.com/api/v3/simple/price?ids=" + symbol + "&vs_currencies=" + currency
+    headers = {"accept": "application/json"}
+    
+    try:
+        #make request to endpoint
+        response = requests.get(url, headers=headers)
+        
+        #process response
+        if response.status_code == 200:
+            data = response.json()
+            usd_value = data[symbol][currency]
+            return f"The price of " + symbol + " in " + currency + " is " + f"${usd_value:,.2f}" #format as currency
+        else:
+            # returrn verbose error, so the model can observe and ReAct
+            return f"Error: {response.status_code}, {response.text}"
+    except Exception as e:
+        return f"Error fetching price for '{symbol}' valued in currency '{currency}': {str(e)}"
+
+@tool
+def get_current_time_in_timezone(timezone: str) -> str:
+    """A tool that fetches the current local time in a specified timezone.
+    Args:
+        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)}"
+
 
 final_answer = FinalAnswerTool()
 
+# 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',
+model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
 
+# model = HfApiModel(
+#     model_id="deepseek-ai/DeepSeek-R1",
+#     provider="together",
+#     api_key=os.getenv("together_key")  # Fetch API key from environment
+# )
+
+
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Load system prompt from prompt.yaml file
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,DuckDuckGoSearchTool()], # add your tools here (don't remove final_answer)
+    tools=[final_answer, get_coin_price, get_current_time_in_timezone, DuckDuckGoSearchTool()], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates # Pass system prompt to CodeAgent
+    prompt_templates=prompt_templates
 )