Update app.py

app.py

@@ -278,19 +278,33 @@ 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()
+        else:
+            st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ 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] == 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()
+        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()
+        else:
+            st.error(f"⚠️ Bạn {name} - {major} cần cố gắng hơn! Có nguy cơ trễ 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