Update app.py

app.py

@@ -76,8 +76,9 @@ def user_input(user_question):
     # Extract the thought process
     thought_process = ""
     if "<think>" in response['output_text'] and "</think>" in response['output_text']:
-        thought_process = re.search(r"<think>(.*?)</think>", response['output_text'], re.DOTALL)
-        thought_process = thought_process.group(1).strip() if thought_process else ""
+        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 = re.sub(r"<think>.*?</think>", "", response['output_text']).strip()
@@ -127,4 +128,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-