Update app.py

app.py

@@ -74,7 +74,8 @@ def user_input(user_question):
     )
 
     # Debugging: Print the original response
-    print("Original Response:", response['output_text'])
+    original_response = response['output_text']
+    print("Original Response:", original_response)
 
     # Extract the thought process
     thought_process = ""
@@ -84,7 +85,7 @@ def user_input(user_question):
             thought_process = thought_process_match.group(1).strip()
 
     # Remove the thought process from the main response
-    clean_response = response['output_text'].replace(f"<think>{thought_process}</think>", "").strip()
+    clean_response = original_response.split("<think>")[-1].split("</think>")[0]
 
     # Debugging: Print the cleaned response
     print("Cleaned Response:", clean_response)