Update app.py

app.py

@@ -1,50 +1,14 @@
 import streamlit as st
 import numpy as np
 import joblib
-import gspread
-from oauth2client.service_account import ServiceAccountCredentials
-from datetime import datetime
-import pytz
-
-# ===== 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
-
-    # Lấy ngày giờ hiện tại theo giờ Việt Nam
-    vn_time = datetime.now(pytz.timezone('Asia/Ho_Chi_Minh'))
-    time_string = vn_time.strftime("%Y-%m-%d %H:%M:%S")
-
-    # Kiểm tra nếu sheet chưa có header thì thêm header
-    if worksheet.row_count == 0 or worksheet.cell(1, 1).value is None:
-        header = ["Ngày giờ nhập", "Họ và tên", "Mã sinh viên", "Ngành học"]
-        for i in range(1, 7):
-            hk_label = f"HK{i}_"
-            header += [
-                hk_label + "Số môn 0 thi",
-                hk_label + "Số TC 0 thi",
-                hk_label + "Số TC nợ",
-                hk_label + "Số môn không đạt",
-                hk_label + "Tổng số TC",
-                hk_label + "Tổng số môn",
-                hk_label + "Số môn đạt",
-                hk_label + "GPA",
-                hk_label + "Xếp loại",
-            ]
-        header.append("Kết quả dự đoán")
-        worksheet.append_row(header)
-
-    # Tạo dòng dữ liệu mới
-    new_row = [time_string, name, student_id, major] + semester_data + [prediction_result]
-    worksheet.append_row(new_row)
-
-# ===== CẤU HÌNH TRANG =====
+
+# ===== Load model =====
+model = joblib.load("D:/Pycharm/Project3/TomekLinks_rf_model.pkl")
+
+# ===== Setup page =====
 st.set_page_config(page_title="Dự đoán tốt nghiệp đúng hạn", page_icon="🎓", layout="wide")
 
-# Inject custom CSS
+# ===== Custom CSS =====
 st.markdown("""
     <style>
     html, body, [class*="css"]  {
@@ -65,8 +29,10 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# ===== Load Model =====
-model = joblib.load("TomekLinks_rf_model.pkl")
+# ===== HEADER =====
+st.markdown("<h1 style='text-align: center; color: #003366;'>🎓 DỰ BÁO 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("---")
 
 # ===== BỘ VÍ DỤ MẪU =====
 sample_cntt_example = {
@@ -123,11 +89,13 @@ st.write("---")
 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", min_value=0, value=default_values[0] if default_values else 0, key=f"somon0thi_{semester_label}")
             sotc0thi = st.number_input("Số tín chỉ không thi", min_value=0, value=default_values[1] if default_values else 0, key=f"sotc0thi_{semester_label}")
             sotcno = st.number_input("Số tín chỉ nợ", min_value=0, value=default_values[2] if default_values else 0, key=f"sotcno_{semester_label}")
             mhno = st.number_input("Số môn không đạt", min_value=0, value=default_values[3] if default_values else 0, key=f"mhno_{semester_label}")
+
         with col2:
             TCHK = st.number_input("Tổng số tín chỉ", min_value=0, value=default_values[4] if default_values else 0, key=f"TCHK_{semester_label}")
             smhk = st.number_input("Tổng số môn học kỳ", min_value=0, value=default_values[5] if default_values else 0, key=f"smhk_{semester_label}")
@@ -164,19 +132,16 @@ for idx, sem in enumerate(semesters):
 nganh = 0 if major == "Công nghệ thông tin" else 1
 final_features = np.array(data + [nganh]).reshape(1, -1)
 
-# ===== Predict and Save =====
+# ===== Predict Button =====
 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}! 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} cần cố gắng hơn! Có nguy cơ trễ hạn.")
         st.snow()
 
-    # Gửi dữ liệu lên Google Sheets
-    save_to_google_sheet(name, student_id, major, data, result_text)
+