Update app.py

app.py

@@ -7,155 +7,262 @@ from datetime import datetime
 import json
 
 # ===== Load models =====
-model_full = joblib.load("random_forest_model_full.pkl")
-model_important = joblib.load("random_forest_model_importainfeature.pkl")
+
+# model_full = joblib.load("random_forest_model_full.pkl")
+# model_important = joblib.load("random_forest_model_importainfeature.pkl")
+
+# Dummy models for demonstration if files are not present
+try:
+    model_full = joblib.load("random_forest_model_full.pkl")
+except FileNotFoundError:
+    st.warning("Model file random_forest_model_full.pkl not found. Using dummy model.")
+    class DummyModel:
+        def predict(self, features):
+            # Simulate a prediction (e.g., always predict 1)
+            # You might want to return a random prediction for more realism
+            return np.array([1])
+    model_full = DummyModel()
+
+try:
+    model_important = joblib.load("random_forest_model_importainfeature.pkl")
+except FileNotFoundError:
+    st.warning("Model file random_forest_model_importainfeature.pkl not found. Using dummy model.")
+    class DummyModel:
+        def predict(self, features):
+            # Simulate a prediction (e.g., always predict 1)
+            # You might want to return a random prediction for more realism
+            return np.array([1])
+    model_important = DummyModel()
+
 
 # ===== Setup page =====
+
 st.set_page_config(page_title="Dự đoán tốt nghiệp đúng hạn", page_icon="🎓", layout="wide")
 
 # ===== Custom CSS =====
-st.markdown("""
-    <style>
-    html, body, [class*="css"]  {
-        font-family: 'Poppins', sans-serif;
-    }
-    .block-container {
-        padding: 2rem 2rem 2rem 2rem;
-    }
-    .stButton button {
-        background: linear-gradient(90deg, #4F8BF9 0%, #8E2DE2 100%);
-        border: none;
-        padding: 0.75em 1.5em;
-        border-radius: 8px;
-        color: white;
-        font-size: 18px;
-        font-weight: bold;
-    }
-    /* Make number_input and selectbox shorter */
-    .stNumberInput input, .stSelectbox select {
-        width: 15px;  /* Set the width to a smaller value */
-    }
-    </style>
-""", unsafe_allow_html=True)
 
+st.markdown(""" <style>
+html, body, [class*="css"]  {
+    font-family: 'Poppins', sans-serif;
+}
+.block-container {
+    padding: 2rem 2rem 2rem 2rem;
+}
+.stButton button {
+    background: linear-gradient(90deg, #4F8BF9 0%, #8E2DE2 100%);
+    border: none;
+    padding: 0.75em 1.5em;
+    border-radius: 8px;
+    color: white;
+    font-size: 18px;
+    font-weight: bold;
+}
+
+/* CSS để làm ngắn st.number_input và st.selectbox */
+.stNumberInput, .stSelectbox {
+    width: 180px;  /* Điều chỉnh giá trị này (pixel) để thay đổi độ rộng mong muốn */
+    /* Có thể dùng max-width thay cho width nếu muốn giới hạn độ rộng tối đa */
+    /* max-width: 200px; */
+}
+
+/* Tùy chọn: Đảm bảo input và select bên trong lấp đầy chiều rộng container */
+.stNumberInput input, .stSelectbox select {
+    width: 100%;
+    box-sizing: border-box; /* Đảm bảo padding và border không làm tràn */
+}
+
+/* Điều chỉnh thêm khoảng cách giữa label và input nếu cần */
+.stNumberInput label, .stSelectbox label {
+    margin-bottom: 0.2rem;
+}
+
+
+ </style>
+""", unsafe_allow_html=True)
 
 # ===== GSheet integration =====
+
+# Make sure credentials.json is present or handle its absence
+try:
+    with open("credentials.json") as f:
+        gsheet_credentials = json.load(f)
+except FileNotFoundError:
+    st.error("Google Sheets credentials.json not found. Sheet logging disabled.")
+    gsheet_credentials = None
+
 @st.cache_resource
-def get_gsheet_client():
-    scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
-    creds = ServiceAccountCredentials.from_json_keyfile_name("credentials.json", scope)
-    client = gspread.authorize(creds)
-    return client
+def get_gsheet_client(creds_json):
+    if creds_json is None:
+        return None
+    try:
+        scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+        creds = ServiceAccountCredentials.from_json_keyfile_dict(creds_json, scope)
+        client = gspread.authorize(creds)
+        return client
+    except Exception as e:
+        st.error(f"Error connecting to Google Sheets: {e}")
+        return None
+
 
 # ===== Save to Google Sheets =====
+
 def save_to_gsheet(name, student_id, major, prediction, semester_data, sheet_name="Trang tính1"):
+    client = get_gsheet_client(gsheet_credentials)
+    if client is None:
+        return # Do not proceed if client is not available
+
     # Get the current date and time
     now = datetime.now()
     current_time = now.strftime("%Y-%m-%d %H:%M:%S")
-    
+
     # Ensure prediction is in the expected format (0 or 1)
     if isinstance(prediction, np.ndarray):
         prediction = int(prediction[0])  # Convert to int if it's ndarray
-    
     elif isinstance(prediction, np.int64):
         prediction = int(prediction)  # Convert np.int64 to int
-    
+    elif not isinstance(prediction, int):
+         # Handle cases where prediction might be a list or other format
+         try:
+             prediction = int(prediction[0]) if isinstance(prediction, list) else int(prediction)
+         except:
+             prediction = -1 # Indicate error or unknown prediction format
+
+
     # Flatten semester data to individual columns for each semester
-    semester_data_flat = [str(val) for val in semester_data]  # Convert all semester data values to strings
-    
+    # Ensure all data types are suitable for GSheets (strings or numbers)
+    semester_data_flat = [str(val) if val is not None else "" for val in semester_data]
+
+
     # Create a data row to insert into the sheet
     data_row = [name, student_id, major, prediction, current_time] + semester_data_flat
 
-    # Connect to Google Sheets
-    client = get_gsheet_client()
-
-    # Use the Spreadsheet ID (replace with your actual ID)
-    sheet_id = "1i7bDNvLVLXN93_e-FN0JLzpg1jb64Z_aEuyPjIfwbdQ"  # Use your actual ID here
-    sheet = client.open_by_key(sheet_id).worksheet(sheet_name)
-
-    # Get all values from the sheet to check if the first row is empty (i.e., headers not created yet)
-    all_values = sheet.get_all_values()
-
-    # If the sheet is empty or if headers are missing, insert headers
-    if not all_values or len(all_values[0]) != (len(semester_data_flat) + 5):  # Ensure number of columns matches
-        headers = ["Họ và tên", "MSV", "Khoa", "Dự báo", "Thời gian"] + [
-            f"Số môn không thi - HK{i+1}" for i in range(6)] + [
-            f"Số tín chỉ không thi - HK{i+1}" for i in range(6)] + [
-            f"Số tín chỉ nợ - HK{i+1}" for i in range(6)] + [
-            f"Số môn không đạt - HK{i+1}" for i in range(6)] + [
-            f"Số tín chỉ qua môn - HK{i+1}" for i in range(6)] + [
-            f"Tổng tín chỉ học kỳ - HK{i+1}" for i in range(6)] + [
-            f"Số môn học kỳ - HK{i+1}" for i in range(6)] + [
-            f"Số môn đạt - HK{i+1}" for i in range(6)] + [
-            f"GPA - HK{i+1}" for i in range(6)] + [
-            f"Xếp loại - HK{i+1}" for i in range(6)]
-        
-        sheet.append_row(headers)  # Create headers if missing
-    
-    # Append data to the sheet
-    sheet.append_row(data_row)
-
-
-# ===== List all sheet names =====
-# def list_sheet_names(sheet_id):
-#     client = get_gsheet_client()
-#     sheet = client.open_by_key(sheet_id)
-    
-#     # List all sheet names
-#     sheet_names = [worksheet.title for worksheet in sheet.worksheets()]
-#     return sheet_names
-
+    try:
+        # Use the Spreadsheet ID (replace with your actual ID)
+        sheet_id = "1i7bDNvLVLXN93_e-FN0JLzpg1jb64Z_aEuyPjIfwbdQ"  # Use your actual ID here
+        sheet = client.open_by_key(sheet_id).worksheet(sheet_name)
+
+        # Get all values from the sheet to check if the first row is empty (i.e., headers not created yet)
+        # Be cautious with get_all_values on very large sheets
+        all_values = sheet.get_all_values()
+
+        # Define headers based on the model type and expected data structure
+        if sheet_name == "Trang tính1": # Full model
+             headers = ["Họ và tên", "MSV", "Khoa", "Dự báo", "Thời gian"] + [
+                f"Số môn không thi - HK{i+1}" for i in range(6)] + [
+                f"Số tín chỉ không thi - HK{i+1}" for i in range(6)] + [
+                f"Số tín chỉ nợ - HK{i+1}" for i in range(6)] + [
+                f"Số môn không đạt - HK{i+1}" for i in range(6)] + [
+                f"Tổng tín chỉ học kỳ - HK{i+1}" for i in range(6)] + [ # TCHK
+                f"Số môn học kỳ - HK{i+1}" for i in range(6)] + [     # SMHK
+                f"Số môn đạt - HK{i+1}" for i in range(6)] + [        # MH Pass
+                f"GPA - HK{i+1}" for i in range(6)] + [
+                f"Xếp loại - HK{i+1}" for i in range(6)] + [
+                 f"Số tín chỉ qua môn (calc) - HK{i+1}" for i in range(6)] # sotc_qua
+
+        elif sheet_name == "Trang tính2": # Important features model
+             headers = ["Họ và tên", "MSV", "Khoa", "Dự báo", "Thời gian"] + [
+                f"Số tín chỉ đạt - HK{i+1}" for i in range(6)] + [ # sotc_qua
+                f"Số tín chỉ nợ - HK{i+1}" for i in range(6)] + [ # sotcno
+                f"GPA - HK{i+1}" for i in range(6)] # TBCHK
+        else: # Default or unknown sheet
+             headers = ["Họ và tên", "MSV", "Khoa", "Dự báo", "Thời gian"] + [f"Data_{i+1}" for i in range(len(semester_data_flat))]
+
+
+        # If the sheet is empty or if headers are missing/mismatched column count, insert headers
+        # Check column count + 5 base columns (Name, ID, Major, Pred, Time)
+        expected_cols = len(semester_data_flat) + 5
+        if not all_values or len(all_values[0]) != expected_cols or all_values[0][:5] != headers[:5]:
+            sheet.append_row(headers)  # Create headers if missing or wrong
+
+
+        # Append data to the sheet
+        sheet.append_row(data_row)
+        # st.success(f"Đã lưu kết quả dự báo vào Google Sheet '{sheet_name}'.") # Optional confirmation
+    except Exception as e:
+         st.error(f"Lỗi khi lưu dữ liệu vào Google Sheet: {e}")
 
 
 # ===== 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("---")
 
 # ===== 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")
 
-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ế"])
+# Điều chỉnh options cho phù hợp với các ví dụ có sẵn
+sample_options = ["Không ví dụ"]
+if model_type == "Dùng toàn bộ đặc trưng":
+     sample_options.append("Dùng ví dụ mẫu CNTT (toàn bộ)")
+     sample_options.append("Dùng ví dụ mẫu Kinh tế (toàn bộ)")
+else: # Dùng đặc trưng quan trọng
+     sample_options.append("Dùng ví dụ mẫu CNTT (rút gọn)")
+     sample_options.append("Dùng ví dụ mẫu Kinh tế (rút gọn)")
+
+
+sample_option = st.selectbox("📝 Chọn ví dụ:", sample_options)
+
+# ===== Ví dụ mẫu cho mô hình "Dùng toàn bộ đặc trưng" =====
 
-# ===== Ví dụ mẫu cho mô hình "Dùng toàn bộ dữ liệu" =====
 sample_cntt_example_full = {
-    "name": "Nguyễn Văn A",
+    "name": "Nguyễn Văn A (Mẫu Full)",
     "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, # HK1: somon0thi, sotc0thi, sotcno, mhno, TCHK, smhk, mhpass, TBCHK, xep_loai, sotc_qua (added later)
+        0, 0, 0, 0, 17, 7, 7, 8.19, 0, # HK2
+        0, 0, 0, 0, 17, 7, 7, 7.90, 0, # HK3
+        0, 0, 0, 0, 17, 7, 7, 8.19, 0, # HK4
+        0, 0, 0, 0, 19, 7, 6, 8.18, 0, # HK5
+        0, 0, 5, 2, 19, 7, 5, 7.10, 1  # HK6
     ]
 }
+# Thêm sotc_qua cho ví dụ Full (TCHK - sotcno)
+for i in range(6):
+    tchk_idx = i * 9 + 4
+    sotcno_idx = i * 9 + 2
+    sotc_qua_val = sample_cntt_example_full["semester_data"][tchk_idx] - sample_cntt_example_full["semester_data"][sotcno_idx]
+    # Insert sotc_qua at the end of each semester's data in the list
+    sample_cntt_example_full["semester_data"].insert((i+1)*9 + i, sotc_qua_val) # Insert at (idx+1)*9 + current insertion count
+
+# Re-calculate indices based on new structure
+# After inserting 6 values, the length becomes 9*6 + 6 = 60
+# HK1: 0-9, HK2: 10-19, ..., HK6: 50-59
+# Each block is 10 values now: somon0thi, sotc0thi, sotcno, mhno, TCHK, smhk, mhpass, TBCHK, xep_loai, sotc_qua
 
-# ===== Ví dụ mẫu cho mô hình "Dùng mô hình đơn giản" =====
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho Kinh tế =====
-sample_kinhte_example_simple = {
-    "name": "Trần Thị B",
-    "student_id": "11418092",
+sample_kinhte_example_full = {
+    "name": "Trần Thị C (Mẫu Full)",
+    "student_id": "11418093",
     "major": "Kinh tế",
     "semester_data": [
-        17, 7, 7.64,  # Học kỳ 1: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 7, 7.19,  # Học kỳ 2: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 7, 6.94,  # Học kỳ 3: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 7, 7.19,  # Học kỳ 4: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        19, 7, 7.65,  # Học kỳ 5: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        19, 7, 7.49   # Học kỳ 6: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        0, 0, 0, 0, 16, 6, 6, 7.00, 1, # HK1
+        0, 0, 0, 0, 18, 7, 7, 8.20, 1, # HK2
+        0, 0, 0, 0, 17, 7, 7, 7.80, 1, # HK3
+        0, 0, 0, 0, 17, 7, 6, 7.90, 1, # HK4
+        0, 0, 0, 0, 19, 8, 5, 8.10, 0, # HK5
+        0, 0, 8, 2, 19, 7, 6, 7.30, 1  # HK6
     ]
 }
 
+# Thêm sotc_qua cho ví dụ Full Kinh tế (TCHK - sotcno)
+for i in range(6):
+    tchk_idx = i * 9 + 4
+    sotcno_idx = i * 9 + 2
+    sotc_qua_val = sample_kinhte_example_full["semester_data"][tchk_idx] - sample_kinhte_example_full["semester_data"][sotcno_idx]
+     # Insert sotc_qua at the end of each semester's data in the list
+    sample_kinhte_example_full["semester_data"].insert((i+1)*9 + i, sotc_qua_val)
+
+
+# ===== Ví dụ mẫu cho mô hình "Dùng đặc trưng quan trọng" (Simplified) =====
 
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho CNTT =====
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho CNTT =====
 sample_cntt_example_simple = {
-    "name": "Nguyễn Văn B",
+    "name": "Nguyễn Văn B (Mẫu Rút gọn)",
     "student_id": "10117368",
     "major": "Công nghệ thông tin",
     "semester_data": [
@@ -168,118 +275,143 @@ sample_cntt_example_simple = {
     ]
 }
 
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho Kinh tế =====
-sample_kinhte_example_full = {
-    "name": "Trần Thị C",
-    "student_id": "11418093",
+sample_kinhte_example_simple = {
+    "name": "Trần Thị B (Mẫu Rút gọn)",
+    "student_id": "11418092",
     "major": "Kinh tế",
     "semester_data": [
-        0, 0, 0, 0, 16, 6, 6, 7.00, 1,
-        0, 0, 0, 0, 18, 7, 7, 8.20, 1,
-        0, 0, 0, 0, 17, 7, 7, 7.80, 1,
-        0, 0, 0, 0, 17, 7, 6, 7.90, 1,
-        0, 0, 0, 0, 19, 8, 5, 8.10, 0,
-        0, 0, 8, 2, 19, 7, 6, 7.30, 1
+        17, 7, 7.64,  # Học kỳ 1: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        17, 7, 7.19,  # Học kỳ 2: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        17, 7, 6.94,  # Học kỳ 3: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        17, 7, 7.19,  # Học kỳ 4: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        19, 7, 7.65,  # Học kỳ 5: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        19, 7, 7.49   # Học kỳ 6: Số tín chỉ đạt, Số tín chỉ nợ, GPA
     ]
 }
 
+
 # ===== 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)
-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ế"])
+
+# Chọn ví dụ mẫu phù hợp với mô hình đã chọn
+selected_example = None
+if sample_option == "Dùng ví dụ mẫu CNTT (toàn bộ)" and model_type == "Dùng toàn bộ đặc trưng":
+    selected_example = sample_cntt_example_full
+elif sample_option == "Dùng ví dụ mẫu Kinh tế (toàn bộ)" and model_type == "Dùng toàn bộ đặc trưng":
+    selected_example = sample_kinhte_example_full
+elif sample_option == "Dùng ví dụ mẫu CNTT (rút gọn)" and model_type == "Dùng đặc trưng quan trọng":
+     selected_example = sample_cntt_example_simple
+elif sample_option == "Dùng ví dụ mẫu Kinh tế (rút gọn)" and model_type == "Dùng đặc trưng quan trọng":
+     selected_example = sample_kinhte_example_simple
+
+
+default_name = selected_example["name"] if selected_example else ""
+default_student_id = selected_example["student_id"] if selected_example else ""
+default_major_index = 0 if selected_example and selected_example["major"] == "Công nghệ thông tin" else (1 if selected_example and selected_example["major"] == "Kinh tế" else 0)
+
+
+name = st.text_input("👤 Họ và tên", value=default_name)
+student_id = st.text_input("🎓 Mã sinh viên", value=default_student_id)
+major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=default_major_index)
+
 
 st.write("---")
 
 # ===== Nhập thông tin học kỳ =====
-def input_semester(semester_label, default_values=None):
+
+def input_semester_full(semester_label, default_values=None):
+    # Default values structure: [somon0thi, sotc0thi, sotcno, mhno, TCHK, smhk, mhpass, TBCHK, xep_loai, sotc_qua]
     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}")
+            # Use get() with a default value (0 or 0.0) to handle None or shorter lists
+            somon0thi = st.number_input("Số môn không thi", min_value=0, value=default_values[0] if default_values and len(default_values)>0 and default_values[0] is not None else 0, key=f"sm0_{semester_label}")
+            sotc0thi = st.number_input("Số tín chỉ không thi", min_value=0, value=default_values[1] if default_values and len(default_values)>1 and default_values[1] is not None else 0, key=f"tc0_{semester_label}")
+            sotcno = st.number_input("Số tín chỉ nợ", min_value=0, value=default_values[2] if default_values and len(default_values)>2 and default_values[2] is not None else 0, key=f"tcno_{semester_label}")
+            mhno = st.number_input("Số môn không đạt", min_value=0, value=default_values[3] if default_values and len(default_values)>3 and default_values[3] is not None else 0, key=f"mhno_{semester_label}")
+            sotc_qua = st.number_input("Số tín chỉ qua môn (TCHK - Số TC nợ)", min_value=0, value=default_values[9] if default_values and len(default_values)>9 and default_values[9] is not None else 0, key=f"tcqua_{semester_label}") # Use the pre-calculated or default
         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]
+            TCHK = st.number_input("Tổng tín chỉ học kỳ", min_value=0, value=default_values[4] if default_values and len(default_values)>4 and default_values[4] is not None else 0, key=f"tchk_{semester_label}")
+            smhk = st.number_input("Số môn học kỳ", min_value=0, value=default_values[5] if default_values and len(default_values)>5 and default_values[5] is not None 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 and len(default_values)>6 and default_values[6] is not None else 0, key=f"mhpass_{semester_label}")
+            TBCHK = st.number_input("GPA", min_value=0.0, max_value=10.0, value=float(default_values[7]) if default_values and len(default_values)>7 and default_values[7] is not None else 0.0, step=0.01, key=f"gpa_{semester_label}")
+            # Ensure the default index is valid for the selectbox options (0 to 6)
+            default_xeploai_index = int(default_values[8]) if default_values and len(default_values)>8 and default_values[8] is not None and 0 <= int(default_values[8]) <= 6 else 0
+            xep_loai_selected = st.selectbox("Xếp loại", list(range(7)), index=default_xeploai_index, key=f"xeploai_{semester_label}")
+
+    # Return data in the order expected by the full model (9 features per semester + sotc_qua)
+    # [somon0thi, sotc0thi, sotcno, mhno, TCHK, smhk, mhpass, TBCHK, xep_loai, sotc_qua]
+    # The order for the model features must match the training data!
+    # Let's assume the training data columns per semester were:
+    # ['So mon khong thi HK1', 'So tin chi khong thi HK1', 'So tin chi no HK1',
+    #  'So mon khong dat HK1', 'So tin chi qua mon HK1', 'Tong tin chi hoc ky HK1',
+    #  'So mon hoc ky HK1', 'So mon dat HK1', 'GPA HK1', 'Xep loai HK1']
+    # So we need to return: somon0thi, sotc0thi, sotcno, mhno, sotc_qua, TCHK, smhk, mhpass, TBCHK, xep_loai
+    return [somon0thi, sotc0thi, sotcno, mhno, sotc_qua, TCHK, smhk, mhpass, TBCHK, xep_loai_selected]
+
 
 def input_important_features(semester_label, default_values=None):
+    # Default values structure: [sotc_qua, sotcno, TBCHK]
     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", min_value=0, value=default_values[0] if default_values and len(default_values)>0 and default_values[0] is not None else 0, key=f"tcqua_imp_{semester_label}")
+            sotcno = st.number_input("Số tín chỉ nợ", min_value=0, value=default_values[1] if default_values and len(default_values)>1 and default_values[1] is not None 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", min_value=0.0, max_value=10.0, value=float(default_values[2]) if default_values and len(default_values)>2 and default_values[2] is not None 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"]
 
-# Mô hình 1: "Dùng toàn bộ dữ liệu"
+# Mô hình 1: "Dùng toàn bộ đặc 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")
     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)
+        default_values = selected_example["semester_data"][idx*10:(idx+1)*10] if selected_example and "semester_data" in selected_example else None # Slice 10 values for full model
+        data += input_semester_full(sem, default_values)
     nganh = 0 if major == "Công nghệ thông tin" else 1
-    final_features = np.array(data + [nganh]).reshape(1, -1)
+    final_features = np.array(data + [nganh]).reshape(1, -1) # Add nganh feature at the end
+
+# Mô hình 2: "Dùng đặc trưng quan trọng"
 
-# Mô hình 2: "Dùng mô hình đơn giản"
-else:
-        
-    # Ensure the example defaults to "Không ví dụ" when model type is 2
-    
+else: # model_type == "Dùng đặc trưng quan trọng"
     st.subheader("✨ Nhập thông tin rút gọn")
-    # Allow selecting an example
     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  # Fix this line to slice 3 values per semester (Số tín chỉ đạt, Số tín chỉ nợ, GPA)
+        default_values = selected_example["semester_data"][idx*3:(idx+1)*3] if selected_example and "semester_data" in selected_example else None  # Slice 3 values for simple model
         data += input_important_features(sem, default_values)
-    final_features = np.array(data).reshape(1, -1)
-
-# ===== Predict =====
-if st.button("🎯 DỰ BÁO"):
-    if model_type == "Dùng toàn bộ đặc trưng":
-        prediction = model_full.predict(final_features)
-        sheet_name = "Trang tính1"  # For full data model, save to Trang tính1
+    final_features = np.array(data).reshape(1, -1) # Simple model doesn't include 'nganh' based on your original code structure
 
-    else:
-        prediction = model_important.predict(final_features)
-        sheet_name = "Trang tính2"  # For full data model, save to Trang tính1
 
+# ===== Predict =====
 
-    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()
-
-    # # Debugging: List all sheet names
-    # sheet_id = "1i7bDNvLVLXN93_e-FN0JLzpg1jb64Z_aEuyPjIfwbdQ"  # Use your actual Spreadsheet ID
-    # sheet_names = list_sheet_names(sheet_id)
-    # st.write("Available sheet names:", sheet_names)
-
-    # Ghi log lên Google Sheets
-    save_to_gsheet(name, student_id, major, prediction, data,sheet_name)  # Pass data to save function
+if st.button("🎯 DỰ BÁO"):
+    try:
+        if model_type == "Dùng toàn bộ đặc trưng":
+            prediction = model_full.predict(final_features)
+            sheet_name = "Trang tính1"  # For full data model, save to Trang tính1
+        else:
+            prediction = model_important.predict(final_features)
+            sheet_name = "Trang tính2"  # For simple model, save to Trang tính2
+
+        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()
+
+        # Ghi log lên Google Sheets
+        # Pass the *actual* data used for the model to the save function
+        # For the full model, this is 'data' which is 60 elements + nganh = 61. But Gsheet wants the 60.
+        # For the simple model, this is 'data' which is 18 elements.
+        data_to_save = data + [nganh] if model_type == "Dùng toàn bộ đặc trưng" else data
+        save_to_gsheet(name, student_id, major, prediction, data_to_save, sheet_name)
+
+    except Exception as e:
+        st.error(f"Đã xảy ra lỗi trong quá trình dự báo hoặc lưu trữ: {e}")
+        # Optional: Print detailed error to console for debugging
+        # import traceback
+        # st.text(traceback.format_exc())