Update app.py

app.py

@@ -9,14 +9,49 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def abe_tool(arg1:int, arg2:int)-> int: #it's import to specify the return type
+def scrape_wiki_fifa_winners(arg1:int, arg2:str)-> 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 accepts two numbers and outputs the sum of them
+    """This tool is used to obtain information regarding FIFA World Cup winners for the male and female soccer competition
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        arg1: this argument accepts an integer denoting the year
+        arg2: this argument accepts a string denoting if the competition to be checked is either for the Male or Female Soccer competition
     """
-    return arg1 +  arg2
+
+    import requests
+    from bs4 import BeautifulSoup
+    
+    def get_fifa_winners():
+        url = "https://en.wikipedia.org/wiki/List_of_FIFA_World_Cup_finals"
+        response = requests.get(url)
+        soup = BeautifulSoup(response.text, 'html.parser')
+        
+        winners = {}
+        tables = soup.find_all("table", {"class": "wikitable"})
+        
+        categories = ["Men", "Women"]
+        
+        for idx, table in enumerate(tables[:2]):  # First two tables (Men's and Women's WC)
+            rows = table.find_all("tr")[1:]  # Skip header row
+            for row in rows:
+                cols = row.find_all("td")
+                if len(cols) >= 2:
+                    year = cols[0].text.strip()
+                    winner = cols[1].text.strip()
+                    winners.setdefault(categories[idx], {})[year] = winner
+        
+        return winners
+    
+    if __name__ == "__main__":
+        if arg2 not in ["Men", "Women"]:
+            return f" Invalid Category"
+        fifa_winners = get_fifa_winners()
+        years=fifa_winners[arg2]
+        if arg1 not in years:
+            return f" Invalid Year"
+            #print(f"\n{category} World Cup Winners:")
+        
+       return f" The winner of the {arg2}'s competition for the year {arg1} is {years[arg1]}"
+            
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +90,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,abe_tool, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer,scrape_wiki_fifa_winners, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,