Update app.py

app.py

@@ -34,8 +34,7 @@ llm = HuggingFacePipeline(pipeline=pipe)
 system_prompt = """You are a helpful assistant tasked with answering questions using a set of tools. 
 Now, 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.
-Your answer should only start with "FINAL ANSWER: ", then follows with the 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."""
 system_message_prompt = SystemMessagePromptTemplate.from_template(system_prompt)
 
 # --- Tools ---
@@ -103,6 +102,12 @@ def create_agent(llm, tools):
     llm_chain = LLMChain(llm=llm, prompt=prompt)
     return llm_chain
 
+def extract_final_answer(full_response):
+    """Extract only the final answer from the agent's response."""
+    if "FINAL ANSWER:" in full_response:
+        return full_response.split("FINAL ANSWER:")[1].strip()
+    return full_response.strip()
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the EnhancedAgent on them, submits all answers,
@@ -160,16 +165,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            # Get direct answer from the agent (no prefixes or explanations)
-            submitted_answer = agent.run(question_text)
-            # Extract the final answer part
-            if "FINAL ANSWER:" in submitted_answer:
-                final_answer = submitted_answer.split("FINAL ANSWER:")[1].strip()
-            else:
-                final_answer = submitted_answer.strip()
+            # Get the response from the agent
+            agent_response = agent.run(question_text)
+            # Extract just the final answer part
+            final_answer = extract_final_answer(agent_response)
             
+            # Add to payload for submission
             answers_payload.append({"task_id": task_id, "submitted_answer": final_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": final_answer})
+            print(f"Task {task_id}: Processed answer: {final_answer}")
         except Exception as e:
             print(f"Error running agent on task {task_id}: {e}")
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})