Update app.py

app.py

@@ -35,6 +35,44 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+
+@tool 
+def get_last_match(team: str) -> str:
+    """
+    Get the last 5 matches of a team introduced.
+
+    Args:
+        team: string representing a football team name
+    """
+    search_query = f"{team} last 5 matches"
+    search_results = DuckDuckGoSearchTool().run(search_query)
+    
+    matches = []
+    # Iterate through search results to find matches
+    for result in search_results:
+        snippet = result.get('snippet', '').lower()
+        # Check if snippet contains match-related keywords
+        if ' vs ' in snippet or 'match' in snippet or 'result' in snippet:
+            # Split snippet into lines to parse each match
+            lines = result['snippet'].split('\n')
+            for line in lines:
+                # Look for lines indicating a match (e.g., containing 'vs' or dates)
+                if ' vs ' in line or (' vs. ' in line) or (' - ' in line and ('win' in line or 'draw' in line or 'loss' in line)):
+                    match_info = line.strip()
+                    matches.append(match_info)
+                    if len(matches) >= 5:
+                        break
+            if len(matches) >= 5:
+                break
+    
+    if not matches:
+        return f"No recent matches found for the team."
+    
+    # Format the last 5 matches
+    formatted_matches = "\n".join(matches[:5])
+    return formatted_matches
+
+
 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:
@@ -64,7 +102,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_last_match, image_generation_tool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,