Update pages/Sql.py

pages/Sql.py

@@ -5,9 +5,23 @@ from langchain_core.messages import HumanMessage, SystemMessage
 
 # Set environment variables
 hf = os.getenv('HF_TOKEN')
-os.environ['HUGGINGFACEHUB_API_TOKEN'] = hf
+os.environ['HUGGINGFACEHUB_API_KEY'] = hf
 os.environ['HF_TOKEN'] = hf
-# Page setup
+
+# Initialize DeepSeek model (you can replace this with your own model if needed)
+def load_deepseek_model():
+    deep_seek_model = HuggingFaceEndpoint(
+        repo_id="deepseek-ai/DeepSeek-R1",
+        provider="nebius",
+        temperature=0.7,
+        max_new_tokens=150,
+        task="conversational"
+    )
+    return ChatHuggingFace(llm=deep_seek_model)
+
+sql_mentor = load_deepseek_model()
+
+# Streamlit page setup
 st.set_page_config(page_title="SQL Mentor Chat", layout="centered")
 
 # Improved custom CSS
@@ -76,6 +90,13 @@ st.markdown("<p style='margin-top: 0.2rem;'>Learn SQL interactively from your AI
 st.sidebar.title("Mentor Preferences")
 exp = st.sidebar.selectbox("Choose your experience level:", ["Beginner", "Intermediate", "Expert"])
 
+# Session state key
+PAGE_KEY = "chat_history"
+
+# Initialize session state
+if PAGE_KEY not in st.session_state:
+    st.session_state[PAGE_KEY] = []
+
 # Chat input box
 st.markdown("<hr>", unsafe_allow_html=True)
 with st.form(key="chat_form"):