Update app.py

app.py

@@ -31,27 +31,20 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-
 # ===== HEADER =====
 st.markdown("<h1 style='text-align: center; color: #003366;'>🎓 DỰ ĐOÁN 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("---")
 
-
 # ===========================
 #     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",
@@ -60,7 +53,6 @@ sample_options = [
 
 sample_option = st.selectbox("📝 Chọn ví dụ:", sample_options)
 
-
 # ===== FULL DATA EXAMPLES =====
 sample_cntt_full_ok = {
     "name": "Nguyễn Văn A",
@@ -175,13 +167,10 @@ sample_kinhte_simple_fail = {
     ]
 }
 
-
 # ==============================
-#     LOAD EXAMPLE SELECTED
+#     LOAD SELECTED EXAMPLE
 # ==============================
-
 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":
@@ -199,51 +188,73 @@ elif sample_option == "Kinh tế (Simple) – Đúng hạn":
 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
+#     PERSONAL INFO
 # ===============================
-
 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ế"],
+    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("---")
 
-
 # ===============================
-#          INPUT SEMESTER
+#     INPUT FUNCTIONS
 # ===============================
-
 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}")
+            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 else 0
+                default_tc_qua = default_values[4] - default_values[2]
             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}")
+            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}")
+            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}")
@@ -253,7 +264,6 @@ def input_semester(semester_label, default_values=None):
         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)
@@ -270,19 +280,19 @@ def input_important_features(semester_label, default_values=None):
         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}")
+                                    step=0.01,
+                                    key=f"gpa_imp_{semester_label}")
 
     return [sotc_qua, sotcno, TBCHK]
 
-
 # ===============================
-#       GENERATE FEATURES
+#     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"])
+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ỳ (Full Data)")
@@ -310,11 +320,9 @@ else:
 
     final_features = np.array(data).reshape(1, -1)
 
-
 # ===============================
 #         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)