Update app.py

app.py

@@ -2,8 +2,9 @@ import streamlit as st
 import numpy as np
 import joblib
 
-# ===== Load model =====
-model = joblib.load("TomekLinks_rf_model.pkl")
+# ===== Load models =====
+model_full = joblib.load("TomekLinks_rf_model.pkl")
+model_important = joblib.load("random_forest_model_importainfeature.pkl")
 
 # ===== Setup page =====
 st.set_page_config(page_title="Dự đoán tốt nghiệp đúng hạn", page_icon="🎓", layout="wide")
@@ -34,18 +35,23 @@ st.markdown("<h1 style='text-align: center; color: #003366;'>🎓 DỰ BÁO KH
 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("---")
 
-# ===== BỘ VÍ DỤ MẪU =====
+# ===== SIDEBAR: chọn ví dụ và mô hình =====
+model_type = st.sidebar.radio("🧠 Chọn mô hình dự báo:", ["Dùng toàn bộ dữ liệu", "Dùng mô hình đơn giản (các đặc trưng quan trọng)"])
+
+sample_option = st.sidebar.selectbox("📝 Chọn ví dụ:", ["Không ví dụ", "Dùng ví dụ mẫu ngành Công nghệ thông tin", "Dùng ví dụ mẫu ngành Kinh tế"])
+
+# ===== Ví dụ mẫu =====
 sample_cntt_example = {
     "name": "Nguyễn Văn A",
     "student_id": "10117367",
     "major": "Công nghệ thông tin",
     "semester_data": [
-        0, 0, 0, 0, 17, 7, 7, 8.73, 0,   
-        0, 0, 0, 0, 17, 7, 7, 8.19, 0,   
-        0, 0, 0, 0, 17, 7, 7, 7.90, 0,   
-        0, 0, 0, 0, 17, 7, 7, 8.19, 0,   
-        0, 0, 0, 0, 19, 7, 6, 8.18, 0,   
-        0, 0, 5, 2, 19, 7, 5, 7.10, 1    
+        0, 0, 0, 0, 17, 7, 7, 8.73, 0,
+        0, 0, 0, 0, 17, 7, 7, 8.19, 0,
+        0, 0, 0, 0, 17, 7, 7, 7.90, 0,
+        0, 0, 0, 0, 17, 7, 7, 8.19, 0,
+        0, 0, 0, 0, 19, 7, 6, 8.18, 0,
+        0, 0, 5, 2, 19, 7, 5, 7.10, 1
     ]
 }
 
@@ -54,94 +60,80 @@ sample_kinhte_example = {
     "student_id": "11418092",
     "major": "Kinh tế",
     "semester_data": [
-        0, 0, 0, 0, 17, 7, 7, 7.64, 1,   
-        0, 0, 0, 0, 17, 7, 7, 7.19, 1,   
-        0, 0, 0, 0, 17, 7, 7, 6.94, 1,   
-        0, 0, 0, 0, 17, 7, 7, 7.19, 1,   
-        0, 0, 0, 0, 19, 7, 5, 7.65, 1,   
-        0, 0, 7, 3, 19, 7, 4, 7.49, 1    
+        0, 0, 0, 0, 17, 7, 7, 7.64, 1,
+        0, 0, 0, 0, 17, 7, 7, 7.19, 1,
+        0, 0, 0, 0, 17, 7, 7, 6.94, 1,
+        0, 0, 0, 0, 17, 7, 7, 7.19, 1,
+        0, 0, 0, 0, 19, 7, 5, 7.65, 1,
+        0, 0, 7, 3, 19, 7, 4, 7.49, 1
     ]
 }
 
-# ===== THÔNG TIN CÁ NHÂN =====
-with st.container():
-    sample_option = st.selectbox(
-        "📝 Chọn ví dụ:",
-        ["Không ví dụ","Dùng ví dụ mẫu ngành Công nghệ thông tin", "Dùng ví dụ mẫu ngành Kinh tế"]
-    )
-
-    if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin":
-        name = st.text_input("👤 Họ và tên", value=sample_cntt_example["name"])
-        student_id = st.text_input("🎓 Mã sinh viên", value=sample_cntt_example["student_id"])
-        major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=0)
-    elif sample_option == "Dùng ví dụ mẫu ngành Kinh tế":
-        name = st.text_input("👤 Họ và tên", value=sample_kinhte_example["name"])
-        student_id = st.text_input("🎓 Mã sinh viên", value=sample_kinhte_example["student_id"])
-        major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=1)
-    else:
-        name = st.text_input("👤 Họ và tên")
-        student_id = st.text_input("🎓 Mã sinh viên")
-        major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"])
+# ===== Thông tin cá nhân =====
+if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin":
+    name = st.text_input("👤 Họ và tên", value=sample_cntt_example["name"])
+    student_id = st.text_input("🎓 Mã sinh viên", value=sample_cntt_example["student_id"])
+    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=0)
+elif sample_option == "Dùng ví dụ mẫu ngành Kinh tế":
+    name = st.text_input("👤 Họ và tên", value=sample_kinhte_example["name"])
+    student_id = st.text_input("🎓 Mã sinh viên", value=sample_kinhte_example["student_id"])
+    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=1)
+else:
+    name = st.text_input("👤 Họ và tên")
+    student_id = st.text_input("🎓 Mã sinh viên")
+    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"])
 
 st.write("---")
 
-# ===== Input Semester Info =====
+# ===== Hàm nhập liệu =====
 def input_semester(semester_label, default_values=None):
     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=default_values[0] if default_values else 0, key=f"somon0thi_{semester_label}")
-            sotc0thi = st.number_input("Số tín chỉ không thi", min_value=0, value=default_values[1] if default_values else 0, key=f"sotc0thi_{semester_label}")
-            sotcno = st.number_input("Số tín chỉ nợ", min_value=0, value=default_values[2] if default_values else 0, key=f"sotcno_{semester_label}")
-            mhno = st.number_input("Số môn không đạt", min_value=0, value=default_values[3] if default_values else 0, key=f"mhno_{semester_label}")
-
+            somon0thi = st.number_input("Số môn không thi", 0, value=default_values[0] if default_values else 0, key=f"sm0_{semester_label}")
+            sotc0thi = st.number_input("Số tín chỉ không thi", 0, value=default_values[1] if default_values else 0, key=f"tc0_{semester_label}")
+            sotcno = st.number_input("Số tín chỉ nợ", 0, value=default_values[2] if default_values else 0, key=f"tcno_{semester_label}")
+            mhno = st.number_input("Số môn không đạt", 0, value=default_values[3] if default_values else 0, key=f"mhno_{semester_label}")
+            sotc_qua = st.number_input("Số tín chỉ qua môn", 0, value=default_values[4] - default_values[2] if default_values else 0, key=f"tcqua_{semester_label}")
         with col2:
-            TCHK = st.number_input("Tổng số tín chỉ", min_value=0, value=default_values[4] if default_values else 0, key=f"TCHK_{semester_label}")
-            smhk = st.number_input("Tổng số môn học kỳ", min_value=0, value=default_values[5] if default_values else 0, key=f"smhk_{semester_label}")
-            mhpass = st.number_input("Số môn đạt", min_value=0, value=default_values[6] if default_values else 0, key=f"mhpass_{semester_label}")
-            TBCHK = st.number_input("GPA", min_value=0.0, max_value=10.0, value=default_values[7] if default_values else 0.0, step=0.01, key=f"TBCHK_{semester_label}")
-
-            xep_loai_option = {
-                'Giỏi': 0,
-                'Khá': 1,
-                'TB Khá': 2,
-                'Kém': 3,
-                'Xuất sắc': 4,
-                'Trung Bình': 5,
-                'Yếu': 6
-            }
-            xep_loai_selected = st.selectbox("Xếp loại", list(xep_loai_option.keys()), index=default_values[8] if default_values else 0, 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 =====
+            TCHK = st.number_input("Tổng tín chỉ học kỳ", 0, value=default_values[4] if default_values else 0, key=f"tchk_{semester_label}")
+            smhk = st.number_input("Số môn học kỳ", 0, value=default_values[5] if default_values else 0, key=f"smhk_{semester_label}")
+            mhpass = st.number_input("Số môn đạt", 0, value=default_values[6] if default_values else 0, key=f"mhpass_{semester_label}")
+            TBCHK = st.number_input("GPA", 0.0, 10.0, value=default_values[7] if default_values else 0.0, step=0.01, key=f"gpa_{semester_label}")
+            xep_loai_selected = st.selectbox("Xếp loại", list(range(7)), index=default_values[8] if default_values else 0, key=f"xeploai_{semester_label}")
+    return [somon0thi, sotc0thi, sotcno, mhno, TCHK, smhk, mhpass, TBCHK, xep_loai_selected, sotc_qua]
+
+# ===== Giao diện theo mô hình =====
 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 idx, sem in enumerate(semesters):
-    if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin":
-        data += input_semester(sem, sample_cntt_example["semester_data"][idx*9:(idx+1)*9])
-    elif sample_option == "Dùng ví dụ mẫu ngành Kinh tế":
-        data += input_semester(sem, sample_kinhte_example["semester_data"][idx*9:(idx+1)*9])
-    else:
-        data += input_semester(sem)
-
-# ===== Prepare Data for Prediction =====
-nganh = 0 if major == "Công nghệ thông tin" else 1
-final_features = np.array(data + [nganh]).reshape(1, -1)
-
-# ===== Predict Button =====
-st.write("")
+if model_type == "Dùng toàn bộ dữ liệu":
+    st.subheader("🔢 Nhập thông tin học kỳ chi tiết")
+    for idx, sem in enumerate(semesters):
+        example_data = sample_cntt_example if "CNTT" in sample_option else sample_kinhte_example if "Kinh tế" in sample_option else None
+        default_values = example_data["semester_data"][idx*9:(idx+1)*9] if example_data else None
+        data += input_semester(sem, default_values)
+    nganh = 0 if major == "Công nghệ thông tin" else 1
+    final_features = np.array(data + [nganh]).reshape(1, -1)
+else:
+    st.subheader("✨ Nhập thông tin rút gọn")
+    for idx, sem in enumerate(semesters):
+        example_data = sample_cntt_example if "CNTT" in sample_option else sample_kinhte_example if "Kinh tế" in sample_option else None
+        raw = example_data["semester_data"][idx*9:(idx+1)*9] if example_data else None
+        default_values = [raw[4], raw[2], raw[7]] if raw else None
+        data += input_important_features(sem, default_values)
+    final_features = np.array(data).reshape(1, -1)
+
+# ===== Predict =====
 if st.button("🎯 DỰ BÁO"):
-    prediction = model.predict(final_features)
+    if model_type == "Dùng toàn bộ dữ liệu":
+        prediction = model_full.predict(final_features)
+    else:
+        prediction = model_important.predict(final_features)
 
     if prediction[0] == 1:
         st.success(f"🎉 Chúc mừng bạn {name} - {major}! Bạn có khả năng tốt nghiệp đúng hạn!")
         st.balloons()
     else:
         st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ hạn.")
-        st.snow()
-
-
+        st.snow()