Create pages/6 Deployment.py

pages/6 Deployment.py

@@ -0,0 +1,53 @@
+import streamlit as st
+import pandas as pd
+import pickle
+
+# Page Title
+st.title("🎯 Student Depression Prediction")
+st.subheader("🧾 Please fill in the student's details")
+
+# Input Fields
+Gender = st.radio("👤 Gender:", ["Male", "Female"])
+Age = st.number_input("🎂 Age:", min_value=18.0, max_value=59.0, step=1.0)
+Academic_pressure = st.radio("📚 Academic Pressure (1 = Low, 5 = High):", [1., 2., 3., 4., 5.])
+cgpa = st.number_input("📈 CGPA (0 - 10):", min_value=0.0, max_value=10.0)
+study_satisfaction = st.radio("😊 Study Satisfaction (1 = Low, 5 = High):", [1., 2., 3., 4., 5.])
+Sleep_Duration = st.radio("🛌 Sleep Duration:", [
+    "Less than 5 hours", "5-6 hours", "7-8 hours", "More than 8 hours"])
+dietary_habits = st.radio("🥗 Dietary Habits:", ["Unhealthy", "Moderate", "Healthy"])
+workStudy_Hours = st.slider("💻 Work/Study Hours per Day:", min_value=1.0, max_value=12.0, step=1.0)
+Financial_Stress = st.radio("💸 Financial Stress (1 = Low, 5 = High):", [1.0, 2.0, 3.0, 4.0, 5.0])
+Family_History_of_Mental_Illness = st.radio("🧬 Family History of Mental Illness:", ["Yes", "No"])
+
+# Load trained model
+with open("demo.pkl", "rb") as f:
+    knn = pickle.load(f)
+
+# Predict on submit
+if st.button("🚀 Predict Depression Status"):
+    input_data = pd.DataFrame([{
+        'Gender': Gender,
+        'Age': Age,
+        'Academic Pressure': Academic_pressure,
+        'CGPA': cgpa,
+        'Study Satisfaction': study_satisfaction,
+        'Sleep Duration': Sleep_Duration,
+        'Dietary Habits': dietary_habits,
+        'Work/Study Hours': workStudy_Hours,
+        'Financial Stress': Financial_Stress,
+        'Family History of Mental Illness': Family_History_of_Mental_Illness
+    }])
+
+    prediction = knn.predict(input_data)
+
+    st.markdown("### 🧠 Depression Prediction Result")
+    if prediction[0] == 1:
+        st.error("🔴 The student is likely experiencing **depression**.")
+    else:
+        st.success("🟢 The student is **not likely** experiencing depression.")
+
+if st.button("<< Back"):
+    st.switch_page(r"pages\5 Model Building.py")
+
+if st.button("Go to Main Page>>"):
+    st.switch_page("app.py")