Update app.py: feat: add trivia question tool and integrate with CodeAgent

### Added
- Implemented `get_trivia_question` tool to fetch random trivia questions from Open Trivia Database.
- Integrated the new tool into the `CodeAgent` for use in the system.

### Changed
- Removed placeholder tool `my_cutom_tool` and replaced it with `get_trivia_question`.
- Updated imports: added `random` for randomizing question type in `get_trivia_question`.

### Fixed
- N/A

### Notes
- Trivia questions now support predefined categories and require difficulty level.
- Question type (`multiple` or `boolean`) is randomly selected.

app.py

@@ -1,49 +1,96 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
 import yaml
+import random
+
 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 my_cutom_tool(arg1:str, arg2:int)-> 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 does nothing yet 
+def get_trivia_question(category: str, difficulty: str) -> str:
+    """Fetches a random trivia question from the Open Trivia Database.
+
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        category: The category of the question. Must be one of:
+            - 'General Knowledge'
+            - 'Books'
+            - 'Film'
+            - 'Music'
+            - 'Video Games'
+            - 'Computers'
+            - 'Mathematics'
+            - 'Mythology'
+            - 'Sports'
+        difficulty: The difficulty level of the question. Must be 'easy', 'medium', or 'hard'.
+
+    Returns:
+        A formatted trivia question with possible answers.
     """
-    return "What magic will you build ?"
+    category_map = {
+        "General Knowledge": 9,
+        "Books": 10,
+        "Film": 11,
+        "Music": 12,
+        "Video Games": 15,
+        "Computers": 18,
+        "Mathematics": 19,
+        "Mythology": 20,
+        "Sports": 21
+    }
+    if category not in category_map:
+        return f"Invalid category. Must be one of: {', '.join(category_map.keys())}"
+    if difficulty not in ["easy", "medium", "hard"]:
+        return "Invalid difficulty. Must be 'easy', 'medium', or 'hard'."
+    question_type = random.choice(["multiple", "boolean"])
+    params = {
+        "amount": 1,
+        "category": category_map[category],
+        "difficulty": difficulty,
+        "type": question_type
+    }
+    response = requests.get("https://opentdb.com/api.php", params=params)
+    if response.status_code != 200:
+        return "An error occurred while fetching a trivia question."
+    data = response.json()
+    if data["response_code"] != 0:
+        return "No question found for the given criteria."
+    question_data = data["results"][0]
+    question = question_data["question"]
+    correct_answer = question_data["correct_answer"]
+    incorrect_answers = question_data["incorrect_answers"]
+    if question_type == "multiple":
+        options = incorrect_answers + [correct_answer]
+        random.shuffle(options)
+        options_text = "\n".join([f"- {opt}" for opt in options])
+        return f"Trivia Question: {question}\n\nPossible answers:\n{options_text}\n\nWhat is your answer?"
+    else:
+        return f"Trivia Question: {question}\n\nAnswer: True or False?"
 
 @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()
 model = HfApiModel(
-max_tokens=2096,
-temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
-custom_role_conversions=None,
+    max_tokens=2096,
+    temperature=0.5,
+    model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+    custom_role_conversions=None,
 )
 
-
-# Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
 with open("prompts.yaml", 'r') as stream:
@@ -51,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_trivia_question, get_current_time_in_timezone],
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -61,5 +108,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-
-GradioUI(agent).launch()
+GradioUI(agent).launch()