First agent

.gitignore

@@ -0,0 +1,2 @@
+.env
+.vscode

app.py

@@ -3,9 +3,13 @@ import datetime
 import requests
 import pytz
 import yaml
+import os
+from Gradio_UI import GradioUI
 from tools.final_answer import FinalAnswerTool
+from tools.web_search import DuckDuckGoSearchTool
+from dotenv import load_env
 
-from Gradio_UI import GradioUI
+load_env()
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -33,8 +37,47 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_laliga_team_ranking(team_name: str) -> int:
+    """Fetches the current ranking of a LaLiga team.
+    
+    Args:
+        team_name: The name of the LaLiga team (e.g., 'Real Madrid CF').
+    
+    Returns:
+        The team's ranking as an integer.
+    """
+    try:
+        url = "https://api.football-data.org/v4/competitions/PD/standings"
+        headers = {"X-Auth-Token": os.getenv("FOOTBALL_DATA_API")}  # Replace with a valid API key
+        response = requests.get(url, headers=headers)
+        data = response.json()
+
+        for table in data['standings']:
+            for team in table['table']:
+                if team['team']['name'].lower() == team_name.lower():
+                    return team['position']
+        
+        return -1  # Return -1 if the team is not found
+    except Exception as e:
+        return -1  # Return -1 if there is an error
+
+
+@tool
+def generate_image(prompt: str) -> str:
+    """Generates an image based on a given text prompt.
+    
+    Args:
+        prompt: A text description of the image to generate.
+    
+    Returns:
+        A URL to the generated image.
+    """
+    return image_generation_tool.run(prompt)
+
 
 final_answer = FinalAnswerTool()
+web_search = DuckDuckGoSearchTool()
 
 # 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' 
@@ -55,7 +98,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, web_search, get_laliga_team_ranking, generate_image], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,