Update app.py

app.py

@@ -1,11 +1,22 @@
 import streamlit as st
 import numpy as np
 import joblib
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
 
-# Load model
-model = joblib.load("TomekLinks_rf_model.pkl")
+# ===== CẤU HÌNH GOOGLE SHEETS =====
+def save_to_google_sheet(name, student_id, major, semester_data, prediction_result):
+    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)
+    sheet = client.open_by_url("https://docs.google.com/spreadsheets/d/1i7bDNvLVLXN93_e-FN0JLzpg1jb64Z_aEuyPjIfwbdQ/edit?usp=sharing")
+    worksheet = sheet.sheet1
+
+    # Tạo dòng dữ liệu
+    new_row = [name, student_id, major] + semester_data + [prediction_result]
+    worksheet.append_row(new_row)
 
-# Setup page
+# ===== CẤU HÌNH TRANG =====
 st.set_page_config(page_title="Dự đoán tốt nghiệp đúng hạn", page_icon="🎓", layout="wide")
 
 # Inject custom CSS
@@ -34,6 +45,9 @@ st.markdown("<h1 style='text-align: center; color: #003366;'>🎓 DỰ BÁO 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("---")
 
+# ===== Load Model =====
+model = joblib.load("TomekLinks_rf_model.pkl")
+
 # ===== Thông tin cá nhân =====
 with st.container():
     col1, col2, col3 = st.columns(3)
@@ -50,13 +64,11 @@ st.write("---")
 def input_semester(semester_label):
     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", min_value=0, value=0, key=f"somon0thi_{semester_label}")
             sotc0thi = st.number_input("Số tín chỉ không thi", min_value=0, value=0, key=f"sotc0thi_{semester_label}")
             sotcno = st.number_input("Số tín chỉ nợ", min_value=0, value=0, key=f"sotcno_{semester_label}")
             mhno = st.number_input("Số môn không đạt", min_value=0, value=0, key=f"mhno_{semester_label}")
-
         with col2:
             TCHK = st.number_input("Tổng số tín chỉ", min_value=0, value=0, key=f"TCHK_{semester_label}")
             smhk = st.number_input("Tổng số môn học kỳ", min_value=0, value=0, key=f"smhk_{semester_label}")
@@ -87,14 +99,19 @@ for sem in semesters:
 nganh = 0 if major.startswith("0") else 1
 final_features = np.array(data + [nganh]).reshape(1, -1)
 
-# ===== Predict Button =====
+# ===== Predict and Save =====
 st.write("")
 if st.button("🎯 DỰ BÁO"):
     prediction = model.predict(final_features)
 
     if prediction[0] == 1:
+        result_text = "Đúng hạn"
         st.success(f"🎉 Chúc mừng bạn {name} - {major.split('-')[1].strip()}! Bạn có khả năng tốt nghiệp đúng hạn!")
         st.balloons()
     else:
+        result_text = "Trễ hạn"
         st.error(f"⚠️ Bạn {name} - {major.split('-')[1].strip()} cần cố gắng hơn! Có nguy cơ trễ hạn.")
-        st.snow()
+        st.snow()
+
+    # Gửi dữ liệu lên Google Sheets
+    save_to_google_sheet(name, student_id, major.split('-')[1].strip(), data, result_text)