Update app.py

app.py

@@ -2,72 +2,37 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import CodeAgent, HfApiModel, tool
+from smolagents import HfApiModel
 
-# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# ====================== TOOLS ======================
-@tool
-def web_search(query: str) -> str:
-    """Search the web."""
-    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:
-    """Simple math calculation."""
-    try:
-        import math
-        return str(eval(expression, {"__builtins__": {}}, {"math": math}))
-    except:
-        return "Calc failed."
-
-# ====================== FINAL 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],
-            add_base_tools=True,
-            verbosity_level=0,        # CRITICAL: no thinking noise
-            max_steps=8,
-            planning_interval=3
-        )
 
     def __call__(self, question: str) -> str:
         try:
-            # Run the agent
-            raw_output = self.agent.run(question)
-            
-            # === CLEAN OUTPUT - THIS IS THE MOST IMPORTANT PART ===
-            answer = str(raw_output).strip()
+            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)
+            answer = str(output).strip()
             
-            # Remove common extra text that kills the score
-            if "Final Answer:" in answer:
-                answer = answer.split("Final Answer:")[-1].strip()
+            # Final cleaning
             if "Answer:" in answer:
                 answer = answer.split("Answer:")[-1].strip()
-            
-            # Take only the last line if it's very long
-            if len(answer) > 400:
-                answer = answer.split("\n")[-1].strip()
-            
-            # Final safety: keep it short and clean
-            return answer[:600].strip()
-            
+            return answer[:500].strip()
+
         except Exception as e:
             return f"Error: {str(e)[:100]}"
 
 
-# ====================== SUBMISSION ======================
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not profile:
         return "Please login with Hugging Face first.", None
@@ -82,7 +47,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else ""
 
-    # Fetch questions
     response = requests.get(questions_url, timeout=20)
     questions_data = response.json()
 
@@ -103,7 +67,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             "Answer": answer[:150] + "..." if len(answer) > 150 else answer
         })
 
-    # Submit
     submission_data = {
         "username": username,
         "agent_code": agent_code,
@@ -115,16 +78,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         response.raise_for_status()
         result = response.json()
         score = result.get("score", 0)
-        status = f"✅ Done!\nScore: {score}% ({result.get('correct_count',0)}/20)\nMessage: {result.get('message','')}"
+        status = f"✅ Submission Successful!\nScore: {score}% ({result.get('correct_count',0)}/20)\nMessage: {result.get('message','')}"
         return status, pd.DataFrame(results_log)
     except Exception as e:
         return f"Submission failed: {str(e)}", pd.DataFrame(results_log)
 
 
-# ====================== GRADIO ======================
 with gr.Blocks() as demo:
-    gr.Markdown("# Unit 4 - Final GAIA Agent")
-    gr.Markdown("**Login → Click the button** (takes 8-15 min)")
+    gr.Markdown("# Unit 4 - GAIA Agent (Simple Version)")
+    gr.Markdown("Login → Click the button below.")
 
     gr.LoginButton()
     btn = gr.Button("🚀 Run Evaluation & Submit All Answers", variant="primary", size="large")