Update app.py

app.py

@@ -3,6 +3,8 @@ 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")
@@ -108,7 +110,7 @@ def log_to_gsheet(name, student_id, major, model_type, prediction_result, final_
         st.warning(f"Không thể lưu dữ liệu vào Google Sheets: {e}")
 
 # ===== 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("<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("---")
 
@@ -120,8 +122,8 @@ st.subheader("🔢 Chọn ví dụ mẫu hoặc nhập thông tin cá nhân")
 
 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 =====
-sample_cntt_example = {
+# ===== Ví dụ mẫu cho mô hình "Dùng toàn bộ dữ liệu" =====
+sample_cntt_example_full = {
     "name": "Nguyễn Văn A",
     "student_id": "10117367",
     "major": "Công nghệ thông tin",
@@ -135,7 +137,8 @@ sample_cntt_example = {
     ]
 }
 
-sample_kinhte_example = {
+# ===== Ví dụ mẫu cho mô hình "Dùng mô hình đơn giản" =====
+sample_kinhte_example_simple = {
     "name": "Trần Thị B",
     "student_id": "11418092",
     "major": "Kinh tế",
@@ -149,14 +152,44 @@ sample_kinhte_example = {
     ]
 }
 
+# ===== Ví dụ cho mô hình "Dùng mô hình đơn giản" cho CNTT =====
+sample_cntt_example_simple = {
+    "name": "Nguyễn Văn B",
+    "student_id": "10117368",
+    "major": "Công nghệ thông tin",
+    "semester_data": [
+        0, 0, 0, 0, 15, 8, 8, 7.50, 0,
+        0, 0, 0, 0, 17, 7, 7, 8.10, 0,
+        0, 0, 0, 0, 18, 7, 7, 7.90, 0,
+        0, 0, 0, 0, 17, 7, 7, 7.80, 0,
+        0, 0, 0, 0, 19, 8, 6, 8.30, 0,
+        0, 0, 6, 3, 19, 7, 5, 7.20, 1
+    ]
+}
+
+# ===== 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",
+    "major": "Kinh tế",
+    "semester_data": [
+        0, 0, 0, 0, 16, 6, 6, 7.00, 1,
+        0, 0, 0, 0, 18, 7, 7, 8.20, 1,
+        0, 0, 0, 0, 17, 7, 7, 7.80, 1,
+        0, 0, 0, 0, 17, 7, 6, 7.90, 1,
+        0, 0, 0, 0, 19, 8, 5, 8.10, 0,
+        0, 0, 8, 2, 19, 7, 6, 7.30, 1
+    ]
+}
+
 # ===== 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["name"])
-    student_id = st.text_input("🎓 Mã sinh viên", value=sample_cntt_example["student_id"])
+    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)
 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["name"])
-    student_id = st.text_input("🎓 Mã sinh viên", value=sample_kinhte_example["student_id"])
+    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)
 else:
     name = st.text_input("👤 Họ và tên")
@@ -165,7 +198,7 @@ else:
 
 st.write("---")
 
-# ===== Hàm nhập liệu =====
+# ===== 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)
@@ -204,7 +237,7 @@ semesters = ["HỌC KỲ I", "HỌC KỲ II", "HỌC KỲ III", "HỌC KỲ IV",
 if model_type == "Dùng toàn bộ dữ liệu":
     st.subheader("🔢 Nhập thông tin học kỳ chi tiết")
     for idx, sem in enumerate(semesters):
-        example_data = sample_cntt_example if "CNTT" in sample_option else sample_kinhte_example if "Kinh tế" in sample_option else None
+        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
@@ -212,21 +245,8 @@ if model_type == "Dùng toàn bộ dữ liệu":
 else:
     st.subheader("✨ Nhập thông tin rút gọn")
     for idx, sem in enumerate(semesters):
-        example_data = None
-        if sample_option == "Dùng ví dụ mẫu ngành Công nghệ thông tin":
-            example_data = sample_cntt_example
-        elif sample_option == "Dùng ví dụ mẫu ngành Kinh tế":
-            example_data = sample_kinhte_example
-
-        if example_data:
-            raw = example_data["semester_data"][idx*9:(idx+1)*9]
-            if len(raw) >= 8:
-                default_values = [raw[4], raw[2], raw[7]]
-            else:
-                default_values = None
-        else:
-            default_values = None
-
+        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*9:(idx+1)*9] if example_data else None
         data += input_important_features(sem, default_values)
     final_features = np.array(data).reshape(1, -1)
 
@@ -245,4 +265,4 @@ if st.button("🎯 DỰ BÁO"):
         st.snow()
 
     # Ghi log lên Google Sheets
-    log_to_gsheet(name, student_id, major, model_type, prediction[0], final_features)
+    log_to_gsheet(name, student_id, major, model_type, prediction[0], final_features, model_type, data)