Update pages/deep_learning.py

pages/deep_learning.py

@@ -43,35 +43,38 @@ deep_seek = ChatHuggingFace(
     task='conversational'
 )
 PAGE_KEY = "deep_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 deep 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 deep learning then you must not answer and tell that it is out of topic query."
-
-    # 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 deep 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 deep learning then you must not answer and tell that it is out of topic query."
+    
+        # 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!')
 # for i, (user, bot) in enumerate(st.session_state.chat_history):
 #     st.markdown(f"**You:** {user}")
 #     st.markdown(f"**Mentor:** {bot}")