Update app.py

app.py

@@ -13,23 +13,36 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
-
-        hf_token = os.getenv("HF_TOKEN")
-        if not hf_token:
-            raise ValueError("HF_TOKEN environment variable is not set.")
+        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
+        )
 
-        self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()],
-            model=HfApiModel(token=hf_token, model="microsoft/phi-2")
+        # single system prompt once
+        self.system = (
+            "You are a helpful assistant.  "
+            "Answer each question in ONE line that starts with “ANSWER: ” "
+            "and nothing else."
         )
 
     def __call__(self, question: str) -> str:
-        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)
-        print(fixed_answer)
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        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
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """