Update app.py

app.py

@@ -13,36 +13,23 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        token = os.getenv("HF_TOKEN")
-        if not token:
-            raise ValueError("HF_TOKEN secret missing in Space settings")
-
-        # load your **ONLY** model
-        self.llm = HfApiModel(
-            token=token,
-            model="microsoft/phi-2",        # <— change this line for any other model
-            stop_sequences=["ANSWER:"],     # stop as soon as it produces an answer
-        )
+        print("BasicAgent initialized.")
+
+        hf_token = os.getenv("HF_TOKEN")
+        if not hf_token:
+            raise ValueError("HF_TOKEN environment variable is not set.")
 
-        # single system prompt once
-        self.system = (
-            "You are a helpful assistant.  "
-            "Answer each question in ONE line that starts with “ANSWER: ” "
-            "and nothing else."
+        self.agent = CodeAgent(
+            tools=[DuckDuckGoSearchTool()],
+            model=HfApiModel(token=hf_token, model="microsoft/phi-2")
         )
 
     def __call__(self, question: str) -> str:
-        messages = [
-            {"role": "system", "content": self.system},
-            {"role": "user",   "content": question},
-        ]
-        reply = self.llm(messages)      # one HTTP request
-        text  = reply["content"].strip()
-        # ensure we only return the line after “ANSWER:”
-        if "ANSWER" in text:
-            text = text.split("ANSWER", 1)[-1].lstrip(": ").splitlines()[0]
-            text = f"ANSWER: {text}"
-        return text
+        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        fixed_answer = self.agent.run("You are a helpful assistant answering short factual questions. Answer the question: " + question, max_steps=1)
+        print(fixed_answer)
+        print(f"Agent returning fixed answer: {fixed_answer}")
+        return fixed_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """