Upload predict.py

predict.py

@@ -55,25 +55,44 @@ def run():
     st.title("🏃 Customer Churn Prediction")
     st.markdown('---')
 
+    st.sidebar.markdown("---")
+    
+    st.sidebar.write("### BERT Model")
+    st.sidebar.write("This application uses a BERT model to analyze the sentiment of customer feedback. BERT is highly effective at understanding context and nuances in text, even for long feedback, despite being trained on shorter sentences in our dataset.")
+    st.sidebar.image("bert.jpg", width=200)
+    st.sidebar.write("BERT (Bidirectional Encoder Representations from Transformers) works by understanding the context of words in a sentence by looking at the words that come before and after them. This bidirectional approach allows BERT to capture the meaning of words more accurately.")
+    
+    st.sidebar.markdown('---')  # Line separating between models
+    
+    st.sidebar.write("### SVM Classifier")
+    st.sidebar.write("After sentiment analysis, an SVM classifier is used to predict whether a customer is likely to churn based on the sentiment and other features provided.")
+    st.sidebar.image("svm.png", width=200)
+    st.sidebar.write("SVM (Support Vector Machine) works by finding the hyperplane that best separates the data into different classes. In this case, it uses the sentiment and other features to predict customer churn.")
+
     st.write("## ✍🏻📑 Input Data: ")
     st.write("")  # Add a blank line for spacing
+    st.write("This prediction will be done by first analyzing the sentiment from the feedback provided using a BERT model. The sentiment, along with other features, will then be used to classify whether the customer is likely to churn or not churn using an SVM classifier.")
     with st.form(key="data"):
         col1, col2 = st.columns(2)
         with col1:
-            customer_id = st.text_input("Customer ID")
-            tenure = st.number_input("Tenure")
+            customer_id = st.text_input("Customer ID", value="CUST001")
+            tenure = st.number_input("Tenure (in months)", value=12)
         with col2:
             contract = st.selectbox(
-                "contract", ['one year', 'month-to-month', 'two year']
+                "contract", ['one year', 'month-to-month', 'two year'], index=1
             )
             payment_method = st.selectbox(
-                "payment_method", ['credit card', 'electronic check', 'bank transfer', 'mailed check']
+                "payment_method", ['credit card', 'electronic check', 'bank transfer', 'mailed check'],index=0
             )
-        monthly_charges = st.number_input("Monthly charges")
-        total_charges = st.number_input("Total charges")
-        feedback = st.text_area("Feedback")
+        monthly_charges = st.number_input("Monthly charges (in $)",step=1.00)
+        total_charges = st.number_input("Total charges (in $)",step=1.00)
+        feedback = st.text_area(
+            "Feedback", 
+            value="As an event planner, I've worked with many florists, but this one stands out from the crowd. For a recent high-profile corporate gala, they provided centerpieces that were nothing short of spectacular. The creativity and artistry in their designs elevated the entire event. They listened carefully to our theme and color scheme, then created arrangements that perfectly captured the essence of the evening. The flowers were fresh, vibrant, and lasted throughout the night, even under bright lights and air conditioning. What impressed me most was their flexibility - when we needed last-minute changes due to a shift in table arrangements, they accommodated without hesitation. Their team was professional, punctual, and a joy to work with. They arrived well before the event to set up and stayed until everything was perfect. The value for money was excellent, considering the high-quality results and the level of service provided. I've already booked them for several upcoming events, including a charity fundraiser and a product launch. It's rare to find a vendor that consistently exceeds expectations, but this florist does just that.",
+            height=300
+        )
         topic = st.selectbox(
-            "topic", ['bouquet preferences', 'delivery issues', 'general feedback', 'price complaints', 'delivery quality', 'product quality', 'customer service', 'price appreciation']
+            "topic", ['bouquet preferences', 'delivery issues', 'general feedback', 'price complaints', 'delivery quality', 'product quality', 'customer service', 'price appreciation'], index=2
         )
         # Submit button
         submit = st.form_submit_button("🔘 Predict")
@@ -141,4 +160,4 @@ def run():
             st.balloons()
 
 if __name__ == "__main__":
-    run()
+    run()