Remove <think> tag from submitted answers in processing logic

app.py

@@ -107,6 +107,9 @@ async def run_and_submit_all( profile: gr.OAuthProfile | None):
                 prompt += f'\nHere is the attached file you might need to answer the question: files/{item.get("file_name", "")}'
             r = await agent.run(prompt)
             submitted_answer = r.response.blocks[0].text
+            # Remove <think> tag and its contents if present in the submitted answer
+            if "</think>" in submitted_answer:
+                submitted_answer = submitted_answer.split("</think>")[-1].strip()
             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})
             print(f"Successfully processed task ID: {task_id}, Question: {question_text}, Submitted Answer: {submitted_answer}")

test.ipynb

@@ -302,33 +302,31 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": 15,
    "id": "ecc8d8be",
    "metadata": {},
    "outputs": [
     {
-     "name": "stdout",
-     "output_type": "stream",
-     "text": [
-      "You are a general AI assistant. I will ask you a question. Report your thoughts, and return\n",
-      "your answer.\n",
-      "\n",
-      "YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of\n",
-      "numbers and/or strings.\n",
-      "If you are asked for a number, don’t use comma to write your number neither use units such as $ or percent\n",
-      "sign unless specified otherwise.\n",
-      "If you are asked for a string, don’t use articles, neither abbreviations (e.g. for cities), and write the digits in\n",
-      "plain text unless specified otherwise.\n",
-      "If you are asked for a comma separated list, apply the above rules depending of whether the element to be put\n",
-      "in the list is a number or a string.\n",
-      "\n",
-      "Here is the question: \n",
-      ".rewsna eht sa \"tfel\" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI\n"
-     ]
+     "data": {
+      "text/plain": [
+       "'132,133,134,197,245'"
+      ]
+     },
+     "execution_count": 15,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
-    "print(prompt)"
+    "submitted_answer = \"\"\"<think>\n",
+    "Okay, let me see. The user needs the page numbers from the audio recording for their Calculus mid-term. I used the audio_to_text tool to convert the MP3 file to text. The transcription mentions pages 245, 197, 132, 133, and 134. Now I need to list these in ascending order. Let me check the numbers again.\n",
+    "\n",
+    "Pages mentioned are 132, 133, 134, 197, and 245. Wait, the user wants them in ascending order. Let me sort them: 132, 133, 134, 197, 245. That's correct. The answer should be a comma-separated list without any extra text. Just the numbers in order. I need to make sure there are no duplicates and that each page is listed once. The user also wants them in ascending order, so I'll arrange them from smallest to largest. Yep, that's right. So the final answer is 132,133,134,197,245.\n",
+    "</think>\n",
+    "\n",
+    "132,133,134,197,245\"\"\"\n",
+    "\n",
+    "submitted_answer.split(\"</think>\")[-1].strip()"
    ]
   }
  ],