Add system prompt to SmolAgent and update response parsing logic

app.py

@@ -9,6 +9,7 @@ from smolagents import CodeAgent, InferenceClientModel
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 HF_TOKEN = os.getenv("HF_TOKEN")
+SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
 
 # --- Smol Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
@@ -28,21 +29,32 @@ class SmolAgent:
         self._agent = CodeAgent(
             tools=[],
             model=model,
+            instructions=SYSTEM_PROMPT,
         )
         print("SmolAgent initialized.")
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        # 6. Prompt carefully - optimized for evaluation tasks
-        prompt = f"""Answer the following question with a short, direct response. Be concise and accurate:
-
-Question: {question}
-
-Answer:"""
         try:
-            answer = self._agent.run(prompt)
-            print(f"Agent returning answer: {answer}")
-            return str(answer)
+            # The system prompt is already set, just pass the question.
+            full_response = self._agent.run(question)
+            print(f"Agent full response: {full_response}")
+
+            # Parse the response to extract the final answer
+            if "FINAL ANSWER:" in full_response:
+                # Split and get the part after "FINAL ANSWER:"
+                final_answer_part = full_response.split("FINAL ANSWER:")[1]
+                # Remove brackets if they exist and strip whitespace
+                final_answer = final_answer_part.strip()
+                if final_answer.startswith('[') and final_answer.endswith(']'):
+                    final_answer = final_answer[1:-1]
+                
+                print(f"Agent returning parsed answer: {final_answer}")
+                return final_answer
+            else:
+                print("Warning: 'FINAL ANSWER:' not found in response. Returning full response.")
+                return full_response
+
         except Exception as e:
             print(f"Error running agent: {e}")
             return f"AGENT ERROR: {e}"