Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import numpy as np
 import joblib
+import json
 
 # ===== Load models =====
 model_full = joblib.load("random_forest_model_full.pkl")
@@ -16,7 +17,7 @@ st.markdown("""
         font-family: 'Poppins', sans-serif;
     }
     .block-container {
-        padding: 2rem;
+        padding: 2rem 2rem 2rem 2rem;
     }
     .stButton button {
         background: linear-gradient(90deg, #4F8BF9 0%, #8E2DE2 100%);
@@ -36,55 +37,40 @@ 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"
-])
+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")
 
+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ế"])
 
-# ===== Ví dụ mẫu cho FULL MODEL (9 feature/kỳ) =====
+# ===== Ví dụ mẫu =====
 sample_cntt_example_full = {
     "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,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,0,0,19,7,5,7.10,5
+        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
     ]
 }
 
-sample_kinhte_example_full = {
-    "name": "Trần Thị C",
-    "student_id": "11418093",
-    "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
-    ]
-}
-
-# ===== Ví dụ cho SIMPLE MODEL (3 feature/kỳ) =====
 sample_cntt_example_simple = {
     "name": "Nguyễn Văn B",
     "student_id": "10117368",
     "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
+        15, 8, 6.50,
+        17, 15, 6.10,
+        18, 7, 5.90,
+        17, 7, 5.80,
+        19, 8, 6.30,
+        19, 10, 6.20
     ]
 }
 
@@ -93,113 +79,126 @@ sample_kinhte_example_simple = {
     "student_id": "11418092",
     "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
+        17, 0, 7.64,
+        17, 0, 7.19,
+        17, 2, 6.94,
+        17, 0, 7.19,
+        19, 0, 7.65,
+        19, 0, 7.49
     ]
 }
 
-
-# ===== CHỌN VÍ DỤ =====
-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_kinhte_example_full = {
+    "name": "Trần Thị C",
+    "student_id": "11418093",
+    "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
+    ]
+}
 
 # ===== Thông tin cá nhân =====
 if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin":
-    example = sample_cntt_example_full if model_type == "Dùng toàn bộ đặc trưng" else sample_cntt_example_simple
+    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ế":
-    example = sample_kinhte_example_full if model_type == "Dùng toàn bộ đặc trưng" else sample_kinhte_example_simple
+    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)
 else:
-    example = None
-
-name = st.text_input("👤 Họ và tên", value=example["name"] if example else "")
-student_id = st.text_input("🎓 Mã sinh viên", value=example["student_id"] if example else "")
-major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"],
-                     index=0 if not example else (0 if example["major"] == "Công nghệ thông tin" else 1))
+    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 DỮ LIỆU =====
-def input_full_semester(label, default):
-    with st.expander(label, expanded=True):
-        results = []
-        fields = [
-            "Số môn không thi", "Số tín chỉ không thi", "Số tín chỉ nợ", "Số môn không đạt",
-            "Tổng tín chỉ học kỳ", "Số môn học kỳ", "Số môn đạt", "GPA", "Xếp loại"
-        ]
-        
-        col1, col2, col3 = st.columns(3)
-        for i, field in enumerate(fields):
-            value = default[i] if default else 0
-            with (col1 if i < 3 else col2 if i < 6 else col3):
-                if i == 7:
-                    value = st.number_input(f"{field}", 0.0, 10.0, value=float(value), step=0.01)
-                elif i == 8:
-                    value = st.selectbox(f"{field}", list(range(7)), index=value)
-                else:
-                    value = st.number_input(f"{field}", 0, value=value)
-            results.append(value)
-
-    return results
-
-
-def input_simple_semester(label, default):
-    with st.expander(label, expanded=True):
-        col1, col2, col3 = st.columns(3)
-        sotc_qua = col1.number_input(f"Số tín chỉ đạt - {label}", 0, value=default[0] if default else 0)
-        sotc_no = col2.number_input(f"Số tín chỉ nợ - {label}", 0, value=default[1] if default else 0)
-        gpa = col3.number_input(f"GPA - {label}", 0.0, 10.0, value=default[2] if default else 0.0, step=0.01)
-    return [sotc_qua, sotc_no, gpa]
-
-
-# ===== NHẬP DỮ LIỆU =====
-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"]
+# ===== Nhập thông tin học kỳ =====
+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 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ộ đặc trưng":
-    st.subheader("🔢 Nhập thông tin đầy đủ 6 học kỳ")
-
-    for i, sem in enumerate(semesters):
-        default_values = example["semester_data"][i*9:(i+1)*9] if example else None
-        data += input_full_semester(sem, default_values)
-
-    final_features = np.array(data).reshape(1, -1)
+    st.subheader("🔢 Nhập thông tin học kỳ chi tiết")
+    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
+        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
 
-    for i, sem in enumerate(semesters):
-        default_values = example["semester_data"][i*3:(i+1)*3] if example else None
-        data += input_simple_semester(sem, default_values)
+        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}")
 
-    final_features = np.array(data).reshape(1, -1)
+        data += [sotc_qua, sotcno, TBCHK]
 
+    final_features = np.array(data).reshape(1, -1)
 
-# ===== PREDICT =====
+# ===== Predict =====
 if st.button("🎯 DỰ BÁO"):
     if model_type == "Dùng toàn bộ đặc trưng":
-        prediction = model_full.predict(final_features)[0]
-        
-        if prediction == 0:   # 0 = tốt nghiệp đúng hạn (theo mô hình của bạn)
-            st.success(f"🎉 Chúc mừng {name}! Bạn có khả năng tốt nghiệp đúng hạn!")
+        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.balloons()
         else:
-            st.error(f"⚠️ Bạn có nguy cơ trễ hạn, cần cố gắng hơn!")
+            st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ hạn.")
             st.snow()
 
     else:
-        prediction = model_important.predict(final_features)[0]
-
-        if prediction == 1:   # 1 = đúng hạn trong simple model
-            st.success(f"🎉 Chúc mừng {name}! Bạn có khả năng tốt nghiệp đúng hạn!")
+        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 có nguy cơ trễ hạn, cần cố gắng hơn!")
+            st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ hạn.")
             st.snow()