Update pages/deep_learning.py

pages/deep_learning.py

@@ -3,27 +3,20 @@ import os
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
 
-
-
+# Load Hugging Face token
 hf = os.getenv('Data_science')
 os.environ['HUGGINGFACEHUB_API_TOKEN'] = hf
 os.environ['HF_TOKEN'] = hf
-# --- Config ---
+
+# --- Page Configuration ---
 st.set_page_config(page_title="AI Mentor Chat", layout="centered")
 st.title("🤖 AI Mentor Chat")
 
-# --- Sidebar for selections ---
+# --- Sidebar: Experience Level ---
 st.sidebar.title("Mentor Preferences")
-exp1 = ['<1', '1', '2', '3', '4', '5', '5+']
-exp = st.sidebar.selectbox("Select experience:", exp1)
-
-# Map experience to label
-experience_map = {
-    '<1': 'New bie mentor',
-    '1': '1', '2': '2', '3': '3', '4': '4', '5': '5',
-    '5+': 'Professional'
-}
-experience_label = experience_map[exp]
+experience_label = st.sidebar.selectbox(
+    "Select your experience level:", ["Beginner", "Intermediate", "Experienced"]
+)
 
 # --- Initialize Chat Model ---
 deep_seek_skeleton = HuggingFaceEndpoint(
@@ -42,31 +35,32 @@ deep_seek = ChatHuggingFace(
     max_new_tokens=150,
     task='conversational'
 )
-PAGE_KEY = "deep_learning_chat_history" 
+
+PAGE_KEY = "deep_learning_chat_history"
+
 # --- Session State ---
 if PAGE_KEY not in st.session_state:
     st.session_state[PAGE_KEY] = []
 
-# --- Chat Form ---
+# --- Chat Input Form ---
 with st.form(key="chat_form"):
     user_input = st.text_input("Ask your question:")
     submit = st.form_submit_button("Send")
 
-# --- Chat Logic ---
+# --- Chat Handling ---
 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."
+    # System context based on selected experience level
+    system_prompt = (
+        f"Act as a deep learning mentor with {experience_label.lower()} experience. "
+        f"Teach in a friendly manner and keep answers within 150 words. "
+        f"If a question is outside deep learning, politely decline to answer."
+    )
 
-    # 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 ---
+# --- Chat History Display ---
 st.subheader("🗨️ Chat History")
 for user, bot in st.session_state[PAGE_KEY]:
     st.markdown(f"**You:** {user}")