Update pages/deep_learning.py

pages/deep_learning.py

@@ -1,26 +1,22 @@
 import streamlit as st
 import os
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
-from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
+from langchain_core.messages import HumanMessage, SystemMessage
 
-# Load Hugging Face token
 hf = os.getenv('Data_science')
 os.environ['HUGGINGFACEHUB_API_TOKEN'] = hf
 os.environ['HF_TOKEN'] = hf
 
-# --- Page Configuration ---
-st.set_page_config(page_title="AI Mentor Chat", layout="centered")
-st.title("🤖 AI Mentor Chat")
+st.set_page_config(page_title="Python Mentor Chat", layout="centered")
+st.title("🐍 Python Mentor Chat")
 
-# --- Sidebar: Experience Level ---
 st.sidebar.title("Mentor Preferences")
 experience_label = st.sidebar.selectbox(
     "Select your experience level:", ["Beginner", "Intermediate", "Experienced"]
 )
 
-# --- Initialize Chat Model ---
 deep_seek_skeleton = HuggingFaceEndpoint(
-    repo_id='Qwen/Qwen3-32B',
+    repo_id='meta-llama/Llama-3.2-3B-Instruct',
     provider='sambanova',
     temperature=0.7,
     max_new_tokens=150,
@@ -29,38 +25,32 @@ deep_seek_skeleton = HuggingFaceEndpoint(
 
 deep_seek = ChatHuggingFace(
     llm=deep_seek_skeleton,
-    repo_id='Qwen/Qwen3-32B',
+    repo_id='meta-llama/Llama-3.2-3B-Instruct',
     provider='sambanova',
     temperature=0.7,
     max_new_tokens=150,
     task='conversational'
 )
 
-PAGE_KEY = "deep_learning_chat_history"
+PAGE_KEY = "python_chat_history"
 
-# --- Session State ---
 if PAGE_KEY not in st.session_state:
     st.session_state[PAGE_KEY] = []
 
-# --- Chat Input Form ---
 with st.form(key="chat_form"):
     user_input = st.text_input("Ask your question:")
     submit = st.form_submit_button("Send")
 
-# --- Chat Handling ---
 if submit and user_input:
-    # System context based on selected experience level
     system_prompt = (
-        f"Act as a deep learning mentor with {experience_label.lower()} experience. "
+        f"Act as a python 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."
+        f"If a question is not about python, politely mention it's out of scope."
     )
-
     messages = [SystemMessage(content=system_prompt), HumanMessage(content=user_input)]
     result = deep_seek.invoke(messages)
     st.session_state[PAGE_KEY].append((user_input, result.content))
 
-# --- Chat History Display ---
 st.subheader("🗨️ Chat History")
 for user, bot in st.session_state[PAGE_KEY]:
     st.markdown(f"**You:** {user}")