Update app.py

app.py

@@ -107,7 +107,62 @@ elif st.session_state.page == "python":
 
 elif st.session_state.page == "sql":
     st.title("SQL Chatbot 🛢️")
-    st.button("⬅️ Back to Home", on_click=lambda: switch_page("home"))
+    if not hf_token:
+    st.error("Please add your Hugging Face API token as an environment variable.")
+    st.stop()
+
+# Initialize the LLaMA model from HuggingFace (via Nebius provider)
+    llama_model = HuggingFaceEndpoint(
+    repo_id="meta-llama/Llama-3.1-8B-Instruct",
+    provider="nebius",
+    temperature=0.7,
+    max_new_tokens=512,
+    task="conversational",
+    huggingfacehub_api_token=hf_token,
+    )
+
+    llama = ChatHuggingFace(
+    llm=llama_model,
+    repo_id="meta-llama/Llama-3.1-8B-Instruct",
+    provider="nebius",
+    temperature=0.7,
+    max_new_tokens=512,
+    task="conversational"
+    )
+
+# Streamlit A
+
+st.markdown("Ask anything related to SQL interviews!")
+
+# Initialize chat history
+    if "messages" not in st.session_state:
+        st.session_state.messages = [
+        SystemMessage(content="Answer clearly like a technical SQL interviewer.")
+    ]
+
+# User input
+    user_input = st.text_input("💡 Ask your SQL interview question:", placeholder="e.g., give me 10 SQL interview questions with answers")
+
+    def generate_response(user_input):
+        st.session_state.messages.append(HumanMessage(content=user_input))
+        response = llama.invoke(st.session_state.messages)
+        st.session_state.messages.append(AIMessage(content=response))
+        return response
+
+# Display response
+    if user_input:
+        with st.spinner("Thinking..."):
+            answer = generate_response(user_input)
+            st.markdown(f"**Answer:** {answer}")
+
+# Show chat history
+st.markdown("### 📜 Chat History")
+    for msg in st.session_state.messages[1:]:  # Skip SystemMessage
+        if isinstance(msg, HumanMessage):
+            st.markdown(f"**You:** {msg.content}")
+        elif isinstance(msg, AIMessage):
+            st.markdown(f"**Bot:** {msg.content}")
+        st.button("⬅️ Back to Home", on_click=lambda: switch_page("home"))
     # Load SQL chatbot here
 
 elif st.session_state.page == "powerbi":