Update app.py

app.py

@@ -252,8 +252,56 @@ def assistant(state: AgentState):
     """
     system_msg = SystemMessage(
         content=(
-            "Reply with the final answer only. Do not include any preamble, "
-            "explanations, or the text 'FINAL ANSWER'."
+            "You are an expert research and reasoning agent.\n\n"
+            "You have access to tools for:\n"
+            "- web search\n"
+            "- webpage retrieval\n"
+            "- PDF reading\n"
+            "- Python execution\n"
+            "- task file retrieval\n\n"
+            "Always use tools when information is missing, uncertain, "
+            "requires computation, or depends on external documents.\n\n"
+            "When solving a task:\n\n"
+            "1. Read the question carefully.\n"
+            "2. Use tools whenever necessary.\n"
+            "3. Verify facts before answering.\n"
+            "4. Perform calculations with Python rather than mental math.\n"
+            "5. Read files and PDFs when relevant.\n"
+            "6. Continue gathering evidence until you are confident.\n\n"
+            "If the task references a file, use the task file tool.\n\n"
+            "IMPORTANT OUTPUT RULES:\n\n"
+            "- Return ONLY the final answer.\n"
+            "- Do NOT explain your reasoning.\n"
+            "- Do NOT provide analysis.\n"
+            "- Do NOT provide step-by-step solutions.\n"
+            "- Do NOT provide preambles.\n"
+            "- Do NOT provide markdown.\n"
+            "- Do NOT provide bullet points.\n"
+            "- Do NOT provide quotes unless the answer itself requires quotes.\n"
+            "- Do NOT write 'FINAL ANSWER'.\n"
+            "- Do NOT write 'The answer is'.\n"
+            "- Do NOT write any surrounding text.\n\n"
+            "Your entire response must contain only the exact answer.\n\n"
+            "Examples:\n\n"
+            "Correct:\n"
+            "Paris\n\n"
+            "Incorrect:\n"
+            "The answer is Paris\n\n"
+            "Incorrect:\n"
+            "FINAL ANSWER: Paris\n\n"
+            "Correct:\n"
+            "42\n\n"
+            "Incorrect:\n"
+            "42.\n\n"
+            "Correct:\n"
+            "1997\n\n"
+            "Incorrect:\n"
+            "The year is 1997\n\n"
+            "If the answer is a number, output only the number.\n\n"
+            "If the answer is a name, output only the name.\n\n"
+            "If the answer is a date, output only the date.\n\n"
+            "If the answer is a phrase, output only the phrase.\n\n"
+            "Be precise."
         )
     )
 
@@ -297,22 +345,29 @@ class BasicAgent:
     def __init__(self):
         self.graph = graph
 
-    def __call__(self, question: str) -> str:
+    def __call__(self, task_id: str, question: str) -> str:
 
         try:
-
+            prompt = (
+                f"Task ID: {task_id}\n\n"
+                "Question:\n"
+                f"{question}"
+            )
             result = self.graph.invoke(
                 {
                     "messages": [
-                        HumanMessage(content=question)
+                        HumanMessage(content=prompt)
                     ]
                 },
                 config={
                     "recursion_limit": 30
                 }
             )
-
-            return result["messages"][-1].content
+            answer = result["messages"][-1].content
+            answer = answer.strip()
+            if answer.startswith('"') and answer.endswith('"'):
+                answer = answer[1:-1]
+            return answer.strip()
 
         except Exception as e:
             return f"Agent error: {e}"
@@ -378,7 +433,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(task_id, question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: