Update app.py

app.py

@@ -31,60 +31,15 @@ st.markdown("""
         font-size: 18px;
         font-weight: bold;
     }
-
-    /* Styling for the semester box */
-    .semester-box {
-        border: 2px solid #003366;
-        padding: 15px;
-        border-radius: 8px;
-        margin-top: 10px;
-        width: 45%;  /* Adjust width of the box */
-        display: inline-block;
-        margin-right: 20px;  /* Space between boxes */
-        margin-bottom: 20px;
-    }
-
-    /* Styling for the label */
-    .input-label {
-        font-weight: bold;
-        font-size: 14px;
-        margin-bottom: 5px;
-        display: block;
-    }
-
-    /* Styling for the input field */
-    .input-field {
-        border: 2px solid #003366;
-        border-radius: 8px;
-        padding: 8px;
-        font-size: 14px;
-        width: 100%;  /* Ensure inputs are responsive to box size */
-        margin-bottom: 10px;
-    }
-
-    /* Styling for the whole semester input section */
-    .semester-section {
-        display: flex;
-        flex-wrap: wrap;
-        justify-content: space-between;
-        margin-bottom: 20px;
-    }
-
-    /* Input container for the label and field alignment */
-    .input-container {
-        display: flex;
-        flex-direction: column;
-        margin-bottom: 10px;
-    }
-
-    /* Ensure space between the label and field */
-    .semester-box-inner {
-        margin-bottom: 15px;
+    /* Shrink number_input and selectbox */
+    .stNumberInput input, .stSelectbox select {
+        font-size: 14px;  /* Smaller font size */
+        padding: 5px;  /* Reduce padding */
     }
-
     </style>
 """, unsafe_allow_html=True)
 
+
 # ===== GSheet integration =====
 @st.cache_resource
 def get_gsheet_client():
@@ -141,6 +96,18 @@ def save_to_gsheet(name, student_id, major, prediction, semester_data, sheet_nam
     # 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)
@@ -170,6 +137,7 @@ sample_cntt_example_full = {
 }
 
 # ===== 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",
@@ -184,6 +152,8 @@ sample_kinhte_example_simple = {
     ]
 }
 
+
+# ===== 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 =====
 sample_cntt_example_simple = {
     "name": "Nguyễn Văn B",
@@ -199,6 +169,21 @@ sample_cntt_example_simple = {
     ]
 }
 
+# ===== 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_full["name"])
@@ -216,6 +201,38 @@ else:
 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 =====
 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"]
 
@@ -231,32 +248,15 @@ if model_type == "Dùng toàn bộ đặc trưng":
 
 # Mô hình 2: "Dùng mô hình đơn giản"
 else:
-    st.subheader("✨ Nhập thông tin rút gọn")
+        
+    # Ensure the example defaults to "Không ví dụ" when model type is 2
     
+    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)
-        
-        with st.expander(f"📘 {sem}", expanded=True):
-            # Create a box for each semester with 3 rows of label and input box
-            col1 = st.columns(1)[0]  # Ensure single column for each input box
-            
-            # Row 1
-            st.write("Số tín chỉ đạt:")
-            sotc_qua = st.number_input(f"Điền Số tín chỉ đạt - {sem}", 0, value=default_values[0] if default_values else 0, key=f"tcqua_imp_{sem}")
-            
-            # Row 2
-            st.write("Số tín chỉ nợ:")
-            sotcno = st.number_input(f"Điền Số tín chỉ nợ - {sem}", 0, value=default_values[1] if default_values else 0, key=f"tcno_imp_{sem}")
-            
-            # Row 3
-            st.write("Điểm trung bình:")
-            TBCHK = st.number_input(f"Điền Đ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]  
-    
+        data += input_important_features(sem, default_values)
     final_features = np.array(data).reshape(1, -1)
 
 # ===== Predict =====
@@ -269,6 +269,7 @@ if st.button("🎯 DỰ BÁO"):
         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"🎉 Chúc mừng bạn {name} - {major}! Bạn có khả năng tốt nghiệp đúng hạn!")
         st.balloons()
@@ -276,5 +277,10 @@ if st.button("🎯 DỰ BÁO"):
         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
+    save_to_gsheet(name, student_id, major, prediction, data,sheet_name)  # Pass data to save function