Upload predict.py

predict.py

@@ -99,7 +99,7 @@ def run():
         "Sample 3": {
             "customer_id": "CUST003",
             "tenure": 6,
-            "contract": "month-to-month",
+            "contract": "two year",
             "payment_method": "bank transfer",
             "monthly_charges": 400000.00,
             "total_charges": 2400000.00,
@@ -187,7 +187,12 @@ def run():
             labels, confidences = predict_sentiment([feedback])
             label = labels[0]
             confidence = confidences[0]
-            st.success(f"Predicted Sentiment: **{label}**")
+            st.write("**Feedback:**")
+            st.write(feedback)
+            if label == "Negative":
+                st.error(f"Predicted Sentiment: **{label}**")
+            else:
+                st.success(f"Predicted Sentiment: **{label}**")
             st.info(f"Confidence: {confidence * 100:.2f}%")
             data['sentiment'] = label
         else:
@@ -200,7 +205,7 @@ def run():
         # Load customer churn model
         status_text.text("Loading churn prediction model...")
         for i in range(50, 75):
-            time.sleep(0.05)
+            time.sleep(0.01)
             progress_bar.progress(i + 1)
         
         with open('model.pkl', 'rb') as file_1:
@@ -208,7 +213,7 @@ def run():
 
         status_text.text("Predicting churn...")
         for i in range(75, 100):
-            time.sleep(0.05)
+            time.sleep(0.01)
             progress_bar.progress(i + 1)
         
         churn = classification.predict(data)
@@ -227,6 +232,10 @@ def run():
             st.success("🙆 Customer is Not Gonna Churn")
             st.image('sss.png')
             st.balloons()
+            st.write('### 💭 Feedback To Marketing Team: ')
+            st.warning("""
+            The customer is not going to churn as long as we continue delivering consistent value, addressing their needs effectively, and providing exceptional service that keeps them satisfied and engaged. By maintaining open communication, offering personalized experiences, and responding quickly to any issues, we can ensure their loyalty and reduce the likelihood of churn.
+            """)
         else:
             if any(element for element in churn_bool):
                 st.error("🏃 **Customer is Gonna Churn!!**")