Update app.py

app.py

@@ -1,62 +1,71 @@
 import os
 import gradio as gr
 import requests
-import inspect
+from smolagents import HfApiModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool
 import pandas as pd
-from smolagents import HfApiModel
-
+import tempfile
+from pathlib import Path
+import re
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Enhanced Agent Definition ---
-class GAIAAgent:
+# --- File Handling ---
+def download_file_if_any(base_api_url: str, task_id: str) -> str | None:
+    url = f"{base_api_url}/files/{task_id}"
+    try:
+        resp = requests.get(url, timeout=30)
+        if resp.status_code == 404:
+            return None
+        resp.raise_for_status()
+    except requests.exceptions.HTTPError as e:
+        raise e
+    cdisp = resp.headers.get("content-disposition", "")
+    filename = task_id
+    if "filename=" in cdisp:
+        m = re.search(r'filename="([^\"]+)"', cdisp)
+        if m:
+            filename = m.group(1)
+    tmp_dir = Path(tempfile.gettempdir()) / "gaia_files"
+    tmp_dir.mkdir(exist_ok=True)
+    file_path = tmp_dir / filename
+    with open(file_path, "wb") as f:
+        f.write(resp.content)
+    return str(file_path)
+
+# --- Basic Agent Definition ---
+class BasicAgent:
     def __init__(self):
-        print("GAIAAgent with HfApiModel initialized.")
-        self.model = gr.load(
-           "models/deepseek-ai/DeepSeek-R1",
-           provider="novita",
+        model = HfApiModel(
+            model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
+            max_tokens=2096,
+            temperature=0.5,
+            custom_role_conversions=None,
         )
-
-    def format_prompt(self, question: str, file_content: str = None) -> str:
-        prompt = (
-            "You are a helpful AI agent solving a question from the GAIA benchmark. "
-            "Respond only with the final answer."
+        self.agent = CodeAgent(
+            model=model,
+            tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
+            add_base_tools=True,
+            additional_authorized_imports=[]
         )
-        if file_content:
-            prompt += f"\nAttached File Content:\n{file_content}\n"
-        prompt += f"\nQuestion: {question}\nAnswer:"
-        return prompt
-
-    def read_file(self, filename: str) -> str:
-        filepath = os.path.join("./", filename)
-        if filename.endswith(".txt") and os.path.exists(filepath):
-            with open(filepath, "r") as file:
-                return file.read()[:1000]  # limit to 1000 chars
-        return ""
-
-    def __call__(self, question: str, file_name: str = None) -> str:
-        file_content = self.read_file(file_name) if file_name else None
-        prompt = self.format_prompt(question, file_content)
+        print("BasicAgent initialized.")
+
+    def __call__(self, question: str) -> str:
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
         try:
-            print("Prompt sent to model:", prompt)
-            result = self.model(prompt)
-            print("Model raw result:", result)
-            if not result or not isinstance(result, str):
-                return "AGENT ERROR: Empty or invalid response"
-            return result.strip().split("Answer:")[-1].strip()
+            fixed_answer = self.agent.run(question)
+            print(f"Agent returning answer: {fixed_answer}")
+            return fixed_answer
         except Exception as e:
-            print(f"Model inference failed: {e}")
+            print(f"Error during inference: {e}")
             return f"AGENT ERROR: {e}"
 
-
+# --- Run and Submit ---
 def run_and_submit_all(profile: gr.OAuthProfile | None):
-    space_id = os.getenv("SPACE_ID")
+    space_id = "l3xv/Final_Assignment_Template"
     if profile:
         username = f"{profile.username}"
-        print(f"User logged in: {username}")
     else:
-        print("User not logged in.")
         return "Please Login to Hugging Face with the button.", None
 
     api_url = DEFAULT_API_URL
@@ -64,7 +73,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     submit_url = f"{api_url}/submit"
 
     try:
-        agent = GAIAAgent()
+        agent = BasicAgent()
     except Exception as e:
         return f"Error initializing agent: {e}", None
 
@@ -82,11 +91,18 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
-        file_name = item.get("file_name")
+        try:
+            file_path = download_file_if_any(api_url, task_id)
+        except Exception as e:
+            file_path = None
+
+        q_for_agent = f"{question_text}\n\n---\nFile: {file_path}\n---\n" if file_path else question_text
+
         if not task_id or question_text is None:
             continue
+
         try:
-            submitted_answer = agent(question_text, file_name)
+            submitted_answer = agent(q_for_agent)
             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:
@@ -112,9 +128,9 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
-
+# --- UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("# GAIA Agent Evaluation Runner")
+    gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""
         **Instructions:**
         1. Log in to your Hugging Face account.
@@ -125,12 +141,8 @@ with gr.Blocks() as demo:
     run_button = gr.Button("Run Evaluation & Submit All Answers")
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
     results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
-
-    run_button.click(
-        fn=run_and_submit_all,
-        outputs=[status_output, results_table]
-    )
+    run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
 
 if __name__ == "__main__":
     print("Launching GAIA agent app...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)