Update app.py

app.py

@@ -2,40 +2,91 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import HfApiModel
+from smolagents import CodeAgent, HfApiModel, tool
+from PIL import Image
+import io
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# ====================== TOOLS ======================
+@tool
+def web_search(query: str) -> str:
+    """Search the web for information."""
+    try:
+        from duckduckgo_search import DDGS
+        with DDGS() as ddgs:
+            results = list(ddgs.text(query, max_results=3))
+            return "\n".join([f"{r.get('title')}: {r.get('body') or r.get('snippet')}" for r in results])
+    except:
+        return "Search failed."
+
+@tool
+def calculate(expression: str) -> str:
+    """Calculate simple math."""
+    try:
+        import math
+        return str(eval(expression, {"__builtins__": {}}, {"math": math}))
+    except:
+        return "Calc failed."
+
+@tool
+def download_file(task_id: str) -> str:
+    """Download the file (image, csv, audio, etc.) attached to a question."""
+    url = f"{DEFAULT_API_URL}/files/{task_id}"
+    try:
+        r = requests.get(url, timeout=20)
+        r.raise_for_status()
+        content_type = r.headers.get("content-type", "")
+        
+        if "image" in content_type:
+            img = Image.open(io.BytesIO(r.content))
+            return f"Downloaded image: {img.size} {img.format}"
+        else:
+            text = r.text[:1500]
+            return f"Downloaded file content:\n{text}"
+    except Exception as e:
+        return f"File download failed: {str(e)}"
+
+
+# ====================== AGENT ======================
 class BasicAgent:
     def __init__(self):
-        print("🚀 Loading model...")
         self.model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
+        
+        self.agent = CodeAgent(
+            model=self.model,
+            tools=[web_search, calculate, download_file],
+            add_base_tools=True,
+            verbosity_level=0,
+            max_steps=10,
+            planning_interval=3
+        )
 
     def __call__(self, question: str) -> str:
         try:
-            prompt = f"""Answer the following question with ONLY the final answer.
-Do not write any explanation, no reasoning, no "Final Answer", no extra text.
-Just give the direct answer.
-
-Question: {question}
-
-Answer:"""
-
-            output = self.model.generate(prompt, max_new_tokens=200)
+            output = self.agent.run(question)
             answer = str(output).strip()
             
-            # Final cleaning
-            if "Answer:" in answer:
+            # Force clean final answer (this fixes 0%)
+            if "Final Answer:" in answer:
+                answer = answer.split("Final Answer:")[-1].strip()
+            elif "Answer:" in answer:
                 answer = answer.split("Answer:")[-1].strip()
-            return answer[:500].strip()
-
+            
+            # Take last line if too long
+            if len(answer) > 400:
+                answer = answer.split("\n")[-1].strip()
+            
+            return answer[:700].strip()
+            
         except Exception as e:
-            return f"Error: {str(e)[:100]}"
+            return f"Error: {str(e)[:150]}"
 
 
+# ====================== SUBMISSION ======================
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not profile:
-        return "Please login with Hugging Face first.", None
+        return "Please login first.", None
 
     username = profile.username.strip()
     api_url = DEFAULT_API_URL
@@ -61,32 +112,25 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
         answer = agent(question)
         answers_payload.append({"task_id": task_id, "submitted_answer": answer})
-        results_log.append({
-            "Task ID": task_id,
-            "Question": question[:100] + "...",
-            "Answer": answer[:150] + "..." if len(answer) > 150 else answer
-        })
-
-    submission_data = {
-        "username": username,
-        "agent_code": agent_code,
-        "answers": answers_payload
-    }
+        results_log.append({"Task ID": task_id, "Question": question[:100]+"...", "Answer": answer[:150]+"..."})
+
+    submission_data = {"username": username, "agent_code": agent_code, "answers": answers_payload}
 
     try:
         response = requests.post(submit_url, json=submission_data, timeout=90)
         response.raise_for_status()
         result = response.json()
         score = result.get("score", 0)
-        status = f"✅ Submission Successful!\nScore: {score}% ({result.get('correct_count',0)}/20)\nMessage: {result.get('message','')}"
+        status = f"✅ Done!\nScore: {score}% ({result.get('correct_count',0)}/20)\n{result.get('message','')}"
         return status, pd.DataFrame(results_log)
     except Exception as e:
         return f"Submission failed: {str(e)}", pd.DataFrame(results_log)
 
 
+# ====================== UI ======================
 with gr.Blocks() as demo:
-    gr.Markdown("# Unit 4 - GAIA Agent (Simple Version)")
-    gr.Markdown("Login → Click the button below.")
+    gr.Markdown("# Unit 4 - Final GAIA Agent (with file support)")
+    gr.Markdown("Login → Click button (takes 8-15 min)")
 
     gr.LoginButton()
     btn = gr.Button("🚀 Run Evaluation & Submit All Answers", variant="primary", size="large")