switch to deepseek with managed agent

app.py

@@ -5,45 +5,100 @@ import inspect
 import pandas as pd
 import yaml
 
-from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, DuckDuckGoSearchTool, Tool
-from tools import visit_webpage
+from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, DuckDuckGoSearchTool, Tool, VisitWebpageTool
+from tools import visit_webpage, transcribe_audio, analyze_video
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
+# --- Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
+class coder_agent:
+    """Coder agent that is running the Qwen2.5 coder model and can generate and execute python code. It can import the pandas library for """
     def __init__(self):
-        websearchtool = WebSearchTool()
+        model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+        
+        web_agent = CodeAgent(
+            model=model,
+            tools=[
+                #visit_webpage,
+                VisitWebpageTool(),
+                WebSearchTool()
+            ],
+            name="web_agent",
+            description="Browses the web to find information",
+            verbosity_level=0,
+            max_steps=10,
+            additional_authorized_imports=[
+                "geopandas",
+                "plotly",
+                "shapely",
+                "json",
+                "pandas",
+                "numpy",
+            ],
+        )
+
+    def __call__(self, prompt: str) -> str:
+
+        
+#class vision_agent:
+#    """This vision agent is able to be passed base64-encoded image data and will use the OpenAI gpt-4o model to decode it and identify objects within. It will return a textual description of the image."""
+
+    
+class MasterAgent:
+    def __init__(self):
+        #websearchtool = WebSearchTool()
         #search_tool = DuckDuckGoSearchTool()
-        model = InferenceClientModel()
-        # TODO: add additional tools and or subagents (e.g. OpenAI image recognition)
+        model=InferenceClientModel("deepseek-ai/DeepSeek-R1", max_tokens=8096),
 
-        answer_facts = ""
+        try:
+            coder_agent = coder_agent()
+        except Exception as e:
+            print(f"Error instantiating agent: {e}")
+            return f"Error initializing agent: {e}", None
+
+        try:
+            vision_agent = vision_agent()
+        except Exception as e:
+            print(f"Error instantiating agent: {e}")
+            return f"Error initializing agent: {e}", None
+            
         with open("prompts.yaml", 'r') as stream:
             prompt_templates = yaml.safe_load(stream)
     
         self.master_agent = CodeAgent(
-            tools=[websearchtool, visit_webpage],
+            tools=[VisitWebpageTool(),
+                WebSearchTool()],
             model=model,
             add_base_tools=True,
-            planning_interval=3,
-            max_steps=10,
+            planning_interval=5,
+            max_steps=15,
             prompt_templates=prompt_templates,
+            managed_agents=[coder_agent],
+            additional_authorized_imports=[
+                "geopandas",
+                "plotly",
+                "shapely",
+                "json",
+                "pandas",
+                "numpy",
+            ],
         )
-        print("BasicAgent initialized.")
+        print("MasterAgent initialized.")
     
-    def __call__(self, question: str) -> str:
+    def __call__(self, question: str, attached_file: str) -> str:
+        """"""
         print(f"Agent received question (first 50 chars): {question[:50]}...")
+        #TODO Handle the file, some will need to be accessed locally, some can just be passed as public URLs
         agent_answer = self.master_agent.run(question)
         print(f"Agent answer: {agent_answer}")
         return agent_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
+    Fetches all questions, runs the MasterAgent on them, submits all answers,
     and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
@@ -63,7 +118,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = MasterAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -96,9 +151,42 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
+
+    cache_dir = "file_cache"
+    if not os.path.exists(cache_dir):
+        os.makedirs(cache_dir)
+    
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+        attached_file = item.get("file_name")
+        if attached_file != "":
+            local_file_path = os.path.join(cache_dir, attached_file)
+            if not os.path.exists(local_file_path):
+                file_name_no_ext = os.path.splitext(attached_file)[0]  # e.g., 'document' from 'document.pdf'
+                download_url = f"{questions_url}/files/{file_name_no_ext}"
+
+                try:
+                    print(f"Downloading from {download_url}")
+                    response = requests.get(download_url, stream=True)
+                    response.raise_for_status()  # Raises an HTTPError for bad responses
+                    with open(local_file_path, 'wb') as f:
+                        for chunk in response.iter_content(chunk_size=8192):
+                            if chunk:
+                                f.write(chunk)
+                    print(f"File downloaded and cached: {local_file_path}")
+                    return local_file_path
+                
+                except requests.exceptions.HTTPError as e:
+                    print(f"HTTP error downloading {download_url}: {e}")
+                    return None
+                except requests.exceptions.RequestException as e:
+                    print(f"Error downloading {download_url}: {e}")
+                    return None
+                except OSError as e:
+                    print(f"Error saving file to {local_file_path}: {e}")
+                    return None
+        
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
@@ -165,7 +253,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
-    gr.Markdown("# Basic Agent Evaluation Runner")
+    gr.Markdown("# Agent Evaluation Runner")
     gr.Markdown(
         """
         **Instructions:**
@@ -215,5 +303,5 @@ if __name__ == "__main__":
 
     print("-"*(60 + len(" App Starting ")) + "\n")
 
-    print("Launching Gradio Interface for Basic Agent Evaluation...")
+    print("Launching Gradio Interface for Agent Evaluation...")
     demo.launch(debug=True, share=False)