Update app.py

app.py

@@ -1,104 +1,104 @@
-import streamlit as st
-import numpy as np
-import joblib
-
-# Load model
-model = joblib.load("D:/Pycharm/Project3/TomekLinks_rf_model.pkl")
-
-# Setup page
-st.set_page_config(page_title="Dự đoán tốt nghiệp đúng hạn", page_icon="🎓", layout="wide")
-
-# Inject custom CSS
-st.markdown("""
-    <style>
-    html, body, [class*="css"]  {
-        font-family: 'Poppins', sans-serif;
-    }
-    .block-container {
-        padding: 2rem 2rem 2rem 2rem;
-    }
-    .stButton button {
-        background: linear-gradient(90deg, #4F8BF9 0%, #8E2DE2 100%);
-        border: none;
-        padding: 0.75em 1.5em;
-        border-radius: 8px;
-        color: white;
-        font-size: 18px;
-        font-weight: bold;
-    }
-    </style>
-""", unsafe_allow_html=True)
-
-# ===== HEADER =====
-st.markdown("<h1 style='text-align: center; color: #003366;'>🎓 DỰ BÁO KHẢ NĂNG TỐT NGHIỆP ĐÚNG HẠN</h1>", unsafe_allow_html=True)
-st.markdown("<h4 style='text-align: center; color: #666;'>Áp dụng cho sinh viên năm 3</h4>", unsafe_allow_html=True)
-st.write("---")
-
-# ===== Thông tin cá nhân =====
-with st.container():
-    col1, col2, col3 = st.columns(3)
-    with col1:
-        name = st.text_input("👤 Họ và tên")
-    with col2:
-        student_id = st.text_input("🎓 Mã sinh viên")
-    with col3:
-        major = st.selectbox("📚 Ngành học", ["0 - Công nghệ thông tin", "1 - Kinh tế"])
-
-st.write("---")
-
-# ===== Input Semester Info =====
-def input_semester(semester_label):
-    with st.expander(f"📖 {semester_label}", expanded=True):
-        col1, col2 = st.columns(2)
-
-        with col1:
-            somon0thi = st.number_input("Số môn không thi", min_value=0, value=0, key=f"somon0thi_{semester_label}")
-            sotc0thi = st.number_input("Số tín chỉ không thi", min_value=0, value=0, key=f"sotc0thi_{semester_label}")
-            sotcno = st.number_input("Số tín chỉ nợ", min_value=0, value=0, key=f"sotcno_{semester_label}")
-            mhno = st.number_input("Số môn không đạt", min_value=0, value=0, key=f"mhno_{semester_label}")
-
-        with col2:
-            TCHK = st.number_input("Tổng số tín chỉ", min_value=0, value=0, key=f"TCHK_{semester_label}")
-            smhk = st.number_input("Tổng số môn học kỳ", min_value=0, value=0, key=f"smhk_{semester_label}")
-            mhpass = st.number_input("Số môn đạt", min_value=0, value=0, key=f"mhpass_{semester_label}")
-            TBCHK = st.number_input("GPA", min_value=0.0, max_value=10.0, value=0.0, step=0.01, key=f"TBCHK_{semester_label}")
-
-            xep_loai_option = {
-                '0 - Giỏi': 0,
-                '1 - Khá': 1,
-                '2 - TB Khá': 2,
-                '3 - Kém': 3,
-                '4 - Xuất sắc': 4,
-                '5 - Trung Bình': 5,
-                '6 - Yếu': 6
-            }
-            xep_loai_selected = st.selectbox("Xếp loại", list(xep_loai_option.keys()), key=f"xep_loai_{semester_label}")
-            xep_loai_value = xep_loai_option[xep_loai_selected]
-
-    return [somon0thi, sotc0thi, sotcno, mhno, TCHK, smhk, mhpass, TBCHK, xep_loai_value]
-
-# ===== Collect Data =====
-data = []
-semesters = ["HỌC KỲ I", "HỌC KỲ II", "HỌC KỲ III", "HỌC KỲ IV", "HỌC KỲ V", "HỌC KỲ VI"]
-for sem in semesters:
-    data += input_semester(sem)
-
-# ===== Prepare Data for Prediction =====
-nganh = 0 if major.startswith("0") else 1
-final_features = np.array(data + [nganh]).reshape(1, -1)
-
-# ===== Predict Button =====
-st.write("")
-if st.button("🎯 DỰ BÁO"):
-    prediction = model.predict(final_features)
-
-    if prediction[0] == 1:
-        st.success(f"🎉 Chúc mừng bạn {name} - {major.split('-')[1].strip()}! Bạn có khả năng tốt nghiệp đúng hạn!")
-        st.balloons()
-    else:
-        st.error(f"⚠️ Bạn {name} - {major.split('-')[1].strip()} cần cố gắng hơn! Có nguy cơ trễ hạn.")
-        st.snow()
-
-# ===== Footer =====
-st.write("---")
-st.caption("Developed by Quán Trà AI | 2025")
+import streamlit as st
+import numpy as np
+import joblib
+
+# Load model
+model = joblib.load("GradOnTime/TomekLinks_rf_model.pkl")
+
+# Setup page
+st.set_page_config(page_title="Dự đoán tốt nghiệp đúng hạn", page_icon="🎓", layout="wide")
+
+# Inject custom CSS
+st.markdown("""
+    <style>
+    html, body, [class*="css"]  {
+        font-family: 'Poppins', sans-serif;
+    }
+    .block-container {
+        padding: 2rem 2rem 2rem 2rem;
+    }
+    .stButton button {
+        background: linear-gradient(90deg, #4F8BF9 0%, #8E2DE2 100%);
+        border: none;
+        padding: 0.75em 1.5em;
+        border-radius: 8px;
+        color: white;
+        font-size: 18px;
+        font-weight: bold;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+# ===== HEADER =====
+st.markdown("<h1 style='text-align: center; color: #003366;'>🎓 DỰ BÁO KHẢ NĂNG TỐT NGHIỆP ĐÚNG HẠN</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center; color: #666;'>Áp dụng cho sinh viên năm 3</h4>", unsafe_allow_html=True)
+st.write("---")
+
+# ===== Thông tin cá nhân =====
+with st.container():
+    col1, col2, col3 = st.columns(3)
+    with col1:
+        name = st.text_input("👤 Họ và tên")
+    with col2:
+        student_id = st.text_input("🎓 Mã sinh viên")
+    with col3:
+        major = st.selectbox("📚 Ngành học", ["0 - Công nghệ thông tin", "1 - Kinh tế"])
+
+st.write("---")
+
+# ===== Input Semester Info =====
+def input_semester(semester_label):
+    with st.expander(f"📖 {semester_label}", expanded=True):
+        col1, col2 = st.columns(2)
+
+        with col1:
+            somon0thi = st.number_input("Số môn không thi", min_value=0, value=0, key=f"somon0thi_{semester_label}")
+            sotc0thi = st.number_input("Số tín chỉ không thi", min_value=0, value=0, key=f"sotc0thi_{semester_label}")
+            sotcno = st.number_input("Số tín chỉ nợ", min_value=0, value=0, key=f"sotcno_{semester_label}")
+            mhno = st.number_input("Số môn không đạt", min_value=0, value=0, key=f"mhno_{semester_label}")
+
+        with col2:
+            TCHK = st.number_input("Tổng số tín chỉ", min_value=0, value=0, key=f"TCHK_{semester_label}")
+            smhk = st.number_input("Tổng số môn học kỳ", min_value=0, value=0, key=f"smhk_{semester_label}")
+            mhpass = st.number_input("Số môn đạt", min_value=0, value=0, key=f"mhpass_{semester_label}")
+            TBCHK = st.number_input("GPA", min_value=0.0, max_value=10.0, value=0.0, step=0.01, key=f"TBCHK_{semester_label}")
+
+            xep_loai_option = {
+                '0 - Giỏi': 0,
+                '1 - Khá': 1,
+                '2 - TB Khá': 2,
+                '3 - Kém': 3,
+                '4 - Xuất sắc': 4,
+                '5 - Trung Bình': 5,
+                '6 - Yếu': 6
+            }
+            xep_loai_selected = st.selectbox("Xếp loại", list(xep_loai_option.keys()), key=f"xep_loai_{semester_label}")
+            xep_loai_value = xep_loai_option[xep_loai_selected]
+
+    return [somon0thi, sotc0thi, sotcno, mhno, TCHK, smhk, mhpass, TBCHK, xep_loai_value]
+
+# ===== Collect Data =====
+data = []
+semesters = ["HỌC KỲ I", "HỌC KỲ II", "HỌC KỲ III", "HỌC KỲ IV", "HỌC KỲ V", "HỌC KỲ VI"]
+for sem in semesters:
+    data += input_semester(sem)
+
+# ===== Prepare Data for Prediction =====
+nganh = 0 if major.startswith("0") else 1
+final_features = np.array(data + [nganh]).reshape(1, -1)
+
+# ===== Predict Button =====
+st.write("")
+if st.button("🎯 DỰ BÁO"):
+    prediction = model.predict(final_features)
+
+    if prediction[0] == 1:
+        st.success(f"🎉 Chúc mừng bạn {name} - {major.split('-')[1].strip()}! Bạn có khả năng tốt nghiệp đúng hạn!")
+        st.balloons()
+    else:
+        st.error(f"⚠️ Bạn {name} - {major.split('-')[1].strip()} cần cố gắng hơn! Có nguy cơ trễ hạn.")
+        st.snow()
+
+# ===== Footer =====
+st.write("---")
+st.caption("Developed by Quán Trà AI | 2025")