Update app.py

app.py

@@ -72,7 +72,114 @@ model_type = st.sidebar.selectbox("🧠 Chọn mô hình dự báo:", ["Dùng to
 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 =====
-# (phần còn lại giữ nguyên như trước đây)
+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
+    ]
+}
+
+sample_kinhte_example = {
+    "name": "Trần Thị B",
+    "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
+    ]
+}
+
+# ===== 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("---")
+
+# ===== 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", 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}")
+            try:
+                default_tc_qua = default_values[4] - default_values[2] if default_values and len(default_values) >= 5 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}")
+            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]
+
+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}")
+        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}")
+    return [sotc_qua, sotcno, TBCHK]
+
+# ===== 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"]
+
+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 = None
+        if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin":
+            example_data = sample_cntt_example
+        elif sample_option == "Dùng ví dụ mẫu ngành Kinh tế":
+            example_data = sample_kinhte_example
+
+        if example_data:
+            raw = example_data["semester_data"][idx*9:(idx+1)*9]
+            if len(raw) >= 8:
+                default_values = [raw[4], raw[2], raw[7]]
+            else:
+                default_values = None
+        else:
+            default_values = None
+
+        data += input_important_features(sem, default_values)
+    final_features = np.array(data).reshape(1, -1)
 
 # ===== Predict =====
 if st.button("🎯 DỰ BÁO"):