Update pages/machine_learning.py

pages/machine_learning.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import re
 import os
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
@@ -73,7 +74,7 @@ try:
         result = deep_seek.invoke(messages)
     
         # Append to history
-        st.session_state[PAGE_KEY].append((user_input, result.content))
+        st.session_state[PAGE_KEY].append((user_input, re.findall('<think>.*</think>',result.content,flags =  re.DOTALL)))
     
     # --- Display Chat History ---
     st.subheader("🗨️ Chat History")