Update app.py

app.py

@@ -3,45 +3,88 @@ import time
 import requests
 import gradio as gr
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, VisitWebpageTool
+from huggingface_hub import InferenceClient
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
+def web_search(query: str) -> str:
+    """Search using DuckDuckGo"""
+    try:
+        from duckduckgo_search import DDGS
+        with DDGS() as ddgs:
+            results = list(ddgs.text(query, max_results=3))
+        if results:
+            return "\n".join([f"- {r['title']}: {r['body']}" for r in results])
+    except:
+        pass
+    return ""
+
+
 class BasicAgent:
     def __init__(self):
-        print("Initializing HuggingFace agent...")
-        
-        self.model = InferenceClientModel(
-            model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+        print("Initializing agent...")
+        self.client = InferenceClient(
+            model="Qwen/Qwen2.5-72B-Instruct",
             token=os.environ.get("HF_TOKEN"),
         )
-        
-        self.agent = CodeAgent(
-            model=self.model,
-            tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
-            max_steps=3,
-            verbosity_level=0,
-        )
         print("✅ Ready")
     
+    def ask(self, prompt: str) -> str:
+        """Simple LLM call"""
+        try:
+            response = self.client.chat_completion(
+                messages=[{"role": "user", "content": prompt}],
+                max_tokens=50,
+                temperature=0.1,
+            )
+            return response.choices[0].message.content.strip()
+        except Exception as e:
+            print(f"    LLM error: {e}")
+            return ""
+    
     def __call__(self, question: str, task_id: str = None) -> str:
         # Handle reversed text
-        if '.rewsna' in question or 'tfel' in question:
+        if '.rewsna' in question or 'tfel' in question or 'eht fo' in question:
             question = question[::-1]
+            print(f"    [Reversed → {question[:50]}...]")
         
-        try:
-            prompt = f"Answer in 1-5 words only: {question}"
-            answer = str(self.agent.run(prompt)).strip()
-            
-            # Clean
-            for p in ["Answer:", "The answer is", "Final answer:"]:
-                if answer.lower().startswith(p.lower()):
-                    answer = answer[len(p):].strip()
-            return answer.strip('."\'') or "unknown"
-        except Exception as e:
-            print(f"Error: {e}")
+        # Search for context
+        search_results = web_search(question[:100])
+        
+        # Build simple prompt
+        context = f"Search results:\n{search_results}\n\n" if search_results else ""
+        
+        prompt = f"""{context}Question: {question}
+
+Answer with ONLY the final answer. 
+- If it's a number, just the number (e.g., "42")
+- If it's a name, just the name (e.g., "John Smith")
+- If it's a list, comma-separated (e.g., "apple, banana, cherry")
+- Maximum 5 words
+
+Answer:"""
+        
+        answer = self.ask(prompt)
+        
+        # Clean the answer
+        if not answer:
             return "unknown"
+        
+        # Remove common prefixes
+        for prefix in ["Answer:", "The answer is:", "The answer is", "A:", "Final answer:"]:
+            if answer.lower().startswith(prefix.lower()):
+                answer = answer[len(prefix):].strip()
+        
+        # Remove quotes and periods
+        answer = answer.strip('."\'')
+        
+        # If answer is too long or contains excuses, retry with simpler prompt
+        if len(answer) > 100 or any(x in answer.lower() for x in ["i cannot", "i don't", "unable"]):
+            answer = self.ask(f"In 1-3 words, answer: {question}")
+            answer = answer.strip('."\'')
+        
+        return answer if answer else "unknown"
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -51,7 +94,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     username = profile.username
     space_id = os.getenv("SPACE_ID")
     
-    print(f"\nUser: {username}")
+    print(f"\n{'='*40}\nUser: {username}\n{'='*40}")
     
     try:
         agent = BasicAgent()
@@ -73,11 +116,20 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         question = q.get("question", "")
         
         print(f"[{i+1}] {question[:50]}...")
-        answer = agent(question, task_id)
-        print(f"    → {answer[:50]}")
+        
+        try:
+            answer = agent(question, task_id)
+        except Exception as e:
+            print(f"    Error: {e}")
+            answer = "unknown"
+        
+        print(f"    → {answer}")
         
         answers.append({"task_id": task_id, "submitted_answer": answer})
         results.append({"#": i+1, "Q": question[:40]+"...", "A": answer[:50]})
+        
+        # Small delay to avoid rate limits
+        time.sleep(1)
     
     total = time.time() - start
     print(f"\n⏱️ {total:.0f}s")
@@ -85,7 +137,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     try:
         result = requests.post(
             f"{DEFAULT_API_URL}/submit",
-            json={"username": username, "agent_code": f"https://huggingface.co/spaces/{space_id}/tree/main", "answers": answers},
+            json={
+                "username": username,
+                "agent_code": f"https://huggingface.co/spaces/{space_id}/tree/main",
+                "answers": answers
+            },
             timeout=60
         ).json()
         
@@ -101,7 +157,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎯 GAIA Agent - HuggingFace")
+    gr.Markdown("# 🎯 GAIA Agent - Simple Mode")
+    gr.Markdown("Direct search + LLM (no code execution)")
     gr.LoginButton()
     btn = gr.Button("🚀 Run", variant="primary")
     status = gr.Textbox(label="Status", lines=5)