Update app.py

app.py

@@ -43,7 +43,7 @@ def get_gsheet_client():
     return client
 
 # ===== Save to Google Sheets =====
-def save_to_gsheet(name, student_id, major, prediction, semester_data, sheet_name="Trang tính1"):
+def save_to_gsheet(name, student_id, major, prediction, sheet_name="Trang tính1"):
     # Get the current date and time
     now = datetime.now()
     current_time = now.strftime("%Y-%m-%d %H:%M:%S")
@@ -55,50 +55,8 @@ def save_to_gsheet(name, student_id, major, prediction, semester_data, sheet_nam
     elif isinstance(prediction, np.int64):
         prediction = int(prediction)  # Convert np.int64 to int
     
-    # Convert prediction to "Đúng hạn" or "Không đúng hạn"
-    if prediction == 1:
-        prediction_result = "Đúng hạn"
-    else:
-        prediction_result = "Không đúng hạn"
-
     # Create a data row to insert into the sheet
-    data_row = [name, student_id, major, prediction_result, current_time]
-    
-    # Define semester columns for all 6 semesters
-    semester_columns = [
-        # Semester 1
-        "Số môn không thi_HK1", "Số tín chỉ không thi_HK1", "Số tín chỉ nợ_HK1", "Số môn không đạt_HK1", 
-        "Số tín chỉ qua môn_HK1", "Tổng tín chỉ học kỳ_HK1", "Số môn học kỳ_HK1", "Số môn đạt_HK1", 
-        "GPA_HK1", "Xếp loại_HK1",
-        
-        # Semester 2
-        "Số môn không thi_HK2", "Số tín chỉ không thi_HK2", "Số tín chỉ nợ_HK2", "Số môn không đạt_HK2", 
-        "Số tín chỉ qua môn_HK2", "Tổng tín chỉ học kỳ_HK2", "Số môn học kỳ_HK2", "Số môn đạt_HK2", 
-        "GPA_HK2", "Xếp loại_HK2",
-        
-        # Semester 3
-        "Số môn không thi_HK3", "Số tín chỉ không thi_HK3", "Số tín chỉ nợ_HK3", "Số môn không đạt_HK3", 
-        "Số tín chỉ qua môn_HK3", "Tổng tín chỉ học kỳ_HK3", "Số môn học kỳ_HK3", "Số môn đạt_HK3", 
-        "GPA_HK3", "Xếp loại_HK3",
-
-        # Semester 4
-        "Số môn không thi_HK4", "Số tín chỉ không thi_HK4", "Số tín chỉ nợ_HK4", "Số môn không đạt_HK4", 
-        "Số tín chỉ qua môn_HK4", "Tổng tín chỉ học kỳ_HK4", "Số môn học kỳ_HK4", "Số môn đạt_HK4", 
-        "GPA_HK4", "Xếp loại_HK4",
-
-        # Semester 5
-        "Số môn không thi_HK5", "Số tín chỉ không thi_HK5", "Số tín chỉ nợ_HK5", "Số môn không đạt_HK5", 
-        "Số tín chỉ qua môn_HK5", "Tổng tín chỉ học kỳ_HK5", "Số môn học kỳ_HK5", "Số môn đạt_HK5", 
-        "GPA_HK5", "Xếp loại_HK5",
-
-        # Semester 6
-        "Số môn không thi_HK6", "Số tín chỉ không thi_HK6", "Số tín chỉ nợ_HK6", "Số môn không đạt_HK6", 
-        "Số tín chỉ qua môn_HK6", "Tổng tín chỉ học kỳ_HK6", "Số môn học kỳ_HK6", "Số môn đạt_HK6", 
-        "GPA_HK6", "Xếp loại_HK6"
-    ]
-    
-    # Add semester data to the data row
-    data_row += semester_data
+    data_row = [name, student_id, major, prediction, current_time]
 
     # Connect to Google Sheets
     client = get_gsheet_client()
@@ -111,14 +69,24 @@ def save_to_gsheet(name, student_id, major, prediction, semester_data, sheet_nam
     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(data_row):  # Ensure the correct number of columns
-        sheet.append_row([
-            "Họ và tên", "MSV", "Khoa", "Dự báo", "Thời gian"] + semester_columns)  # Create headers
+    if not all_values or len(all_values[0]) != 6:  # Ensure there are 6 columns
+        sheet.append_row(["Họ và tên", "MSV", "Khoa", "Dự báo", "Thời gian"])  # Create headers
     
     # 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
+
+
+
 # ===== 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)
@@ -209,7 +177,39 @@ else:
 st.write("---")
 
 # ===== Nhập thông tin học kỳ =====
-semester_data = []
+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":
@@ -217,16 +217,16 @@ if model_type == "Dùng toàn bộ dữ liệu":
     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
-        semester_data += input_semester(sem, default_values)
+        data += input_semester(sem, default_values)
     nganh = 0 if major == "Công nghệ thông tin" else 1
-    final_features = np.array(semester_data + [nganh]).reshape(1, -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*9:(idx+1)*9] if example_data else None
-        semester_data += input_important_features(sem, default_values)
-    final_features = np.array(semester_data).reshape(1, -1)
+        data += input_important_features(sem, default_values)
+    final_features = np.array(data).reshape(1, -1)
 
 # ===== Predict =====
 if st.button("🎯 DỰ BÁO"):
@@ -242,5 +242,10 @@ if st.button("🎯 DỰ BÁO"):
         st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ hạn.")
         st.snow()
 
-    # Save all data to Google Sheets including semester details
-    save_to_gsheet(name, student_id, major, prediction, semester_data)
+    # # 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)