Update app.py

app.py

@@ -1,10 +1,6 @@
 import streamlit as st
 import numpy as np
 import joblib
-import gspread
-from oauth2client.service_account import ServiceAccountCredentials
-from datetime import datetime
-import json
 
 # ===== Load models =====
 model_full = joblib.load("random_forest_model_full.pkl")
@@ -20,7 +16,7 @@ st.markdown("""
         font-family: 'Poppins', sans-serif;
     }
     .block-container {
-        padding: 2rem 2rem 2rem 2rem;
+        padding: 2rem;
     }
     .stButton button {
         background: linear-gradient(90deg, #4F8BF9 0%, #8E2DE2 100%);
@@ -34,282 +30,176 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# ===== GSheet integration =====
-@st.cache_resource
-def get_gsheet_client():
-    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)
-    return client
-
-# ===== Save to Google Sheets =====
-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")
-    
-    # Ensure prediction is in the expected format (0 or 1)
-    if isinstance(prediction, np.ndarray):
-        prediction = int(prediction[0])  # Convert to int if it's ndarray
-    
-    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] + semester_data_flat
-
-    # Connect to Google Sheets
-    client = get_gsheet_client()
-
-    # Use the Spreadsheet ID (replace with your actual ID)
-    sheet_id = "1i7bDNvLVLXN93_e-FN0JLzpg1jb64Z_aEuyPjIfwbdQ"  # Use your actual ID here
-    sheet = client.open_by_key(sheet_id).worksheet(sheet_name)
-
-    # Get all values from the sheet to check if the first row is empty (i.e., headers not created yet)
-    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(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)
-
-
-# ===== 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)
 st.write("---")
 
-# ===== SIDEBAR: chọn mô hình =====
-model_type = st.sidebar.selectbox("🧠 Chọn mô hình dự báo:", ["Dùng toàn bộ đặc trưng", "Dùng đặc trưng quan trọng"])
 
-# ===== Giao diện nhập ví dụ =====
-st.subheader("🔢 Chọn ví dụ mẫu hoặc nhập thông tin cá nhân")
+# ===== SIDEBAR: chọn mô hình =====
+model_type = st.sidebar.selectbox("🧠 Chọn mô hình dự báo:", [
+    "Dùng toàn bộ đặc trưng",
+    "Dùng đặc trưng quan trọng"
+])
 
-sample_option = st.selectbox("📝 Chọn ví dụ:", ["Không ví dụ", "Dùng ví dụ mẫu ngành Công nghệ thông tin", "Dùng ví dụ mẫu ngành Kinh tế"])
 
-# ===== Ví dụ mẫu cho mô hình "Dùng toàn bộ dữ liệu" =====
+# ===== Ví dụ mẫu cho FULL MODEL (9 feature/kỳ) =====
 sample_cntt_example_full = {
     "name": "Nguyễn Văn A",
     "student_id": "10117367",
     "major": "Công nghệ thông tin",
     "semester_data": [
-        0, 0, 0, 0, 17, 7, 7, 8.73, 6,
-        0, 0, 0, 0, 17, 7, 7, 8.19, 5,
-        0, 0, 0, 0, 17, 7, 7, 7.90, 5,
-        0, 0, 0, 0, 17, 7, 7, 8.19, 6,
-        0, 0, 0, 0, 19, 7, 6, 8.18, 6,
-        0, 0, 5, 2, 19, 7, 5, 7.10, 5
+        0,0,0,0,17,7,7,8.73,6,
+        0,0,0,0,17,7,7,8.19,5,
+        0,0,0,0,17,7,7,7.90,5,
+        0,0,0,0,17,7,7,8.19,6,
+        0,0,0,0,19,7,6,8.18,6,
+        0,0,0,0,19,7,5,7.10,5
     ]
 }
 
-# ===== Ví dụ mẫu cho mô hình "Dùng mô hình đơn giản" =====
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho Kinh tế =====
-sample_kinhte_example_simple = {
-    "name": "Trần Thị B",
-    "student_id": "11418092",
+sample_kinhte_example_full = {
+    "name": "Trần Thị C",
+    "student_id": "11418093",
     "major": "Kinh tế",
     "semester_data": [
-        17, 0, 7.64,  # Học kỳ 1: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 0, 7.19,  # Học kỳ 2: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 2, 6.94,  # Học kỳ 3: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 0, 7.19,  # Học kỳ 4: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        19, 0, 7.65,  # Học kỳ 5: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        19, 0, 7.49   # Học kỳ 6: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        0,0,0,0,16,6,6,7.00,5,
+        0,0,0,0,18,7,7,8.20,6,
+        0,0,0,0,17,7,7,7.80,5,
+        0,0,0,0,17,7,6,7.90,5,
+        0,0,0,0,19,8,5,8.10,6,
+        0,0,8,2,19,7,6,7.30,5
     ]
 }
 
-
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho CNTT =====
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho CNTT =====
+# ===== Ví dụ cho SIMPLE MODEL (3 feature/kỳ) =====
 sample_cntt_example_simple = {
     "name": "Nguyễn Văn B",
     "student_id": "10117368",
     "major": "Công nghệ thông tin",
     "semester_data": [
-        15, 8, 6.50,  # Học kỳ 1: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 15, 6.10,  # Học kỳ 2: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        18, 7, 5.90,  # Học kỳ 3: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        17, 7, 5.80,  # Học kỳ 4: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        19, 8, 6.30,  # Học kỳ 5: Số tín chỉ đạt, Số tín chỉ nợ, GPA
-        19, 10, 6.20   # Học kỳ 6: Số tín chỉ đạt, Số tín chỉ nợ, GPA
+        15,8,6.50,
+        17,15,6.10,
+        18,7,5.90,
+        17,7,5.80,
+        19,8,6.30,
+        19,10,6.20
     ]
 }
 
-# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho Kinh tế =====
-sample_kinhte_example_full = {
-    "name": "Trần Thị C",
-    "student_id": "11418093",
+sample_kinhte_example_simple = {
+    "name": "Trần Thị B",
+    "student_id": "11418092",
     "major": "Kinh tế",
     "semester_data": [
-        0, 0, 0, 0, 16, 6, 6, 7.00, 5,
-        0, 0, 0, 0, 18, 7, 7, 8.20, 6,
-        0, 0, 0, 0, 17, 7, 7, 7.80, 5,
-        0, 0, 0, 0, 17, 7, 6, 7.90, 5,
-        0, 0, 0, 0, 19, 8, 5, 8.10, 6,
-        0, 0, 8, 2, 19, 7, 6, 7.30, 5
+        17,0,7.64,
+        17,0,7.19,
+        17,2,6.94,
+        17,0,7.19,
+        19,0,7.65,
+        19,0,7.49
     ]
 }
 
+
+# ===== CHỌN VÍ DỤ =====
+sample_option = st.selectbox("📝 Chọn ví dụ:", [
+    "Không ví dụ",
+    "Dùng ví dụ mẫu ngành Công nghệ thông tin",
+    "Dùng ví dụ mẫu ngành Kinh tế"
+])
+
 # ===== Thông tin cá nhân =====
 if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin":
-    name = st.text_input("👤 Họ và tên", value=sample_cntt_example_full["name"])
-    student_id = st.text_input("🎓 Mã sinh viên", value=sample_cntt_example_full["student_id"])
-    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=0)
+    example = sample_cntt_example_full if model_type == "Dùng toàn bộ đặc trưng" else sample_cntt_example_simple
 elif sample_option == "Dùng ví dụ mẫu ngành Kinh tế":
-    name = st.text_input("👤 Họ và tên", value=sample_kinhte_example_simple["name"])
-    student_id = st.text_input("🎓 Mã sinh viên", value=sample_kinhte_example_simple["student_id"])
-    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"], index=1)
+    example = sample_kinhte_example_full if model_type == "Dùng toàn bộ đặc trưng" else sample_kinhte_example_simple
 else:
-    name = st.text_input("👤 Họ và tên")
-    student_id = st.text_input("🎓 Mã sinh viên")
-    major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"])
+    example = None
+
+name = st.text_input("👤 Họ và tên", value=example["name"] if example else "")
+student_id = st.text_input("🎓 Mã sinh viên", value=example["student_id"] if example else "")
+major = st.selectbox("📚 Ngành học", ["Công nghệ thông tin", "Kinh tế"],
+                     index=0 if not example else (0 if example["major"] == "Công nghệ thông tin" else 1))
 
 st.write("---")
 
-# ===== Nhập thông tin học kỳ =====
-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 =====
+
+# ===== HÀM NHẬP DỮ LIỆU =====
+def input_full_semester(label, default):
+    with st.expander(label, expanded=True):
+        results = []
+        fields = [
+            "Số môn không thi", "Số tín chỉ không thi", "Số tín chỉ nợ", "Số môn không đạt",
+            "Tổng tín chỉ học kỳ", "Số môn học kỳ", "Số môn đạt", "GPA", "Xếp loại"
+        ]
+        
+        col1, col2, col3 = st.columns(3)
+        for i, field in enumerate(fields):
+            value = default[i] if default else 0
+            with (col1 if i < 3 else col2 if i < 6 else col3):
+                if i == 7:
+                    value = st.number_input(f"{field}", 0.0, 10.0, value=float(value), step=0.01)
+                elif i == 8:
+                    value = st.selectbox(f"{field}", list(range(7)), index=value)
+                else:
+                    value = st.number_input(f"{field}", 0, value=value)
+            results.append(value)
+
+    return results
+
+
+def input_simple_semester(label, default):
+    with st.expander(label, expanded=True):
+        col1, col2, col3 = st.columns(3)
+        sotc_qua = col1.number_input(f"Số tín chỉ đạt - {label}", 0, value=default[0] if default else 0)
+        sotc_no = col2.number_input(f"Số tín chỉ nợ - {label}", 0, value=default[1] if default else 0)
+        gpa = col3.number_input(f"GPA - {label}", 0.0, 10.0, value=default[2] if default else 0.0, step=0.01)
+    return [sotc_qua, sotc_no, gpa]
+
+
+# ===== NHẬP DỮ LIỆU =====
+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"]
 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"]
 
-# Mô hình 1: "Dùng toàn bộ dữ liệu"
 if model_type == "Dùng toàn bộ đặc trưng":
-    st.subheader("🔢 Nhập thông tin học kỳ chi tiết")
-    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
-        data += input_semester(sem, default_values)
-    nganh = 0 if major == "Công nghệ thông tin" else 1
-    final_features = np.array(data + [nganh]).reshape(1, -1)
-
-# Mô hình 2: "Dùng mô hình đơn giản"
+    st.subheader("🔢 Nhập thông tin đầy đủ 6 học kỳ")
+
+    for i, sem in enumerate(semesters):
+        default_values = example["semester_data"][i*9:(i+1)*9] if example else None
+        data += input_full_semester(sem, default_values)
+
+    final_features = np.array(data).reshape(1, -1)
+
 else:
     st.subheader("✨ Nhập thông tin rút gọn")
-    
-    # Allow selecting an example
-    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*3:(idx+1)*3] if example_data else None  # Fix this line to slice 3 values per semester (Số tín chỉ đạt, Số tín chỉ nợ, GPA)
-        
-        # Use expander to group each semester's fields in a collapsible box
-        if idx < 3:  # First row (3 boxes)
-            with st.expander(f"📘 {sem}", expanded=True):
-                col1, col2, col3 = st.columns(3)  # 3 columns for the first row
-                with col1:
-                    sotc_qua = st.number_input(f"Số tín chỉ đạt - {sem}", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{sem}")
-                with col2:
-                    sotcno = st.number_input(f"Số tín chỉ nợ - {sem}", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{sem}")
-                with col3:
-                    TBCHK = st.number_input(f"Điểm trung bình - {sem}", 0.0, 10.0, value=default_values[2] if default_values else 0.0, step=0.01, key=f"gpa_imp_{sem}")
-        else:  # Second row (2 boxes)
-            with st.expander(f"📘 {sem}", expanded=True):
-                col1, col2 = st.columns(2)  # 2 columns for the second row
-                with col1:
-                    sotc_qua = st.number_input(f"Số tín chỉ đạt - {sem}", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{sem}")
-                    sotcno = st.number_input(f"Số tín chỉ nợ - {sem}", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{sem}")
-                with col2:
-                    TBCHK = st.number_input(f"Điểm trung bình - {sem}", 0.0, 10.0, value=default_values[2] if default_values else 0.0, step=0.01, key=f"gpa_imp_{sem}")
-        
-        # Collect semester data
-        data += [sotc_qua, sotcno, TBCHK]  
-    
+
+    for i, sem in enumerate(semesters):
+        default_values = example["semester_data"][i*3:(i+1)*3] if example else None
+        data += input_simple_semester(sem, default_values)
+
     final_features = np.array(data).reshape(1, -1)
 
-# ===== Predict =====
+
+# ===== PREDICT =====
 if st.button("🎯 DỰ BÁO"):
     if model_type == "Dùng toàn bộ đặc trưng":
-        prediction = model_full.predict(final_features)
-        sheet_name = "Trang tính1"  # For full data model, save to Trang tính1
-        if prediction[0] == 0:
-                # st.success(f"Dự đoán là : {prediction[0]}")
-                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()
+        prediction = model_full.predict(final_features)[0]
+        
+        if prediction == 0:   # 0 = tốt nghiệp đúng hạn (theo mô hình của bạn)
+            st.success(f"🎉 Chúc mừng {name}! Bạn có khả năng tốt nghiệp đúng hạn!")
+            st.balloons()
         else:
-            st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ hạn.")
+            st.error(f"⚠️ Bạn có nguy cơ trễ hạn, cần cố gắng hơn!")
             st.snow()
 
     else:
-        prediction = model_important.predict(final_features)
-        sheet_name = "Trang tính2"  # For full data model, save to Trang tính1
-        if prediction[0] == 1:
-                # st.success(f"Dự đoán là : {prediction[0]}")
-                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()
+        prediction = model_important.predict(final_features)[0]
+
+        if prediction == 1:   # 1 = đúng hạn trong simple model
+            st.success(f"🎉 Chúc mừng {name}! Bạn có khả năng tốt nghiệp đúng hạn!")
+            st.balloons()
         else:
-            st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ hạn.")
+            st.error(f"⚠️ Bạn có nguy cơ trễ hạn, cần cố gắng hơn!")
             st.snow()
-
-
-    # if prediction[0] == 0:
-    #     # st.success(f"Dự đoán là : {prediction[0]}")
-    #     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:
-    #     st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ hạn.")
-    #     st.snow()
-
-    # # 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, data,sheet_name)  # Pass data to save function