Update app.py

app.py

@@ -84,11 +84,34 @@ if st.button("Submit"):
     e_salary=np.round(((e_salary-100090.239881)/57510.492818),3)
     
     
-    output = bank_model.predict([[credit_score, geography_value, gender_value, age, tenure, balance,  e_salary,  active_holder_value]])
+    prediction = bank_model.predict([[credit_score, geography_value, gender_value, age, tenure, balance,  e_salary,  active_holder_value]])
 
+      # Define messages and colors
+        review_status = {
+            1: ("✅ The Customer is Interseted in our Bank", "#32CD32"),  # Green
+            0: ("❌ The Customer is Not Interseted in our Bank", "#FF4500")   # Red-Orange
+        }
     
-    if output ==0:
-        st.write("The Customer is Interseted in our Bank")
-    else:
-        st.write("The Customer is Not Interseted in our Bank")
+        # Get message and color based on prediction
+        message, color = review_status.get(prediction, ("❓ Unknown Prediction", "#808080"))
+    
+        # Display styled result
+        st.markdown(f"""
+            <div style="
+                padding: 15px; 
+                background-color: {color}; 
+                border-radius: 10px; 
+                text-align: center; 
+                font-size: 18px; 
+                font-weight: bold; 
+                color: white;">
+                {message}
+            </div>
+        """, unsafe_allow_html=True)
+    
+    except Exception as e:
+        st.error(f"⚠️ Error in prediction: {e}")
+    
+    st.write("")
+