Update pages/machine_learning.py

pages/machine_learning.py

@@ -46,32 +46,35 @@ deep_seek = ChatHuggingFace(
 
 # --- Session State ---
 PAGE_KEY = "machine_learning_chat_history" 
-# --- Session State ---
-if PAGE_KEY not in st.session_state:
-    st.session_state[PAGE_KEY] = []
-
-# --- Chat Form ---
-with st.form(key="chat_form"):
-    user_input = st.text_input("Ask your question:")
-    submit = st.form_submit_button("Send")
-
-# --- Chat Logic ---
-if submit and user_input:
-    # Add system context
-    system_prompt = f"Act as a machine learning mentor who has {experience_label} years of experience who teaches in a very friendly manner and also tells everything in within 150 words. If any question is asked other than machine learning tell politely that this is out of the topic question"
-
-    # Create message list
-    messages = [SystemMessage(content=system_prompt), HumanMessage(content=user_input)]
-
-    # Get model response
-    result = deep_seek.invoke(messages)
-
-    # Append to history
-    st.session_state[PAGE_KEY].append((user_input, result.content))
-
-# --- Display Chat History ---
-st.subheader("🗨️ Chat History")
-for user, bot in st.session_state[PAGE_KEY]:
-    st.markdown(f"**You:** {user}")
-    st.markdown(f"**Mentor:** {bot}")
-    st.markdown("---")
+try:
+    # --- Session State ---
+    if PAGE_KEY not in st.session_state:
+        st.session_state[PAGE_KEY] = []
+    
+    # --- Chat Form ---
+    with st.form(key="chat_form"):
+        user_input = st.text_input("Ask your question:")
+        submit = st.form_submit_button("Send")
+    
+    # --- Chat Logic ---
+    if submit and user_input:
+        # Add system context
+        system_prompt = f"Act as a machine learning mentor who has {experience_label} years of experience who teaches in a very friendly manner and also tells everything in within 150 words. If any question is asked other than machine learning tell politely that this is out of the topic question"
+    
+        # Create message list
+        messages = [SystemMessage(content=system_prompt), HumanMessage(content=user_input)]
+    
+        # Get model response
+        result = deep_seek.invoke(messages)
+    
+        # Append to history
+        st.session_state[PAGE_KEY].append((user_input, result.content))
+    
+    # --- Display Chat History ---
+    st.subheader("🗨️ Chat History")
+    for user, bot in st.session_state[PAGE_KEY]:
+        st.markdown(f"**You:** {user}")
+        st.markdown(f"**Mentor:** {bot}")
+        st.markdown("---")
+except:
+    st.warning('The token limit has reached please revisit in 24 hours!')