Updating app.py

app.py

@@ -68,10 +68,18 @@ class BasicAgent:
             SystemMessage(content=self.system_prompt),
             HumanMessage(content=question)
         ]
+        def extract_model_answer(full_response: str) -> str:
+            marker = "FINAL ANSWER:"
+            idx = full_response.find(marker)
+            if idx == -1:
+                return full_response.strip()  # fallback: return whole response
+            return full_response[idx + len(marker):].strip().rstrip(".")
+            
         response = self.alfred.invoke({"messages": messages})
         model_answer = response["messages"][-1].content
+        fixed_answer = extract_model_answer(model_answer)
         reasoning_trace = "\n".join([f"{m.type}: {m.content}" for m in messages + response["messages"]])
-        return model_answer
+        return fixed_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """