Update app.py

app.py

@@ -1,5 +1,13 @@
 import os
 import gradio as gr
+from PIL import Image
+from smolagents import (
+    CodeAgent, 
+    HfApiModel,       
+    DuckDuckGoSearchTool, 
+    visit_webpage,
+    Tool
+)
 import requests
 import inspect
 import pandas as pd
@@ -40,7 +48,35 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        
+        # 2. Instantiate Internet Search & Browsing Tools
+        ddg_search = DuckDuckGoSearchTool()
+        
+        # 3. Create an Image Execution Tool from the Hugging Face Spaces Engine
+        image_generation_tool = Tool.from_space(
+            space_id="black-forest-labs/FLUX.1-schnell", 
+            name="image_generator", 
+            description="Generates a high-quality visual image based on a descriptive text prompt. Returns a PIL Image object."
+        )
+        
+        # 4. Consolidate your toolbox array
+        all_tools = [
+            ddg_search, 
+            visit_webpage, 
+            image_generation_tool
+        ]
+        
+        # 5. Initialize HfApiModel with the top-tier open-source GAIA benchmark model
+        # Qwen2.5-72B-Instruct provides elite multi-step execution logic over Serverless API infrastructure.
+        model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
+        
+        # 6. Initialize the CodeAgent with the Serverless API backbone
+        agent = CodeAgent(
+            tools=all_tools,
+            model=model,
+            add_base_tools=True,
+            additional_authorized_imports=["time", "math", "json", "PIL"]
+        )
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None