Update app.py

app.py

@@ -73,14 +73,22 @@ def user_input(user_question):
         return_only_outputs=True
     )
 
+    # Debugging: Print the original response
+    print("Original Response:", response['output_text'])
+
     # Extract the thought process
     thought_process = ""
     if "<think>" in response['output_text'] and "</think>" in response['output_text']:
-        thought_process = response['output_text'].split("<think>")[1].split("</think>")[0].strip()
+        thought_process_match = re.search(r"<think>(.*?)</think>", response['output_text'], re.DOTALL)
+        if thought_process_match:
+            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()
 
+    # Debugging: Print the cleaned response
+    print("Cleaned Response:", clean_response)
+
     # Display the model's thought process in the expander
     with st.expander("Model Thought Process"):
         st.write(thought_process)
@@ -126,4 +134,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-