Update app.py

app.py

@@ -10,18 +10,28 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-from smolagents import CodeAgent, DuckDuckGoSearchTool, WikipediaTool, load_tool, TransformersModel
+from smolagents import (
+    CodeAgent,
+    FinalAnswerTool,
+    DuckDuckGoSearchTool,
+    GoogleSearchTool,
+    PythonInterpreterTool,
+    load_tool,
+    TransformersModel,
+    ToolCollection
+)
 
 # 1. Load a powerful model (32B+)
 model = TransformersModel(
-    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",  # Swap for Mixtral 8x22B, DBRX, etc.
+    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",  # Or Mixtral 8x22B, DBRX, etc.
     device_map="auto",
-    max_new_tokens=2048,
     trust_remote_code=True
 )
 
-search_tool = DuckDuckGoSearchTool()  # Web search
-wikipedia_tool = WikipediaTool()      # Wikipedia search
+final_answer_tool = FinalAnswerTool()
+search_tool = DuckDuckGoSearchTool()
+google_tool = GoogleSearchTool()
+python_tool = PythonInterpreterTool()
 
 # Calculator (from Hugging Face Hub)
 calculator_tool = load_tool("smol-ai/calculator", trust_remote_code=True)
@@ -34,7 +44,7 @@ unit_converter_tool = load_tool("smol-ai/unit-converter", trust_remote_code=True
 
 # Date/time tool (from Hugging Face Hub)
 datetime_tool = load_tool("smol-ai/datetime", trust_remote_code=True)
-
+image_tool_collection = ToolCollection.from_hub("huggingface-tools/diffusion-tools-6630bb19a942c2306a2cdb6f")
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -56,18 +66,22 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = CodeAgent(
-        tools=[
+        all_tools = [
+            final_answer_tool,
             search_tool,
-            wikipedia_tool,
+            google_tool,
+            python_tool,
             calculator_tool,
             summarizer_tool,
             unit_converter_tool,
-            datetime_tool
-        ],
-        model=model,
-        add_base_tools=True  # Adds Python REPL, file I/O, etc.
-    )
+            datetime_tool,
+            *image_tool_collection.tools  # Add all image tools from the collection
+        ]
+        agent = CodeAgent(
+            tools=all_tools,
+            model=model,
+            add_base_tools=False  # Already included above
+        )
 
     except Exception as e:
         print(f"Error instantiating agent: {e}")