Update app.py

app.py

@@ -9,27 +9,31 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-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
-    """This tool is used to obtain information regarding FIFA World Cup winners for the male and female soccer competition
+def scrape_wiki_fifa_winners(arg1: int, arg2: str) -> str:
+    """This tool is used to obtain information regarding FIFA World Cup winners 
+    for the male and female soccer competition.
+
     Args:
-        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
-    """
+        arg1 (int): The year of the World Cup.
+        arg2 (str): The competition category, either "Men" or "Women".
 
+    Returns:
+        str: The winner of the given category and year or an error message.
+    """
+    
     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:
@@ -38,19 +42,20 @@ def scrape_wiki_fifa_winners(arg1:int, arg2:str)-> str: #it's import to specify
                     year = cols[0].text.strip()
                     winner = cols[1].text.strip()
                     winners.setdefault(categories[idx], {})[year] = winner
-        
+
         return winners
+
+    if arg2 not in ["Men", "Women"]:
+        return "Error: Invalid category. Choose either 'Men' or 'Women'."
+
+    fifa_winners = get_fifa_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} competition for the year {arg1} is {years[arg1]}"
+    year_str = str(arg1)  # Convert year to string since dictionary keys are strings
+    if year_str not in fifa_winners[arg2]:
+        return f"Error: No data available for the year {arg1} in the {arg2} category."
+
+    return f"The winner of the {arg2} World Cup in {arg1} was {fifa_winners[arg2][year_str]}."
+
             
 
 @tool