Update app.py

app.py

@@ -37,88 +37,192 @@ st.markdown("<h1 style='text-align: center; color: #003366;'>🎓 DỰ ĐOÁN 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("---")
 
-# ===== SIDEBAR: chọn mô hình =====
-model_type = st.sidebar.selectbox("🧠 Chọn mô hình dự báo:", ["Dùng toàn bộ đặc trưng", "Dùng đặc trưng quan trọng"])
 
-# ===== Giao diện nhập ví dụ =====
-st.subheader("🔢 Chọn ví dụ mẫu hoặc nhập thông tin cá nhân")
+# ===========================
+#     BỘ 8 VÍ DỤ
+# ===========================
+
+sample_options = [
+    "Không ví dụ",
+
+    # FULL DATA
+    "CNTT (Full) – Đúng hạn",
+    "CNTT (Full) – Không đúng hạn",
+    "Kinh tế (Full) – Đúng hạn",
+    "Kinh tế (Full) – Không đúng hạn",
+
+    # SIMPLE DATA
+    "CNTT (Simple) – Đúng hạn",
+    "CNTT (Simple) – Không đúng hạn",
+    "Kinh tế (Simple) – Đúng hạn",
+    "Kinh tế (Simple) – Không đúng hạn",
+]
 
-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ế"])
+sample_option = st.selectbox("📝 Chọn ví dụ:", sample_options)
 
-# ===== Ví dụ mẫu =====
-sample_cntt_example_full = {
+
+# ===== FULL DATA EXAMPLES =====
+sample_cntt_full_ok = {
     "name": "Nguyễn Văn A",
-    "student_id": "10117367",
+    "student_id": "10117001",
     "major": "Công nghệ thông tin",
     "semester_data": [
-        0, 0, 0, 0, 17, 7, 7, 8.73, 6,
-        0, 0, 0, 0, 17, 7, 7, 8.19, 5,
-        0, 0, 0, 0, 17, 7, 7, 7.90, 5,
-        0, 0, 0, 0, 17, 7, 7, 8.19, 6,
-        0, 0, 0, 0, 19, 7, 6, 8.18, 6,
-        0, 0, 5, 2, 19, 7, 5, 7.10, 5
+        0,0,0,0,17,7,7,8.5,6,
+        0,0,0,0,17,7,7,8.3,5,
+        0,0,0,0,17,7,7,8.0,5,
+        0,0,0,0,17,7,6,8.2,6,
+        0,0,0,0,19,7,6,8.1,6,
+        0,0,3,1,19,7,6,7.9,5
     ]
 }
 
-sample_cntt_example_simple = {
+sample_cntt_full_fail = {
     "name": "Nguyễn Văn B",
-    "student_id": "10117368",
+    "student_id": "10117002",
     "major": "Công nghệ thông tin",
     "semester_data": [
-        15, 8, 6.50,
-        17, 15, 6.10,
-        18, 7, 5.90,
-        17, 7, 5.80,
-        19, 8, 6.30,
-        19, 10, 6.20
+        1,3,5,2,15,6,4,5.5,2,
+        2,3,6,3,16,6,3,5.2,2,
+        2,4,7,3,17,7,3,4.9,1,
+        3,4,8,4,17,7,3,4.8,1,
+        3,5,9,4,18,7,3,4.7,1,
+        4,6,10,5,18,7,2,4.5,0
     ]
 }
 
-sample_kinhte_example_simple = {
-    "name": "Trần Thị B",
-    "student_id": "11418092",
+sample_kinhte_full_ok = {
+    "name": "Trần Thị C",
+    "student_id": "11418001",
     "major": "Kinh tế",
     "semester_data": [
-        17, 0, 7.64,
-        17, 0, 7.19,
-        17, 2, 6.94,
-        17, 0, 7.19,
-        19, 0, 7.65,
-        19, 0, 7.49
+        0,0,0,0,16,6,6,7.0,5,
+        0,0,0,0,18,7,7,8.2,6,
+        0,0,0,0,17,7,7,7.8,5,
+        0,0,0,0,17,7,7,8.0,6,
+        0,0,0,0,19,7,7,8.3,6,
+        0,0,0,0,19,7,7,8.1,6
     ]
 }
 
-sample_kinhte_example_full = {
-    "name": "Trần Thị C",
-    "student_id": "11418093",
+sample_kinhte_full_fail = {
+    "name": "Trần Thị D",
+    "student_id": "11418002",
+    "major": "Kinh tế",
+    "semester_data": [
+        1,2,3,2,15,6,4,5.6,2,
+        1,3,4,3,15,6,4,5.3,2,
+        2,4,5,3,16,7,4,5.2,2,
+        2,5,6,3,17,7,4,5.0,1,
+        3,6,7,4,17,7,3,4.9,1,
+        4,7,8,5,18,7,3,4.7,1
+    ]
+}
+
+# ===== SIMPLE DATA EXAMPLES =====
+sample_cntt_simple_ok = {
+    "name": "Nguyễn Văn E",
+    "student_id": "10118001",
+    "major": "Công nghệ thông tin",
+    "semester_data": [
+        17,0,7.2,
+        17,0,7.0,
+        18,0,6.8,
+        18,0,6.9,
+        19,0,7.1,
+        19,0,7.0
+    ]
+}
+
+sample_cntt_simple_fail = {
+    "name": "Nguyễn Văn F",
+    "student_id": "10118002",
+    "major": "Công nghệ thông tin",
+    "semester_data": [
+        12,10,5.2,
+        12,12,5.0,
+        13,14,4.8,
+        14,13,4.7,
+        15,12,4.9,
+        15,13,4.6
+    ]
+}
+
+sample_kinhte_simple_ok = {
+    "name": "Trần Thị G",
+    "student_id": "11418003",
     "major": "Kinh tế",
     "semester_data": [
-        0, 0, 0, 0, 16, 6, 6, 7.00, 5,
-        0, 0, 0, 0, 18, 7, 7, 8.20, 6,
-        0, 0, 0, 0, 17, 7, 7, 7.80, 5,
-        0, 0, 0, 0, 17, 7, 6, 7.90, 5,
-        0, 0, 0, 0, 19, 8, 5, 8.10, 6,
-        0, 0, 8, 2, 19, 7, 6, 7.30, 5
+        17,0,7.5,
+        17,0,7.2,
+        17,0,7.0,
+        18,0,7.3,
+        18,0,7.4,
+        19,0,7.5
     ]
 }
 
-# ===== 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_full["name"])
-    student_id = st.text_input("🎓 Mã sinh viên", value=sample_cntt_example_full["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_simple["name"])
-    student_id = st.text_input("🎓 Mã sinh viên", value=sample_kinhte_example_simple["student_id"])
-    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=1)
+sample_kinhte_simple_fail = {
+    "name": "Trần Thị H",
+    "student_id": "11418004",
+    "major": "Kinh tế",
+    "semester_data": [
+        12,10,5.3,
+        12,11,5.0,
+        13,12,4.9,
+        13,13,4.8,
+        14,12,4.7,
+        14,13,4.5
+    ]
+}
+
+
+# ==============================
+#     LOAD EXAMPLE SELECTED
+# ==============================
+
+example_data = None
+
+if sample_option == "CNTT (Full) – Đúng hạn":
+    example_data = sample_cntt_full_ok
+elif sample_option == "CNTT (Full) – Không đúng hạn":
+    example_data = sample_cntt_full_fail
+elif sample_option == "Kinh tế (Full) – Đúng hạn":
+    example_data = sample_kinhte_full_ok
+elif sample_option == "Kinh tế (Full) – Không đúng hạn":
+    example_data = sample_kinhte_full_fail
+elif sample_option == "CNTT (Simple) – Đúng hạn":
+    example_data = sample_cntt_simple_ok
+elif sample_option == "CNTT (Simple) – Không đúng hạn":
+    example_data = sample_cntt_simple_fail
+elif sample_option == "Kinh tế (Simple) – Đúng hạn":
+    example_data = sample_kinhte_simple_ok
+elif sample_option == "Kinh tế (Simple) – Không đúng hạn":
+    example_data = sample_kinhte_simple_fail
+
+
+
+# ===============================
+#     NHẬP THÔNG TIN CÁ NHÂN
+# ===============================
+
+if example_data:
+    name = st.text_input("👤 Họ và tên", value=example_data["name"])
+    student_id = st.text_input("🎓 Mã sinh viên", value=example_data["student_id"])
+    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"],
+                         index=0 if example_data["major"] == "Công nghệ thông tin" else 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("---")
 
-# ===== Nhập thông tin học kỳ =====
+
+# ===============================
+#          INPUT SEMESTER
+# ===============================
+
 def input_semester(semester_label, default_values=None):
     with st.expander(f"📖 {semester_label}", expanded=True):
         col1, col2 = st.columns(2)
@@ -127,78 +231,107 @@ def input_semester(semester_label, default_values=None):
             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}")
+
             try:
                 default_tc_qua = default_values[4] - default_values[2] if default_values else 0
             except:
                 default_tc_qua = 0
+
             sotc_qua = st.number_input("Số tín chỉ qua môn", 0, value=default_tc_qua, key=f"tcqua_{semester_label}")
+
         with col2:
             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}")
+            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]
+            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
+    ]
 
 
 def input_important_features(semester_label, default_values=None):
     with st.expander(f"📘 {semester_label}", expanded=True):
         col1, col2 = st.columns(2)
+
         with col1:
-            sotc_qua = st.number_input("Số tín chỉ đạt", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{semester_label}")
-            sotcno = st.number_input("Số tín chỉ nợ", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{semester_label}")
+            sotc_qua = st.number_input("Số tín chỉ đạt", 0,
+                                       value=default_values[0] if default_values else 0,
+                                       key=f"tcqua_imp_{semester_label}")
+
+            sotcno = st.number_input("Số tín chỉ nợ", 0,
+                                     value=default_values[1] if default_values else 0,
+                                     key=f"tcno_imp_{semester_label}")
+
         with col2:
-            TBCHK = st.number_input("Điểm trung bình", 0.0, 10.0, value=default_values[2] if default_values else 0.0, step=0.01, key=f"gpa_imp_{semester_label}")
+            TBCHK = st.number_input("Điểm trung bình", 0.0, 10.0,
+                                    value=default_values[2] if default_values else 0.0,
+                                    step=0.01, key=f"gpa_imp_{semester_label}")
+
     return [sotc_qua, sotcno, TBCHK]
 
-# ===== Giao diện theo mô hình =====
+
+# ===============================
+#       GENERATE FEATURES
+# ===============================
+
 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"]
 
+model_type = st.sidebar.selectbox("🧠 Chọn mô hình dự báo:", ["Dùng toàn bộ đặc trưng", "Dùng đặc trưng quan trọng"])
+
 if model_type == "Dùng toàn bộ đặc trưng":
-    st.subheader("🔢 Nhập thông tin học kỳ chi tiết")
+    st.subheader("🔢 Nhập thông tin học kỳ (Full Data)")
+
     for idx, sem in enumerate(semesters):
-        example_data = sample_cntt_example_full if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin" else sample_kinhte_example_full if sample_option == "Dùng ví dụ mẫu ngành Kinh tế" else None
-        default_values = example_data["semester_data"][idx*9:(idx+1)*9] if example_data else None
+        default_values = None
+        if example_data and len(example_data["semester_data"]) == 54:
+            default_values = example_data["semester_data"][idx*9:(idx+1)*9]
+
         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_simple if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin" else sample_kinhte_example_simple if sample_option == "Dùng ví dụ mẫu ngành Kinh tế" else None
-        default_values = example_data["semester_data"][idx*3:(idx+1)*3] if example_data else None
+    st.subheader("✨ Nhập thông tin rút gọn (Simple Data)")
 
-        with st.expander(f"📘 {sem}", expanded=True):
-            col1, col2, col3 = st.columns(3)
-            with col1:
-                sotc_qua = st.number_input(f"Số tín chỉ đạt - {sem}", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{sem}")
-            with col2:
-                sotcno = st.number_input(f"Số tín chỉ nợ - {sem}", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{sem}")
-            with col3:
-                TBCHK = st.number_input(f"Điểm trung bình - {sem}", 0.0, 10.0, value=default_values[2] if default_values else 0.0, step=0.01, key=f"gpa_imp_{sem}")
+    for idx, sem in enumerate(semesters):
+        default_values = None
+        if example_data and len(example_data["semester_data"]) == 18:
+            default_values = example_data["semester_data"][idx*3:(idx+1)*3]
 
-        data += [sotc_qua, sotcno, TBCHK]
+        data_inputs = input_important_features(sem, default_values)
+        data += data_inputs
 
     final_features = np.array(data).reshape(1, -1)
 
-# ===== Predict =====
+
+# ===============================
+#         PREDICT BUTTON
+# ===============================
+
 if st.button("🎯 DỰ BÁO"):
     if model_type == "Dùng toàn bộ đặc trưng":
         prediction = model_full.predict(final_features)
+
         if prediction[0] == 0:
-            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.success(f"🎉 Chúc mừng bạn {name}! 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.error(f"⚠️ Bạn {name} có nguy cơ trễ hạn. Hãy cố gắng cải thiện nhé!")
             st.snow()
 
     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.success(f"🎉 Chúc mừng bạn {name}! 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.error(f"⚠️ Bạn {name} có nguy cơ trễ hạn. Cần cố gắng hơn!")
             st.snow()