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, sheet_name="Trang tính1"):
+def save_to_gsheet(name, student_id, major, prediction, semester_data, 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,8 +55,11 @@ def save_to_gsheet(name, student_id, major, prediction, sheet_name="Trang tính1
     elif isinstance(prediction, np.int64):
         prediction = int(prediction)  # Convert np.int64 to int
     
+    # 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
+    
     # Create a data row to insert into the sheet
-    data_row = [name, student_id, major, prediction, current_time]
+    data_row = [name, student_id, major, prediction, current_time] + semester_data_flat
 
     # Connect to Google Sheets
     client = get_gsheet_client()
@@ -69,8 +72,20 @@ def save_to_gsheet(name, student_id, major, prediction, sheet_name="Trang tính1
     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]) != 6:  # Ensure there are 6 columns
-        sheet.append_row(["Họ và tên", "MSV", "Khoa", "Dự báo", "Thời gian"])  # Create 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)
@@ -248,4 +263,4 @@ if st.button("🎯 DỰ BÁO"):
     # st.write("Available sheet names:", sheet_names)
 
     # Ghi log lên Google Sheets
-    save_to_gsheet(name, student_id, major, prediction)
+    save_to_gsheet(name, student_id, major, prediction, data)  # Pass data to save function