Update app.py

app.py

@@ -247,8 +247,12 @@ else:
 if st.button("🎯 DỰ BÁO"):
     if model_type == "Dùng toàn bộ dữ liệu":
         prediction = model_full.predict(final_features)
+        sheet_name = "Trang tính1"  # For full data model, save to Trang tính1
+
     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"🎉 Chúc mừng bạn {name} - {major}! Bạn có khả năng tốt nghiệp đúng hạn!")
@@ -263,4 +267,4 @@ if st.button("🎯 DỰ BÁO"):
     # st.write("Available sheet names:", sheet_names)
 
     # Ghi log lên Google Sheets
-    save_to_gsheet(name, student_id, major, prediction, data)  # Pass data to save function
+    save_to_gsheet(name, student_id, major, prediction, data,sheet_name)  # Pass data to save function