Update pages/python.py

pages/python.py

@@ -46,35 +46,37 @@ deep_seek = ChatHuggingFace(
 
 # --- Session State ---
 PAGE_KEY = "python_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 python mentor who has {experience_label} years of experience who teaches in a very friendly manner, if question is not from python dont answer just tell that this is out of topic query from python"
-
-    # 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 python mentor who has {experience_label} years of experience who teaches in a very friendly manner, if question is not from python dont answer just tell that this is out of topic query from python"
+    
+        # 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!')