Update app.py

app.py

@@ -40,23 +40,70 @@ def get_gsheet_client():
     client = gspread.authorize(creds)
     return client
 
-from datetime import datetime
-import json
-
-def log_to_gsheet(name, student_id, major, model_type, prediction_result, final_features):
+def log_to_gsheet(name, student_id, major, model_type, prediction_result, final_features, model_type_selected, semester_data):
     """Ghi thông tin dự đoán vào Google Sheets, bao gồm lựa chọn mô hình và toàn bộ dữ liệu nhập."""
     try:
+        # Tạo kết nối với Google Sheets
         client = get_gsheet_client()
-        sheet = client.open("Kết quả nhập").sheet1
-        sheet.append_row([
+        
+        # Chọn sheet phù hợp theo mô hình
+        if model_type_selected == "Dùng toàn bộ dữ liệu":
+            sheet = client.open("Kết quả nhập").get_worksheet(0)  # Sheet 1
+        else:
+            sheet = client.open("Kết quả nhập").get_worksheet(1)  # Sheet 2
+
+        # Lấy header (cột) của Google Sheets
+        headers = sheet.row_values(1)
+
+        # Danh sách các cột cần tạo nếu chưa có
+        required_columns_sheet1 = [
+            "Ngày giờ nhập", "Họ và tên", "Mã sinh viên", "Ngành học", "Mô hình", 
+            "Kết quả", "Thời gian dự đoán", "Số môn không thi", "Số tín chỉ không thi",
+            "Số tín chỉ nợ", "Số môn không đạt", "Số tín chỉ qua môn", "Tổng tín chỉ học kỳ", 
+            "Số môn học kỳ", "Số môn đạt", "GPA", "Xếp loại"
+        ]
+        
+        required_columns_sheet2 = [
+            "Ngày giờ nhập", "Họ và tên", "Mã sinh viên", "Ngành học", "Mô hình", 
+            "Kết quả", "Thời gian dự đoán", "Số tín chỉ đạt", "Số tín chỉ nợ", "Điểm trung bình"
+        ]
+        
+        # Chọn các cột cần kiểm tra tùy vào mô hình
+        if model_type_selected == "Dùng toàn bộ dữ liệu":
+            required_columns = required_columns_sheet1
+        else:
+            required_columns = required_columns_sheet2
+        
+        # Thêm cột mới nếu chưa có
+        for column in required_columns:
+            if column not in headers:
+                sheet.add_cols(1)
+                headers = sheet.row_values(1)  # Cập nhật lại headers sau khi thêm cột mới
+
+        # Phân tách dữ liệu học kỳ
+        semester_values = []
+        for semester in semester_data:
+            semester_values.extend(semester)
+
+        # Tạo dòng mới để lưu vào
+        row = [
+            datetime.now().strftime("%Y-%m-%d %H:%M:%S"),  # Ngày giờ nhập
             name,
             student_id,
             major,
             model_type,
-            "Đúng hạn" if prediction_result == 1 else "Trễ hạn",
-            datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-            json.dumps(final_features.tolist()) if isinstance(final_features, np.ndarray) else json.dumps(final_features)
-        ])
+            "Đúng hạn" if prediction_result == 1 else "Trễ hạn",  # Kết quả
+            datetime.now().strftime("%Y-%m-%d %H:%M:%S"),  # Thời gian dự đoán
+        ]
+        
+        if model_type_selected == "Dùng toàn bộ dữ liệu":
+            row.extend(semester_values)  # Dữ liệu đầy đủ cho Sheet 1
+        else:
+            row.extend([semester_values[4], semester_values[2], semester_values[7]])  # Dữ liệu quan trọng cho Sheet 2
+
+        # Lưu dòng vào Google Sheets
+        sheet.append_row(row)
+
     except Exception as e:
         st.warning(f"Không thể lưu dữ liệu vào Google Sheets: {e}")