Update app.py

app.py

@@ -15,7 +15,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class CoderAgent:
-    """Coder agent that is running the Qwen2.5 coder model and can generate and execute python code. It can import the pandas library for """
+    """Coder agent that is running the Qwen2.5 coder model and can generate and execute python code. It can import the pandas library for data analysis and manipulation."""
     def __init__(self):
         model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
         
@@ -40,6 +40,7 @@ class CoderAgent:
                 "numpy",
             ],
         )
+        print("ManagedAgent CoderAgent initialized.")
 
     def __call__(self, prompt: str) -> str:
         agent_answer = self.coder_agent.run(prompt)
@@ -96,11 +97,19 @@ class MasterAgent:
     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)
+
+        additional_args = {}
+        if attached_file:
+            additional_args = {
+                "attached_file" = attached_file
+            }
+            question = question + f" The file in attached_file can be accessed locally at this path 'file_cache/{attached_file}' or as a public URL at f{DEFAULT_API_URL}/files/f{os.path.splitext(attached_file)[0]}"
+            
+        agent_answer = self.master_agent.run(question, additional_args: additional_args)
         print(f"Agent answer: {agent_answer}")
         return agent_answer
 
+        
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the MasterAgent on them, submits all answers,
@@ -169,7 +178,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             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}"
+                download_url = f"{api_url}/files/{file_name_no_ext}"
 
                 try:
                     print(f"Downloading from {download_url}")
@@ -196,7 +205,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, attached_file)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: