Update app.py

app.py

@@ -3,90 +3,45 @@ import time
 import requests
 import gradio as gr
 import pandas as pd
-from groq import Groq
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, VisitWebpageTool
 
-# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# ============================================
-# SIMPLE FAST AGENT
-# ============================================
 
 class BasicAgent:
     def __init__(self):
-        api_key = os.environ.get("GROQ_API_KEY")
-        if not api_key:
-            raise ValueError("GROQ_API_KEY not set!")
-        self.client = Groq(api_key=api_key)
-        print("✅ Agent ready")
-    
-    def search(self, query: str) -> str:
-        """Quick web search"""
-        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.get('title','')}: {r.get('body','')}" for r in results])
-        except:
-            pass
-        return ""
-    
-    def ask(self, prompt: str) -> str:
-        """Ask LLM - use 8B model for speed and rate limits"""
-        try:
-            response = self.client.chat.completions.create(
-                model="llama-3.1-8b-instant",  # Fast, high rate limit
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0,
-                max_tokens=50,  # Short answers only
-            )
-            return response.choices[0].message.content.strip()
-        except Exception as e:
-            if "rate" in str(e).lower():
-                time.sleep(5)
-                try:
-                    response = self.client.chat.completions.create(
-                        model="llama-3.1-8b-instant",
-                        messages=[{"role": "user", "content": prompt}],
-                        temperature=0,
-                        max_tokens=50,
-                    )
-                    return response.choices[0].message.content.strip()
-                except:
-                    return ""
-            return ""
+        print("Initializing HuggingFace agent...")
+        
+        self.model = HfApiModel(
+            model_id="Qwen/Qwen2.5-Coder-32B-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 __call__(self, question: str, task_id: str = None) -> str:
         # Handle reversed text
-        if '.rewsna' in question or 'eht sa' in question or 'tfel' in question:
+        if '.rewsna' in question or 'tfel' in question:
             question = question[::-1]
         
-        # Search for info
-        search_results = self.search(question[:80])
-        
-        # Build prompt
-        context = f"Info: {search_results[:1000]}\n\n" if search_results else ""
-        
-        prompt = f"""{context}Q: {question}
-
-Give only the final answer in 1-5 words. No explanation."""
-        
-        answer = self.ask(prompt)
-        
-        # Clean answer
-        for prefix in ["Answer:", "The answer is", "A:", "Final answer:"]:
-            if answer.lower().startswith(prefix.lower()):
-                answer = answer[len(prefix):].strip()
-        
-        answer = answer.strip('."\'')
-        
-        # Filter bad responses
-        if any(x in answer.lower() for x in ["i cannot", "i'm unable", "no code", "no image", "i don't"]):
-            answer = self.ask(f"Answer in 1-3 words only: {question}")
-            answer = answer.strip('."\'')
-        
-        return answer if answer else "unknown"
+        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}")
+            return "unknown"
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -96,15 +51,12 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     username = profile.username
     space_id = os.getenv("SPACE_ID")
     
-    if not os.environ.get("GROQ_API_KEY"):
-        return "❌ Add GROQ_API_KEY!", None
-    
-    print(f"\n{'='*40}\nUser: {username}\n{'='*40}")
+    print(f"\nUser: {username}")
     
     try:
         agent = BasicAgent()
     except Exception as e:
-        return f"❌ {e}", None
+        return f"❌ Agent failed: {e}", None
     
     try:
         questions = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15).json()
@@ -122,12 +74,10 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         
         print(f"[{i+1}] {question[:50]}...")
         answer = agent(question, task_id)
-        print(f"    → {answer}")
+        print(f"    → {answer[:50]}")
         
         answers.append({"task_id": task_id, "submitted_answer": answer})
-        results.append({"#": i+1, "Q": question[:40]+"...", "A": answer})
-        
-        time.sleep(2)  # Small delay
+        results.append({"#": i+1, "Q": question[:40]+"...", "A": answer[:50]})
     
     total = time.time() - start
     print(f"\n⏱️ {total:.0f}s")
@@ -151,11 +101,13 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 
 
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎯 GAIA Agent")
-    gr.Markdown("Fast mode - ~2 min")
+    gr.Markdown("# 🎯 GAIA Agent - HuggingFace")
     gr.LoginButton()
-    gr.Button("🚀 Run", variant="primary").click(run_and_submit_all, outputs=[gr.Textbox(label="Status", lines=5), gr.DataFrame(label="Results")])
+    btn = gr.Button("🚀 Run", variant="primary")
+    status = gr.Textbox(label="Status", lines=5)
+    table = gr.DataFrame(label="Results")
+    btn.click(run_and_submit_all, outputs=[status, table])
 
 if __name__ == "__main__":
-    print(f"GROQ: {'✅' if os.environ.get('GROQ_API_KEY') else '❌'}")
+    print(f"HF_TOKEN: {'✅' if os.environ.get('HF_TOKEN') else '❌'}")
     demo.launch()