Create app.py

app.py

@@ -0,0 +1,115 @@
+import gradio as gr
+import pandas as pd
+import joblib
+from sklearn.pipeline import Pipeline
+from sklearn.impute import SimpleImputer
+from sklearn.compose import ColumnTransformer
+from sklearn.preprocessing import StandardScaler, OneHotEncoder
+from sklearn.linear_model import LogisticRegression
+
+# Load the saved full pipeline from the file
+full_pipeline = joblib.load('pipe.pkl')
+
+# Define the predict function
+def predict(gender, SeniorCitizen, Partner, Dependents, Contract, tenure, MonthlyCharges,
+            TotalCharges, PaymentMethod, PhoneService, MultipleLines, InternetService,
+            OnlineSecurity, OnlineBackup, DeviceProtection, TechSupport, StreamingTV,
+            StreamingMovies, PaperlessBilling):
+    # Create a DataFrame from the input data
+    input_data = pd.DataFrame({
+        'gender': [gender] if gender else ['Male'],  # Replace None with default value
+        'SeniorCitizen': [SeniorCitizen] if SeniorCitizen is not None else [0],  # Replace None with default value
+        'Partner': [Partner] if Partner else ['No'],  # Replace None with default value
+        'Dependents': [Dependents] if Dependents else ['No'],  # Replace None with default value
+        'tenure': [tenure] if tenure else [1],  # Replace None with default value
+        'PhoneService': [PhoneService] if PhoneService else ['Yes'],  # Replace None with default value
+        'MultipleLines': [MultipleLines] if MultipleLines else ['No'],  # Replace None with default value
+        'InternetService': [InternetService] if InternetService else ['DSL'],  # Replace None with default value
+        'OnlineSecurity': [OnlineSecurity] if OnlineSecurity else ['No'],  # Replace None with default value
+        'OnlineBackup': [OnlineBackup] if OnlineBackup else ['No'],  # Replace None with default value
+        'DeviceProtection': [DeviceProtection] if DeviceProtection else ['No'],  # Replace None with default value
+        'TechSupport': [TechSupport] if TechSupport else ['No'],  # Replace None with default value
+        'StreamingTV': [StreamingTV] if StreamingTV else ['No'],  # Replace None with default value
+        'StreamingMovies': [StreamingMovies] if StreamingMovies else ['No'],  # Replace None with default value
+        'Contract': [Contract] if Contract else ['Month-to-month'],  # Replace None with default value
+        'PaperlessBilling': [PaperlessBilling] if PaperlessBilling else ['No'],  # Replace None with default value
+        'PaymentMethod': [PaymentMethod] if PaymentMethod else ['Electronic check'],  # Replace None with default value
+        'MonthlyCharges': [MonthlyCharges] if MonthlyCharges else [0.0],  # Replace None with default value
+        'TotalCharges': [TotalCharges] if TotalCharges else [0.0]  # Replace None with default value
+    })
+
+
+        # Make predictions using the loaded logistic regression model
+    predictions = full_pipeline.predict(input_data)
+
+    #return predictions[0]
+    if predictions[0] == "Yes":
+        return "Churn"
+    else:
+        return "Not Churn"
+
+# Setting Gradio App Interface
+with gr.Blocks(css=".gradio-container {background-color: grey}") as demo:
+    gr.Markdown("# Teleco Customer Churn Prediction #\n*This App allows the user to predict whether a customer will churn or not by entering values in the given fields. Any field left blank takes the default value.*")
+    
+    # Receiving ALL Input Data here
+    gr.Markdown("**Demographic Data**")
+    with gr.Row():
+        gender = gr.Dropdown(label="Gender", choices=["Male", "Female"])
+        SeniorCitizen = gr.Radio(label="Senior Citizen", choices=[1, 0])
+        Partner = gr.Radio(label="Partner", choices=["Yes", "No"])
+        Dependents = gr.Radio(label="Dependents", choices=["Yes", "No"])
+
+    gr.Markdown("**Service Length and Charges (USD)**")
+    with gr.Row():
+        Contract = gr.Dropdown(label="Contract", choices=["Month-to-month", "One year", "Two year"])
+        tenure = gr.Slider(label="Tenure (months)", minimum=1, step=1, interactive=True)
+        MonthlyCharges = gr.Slider(label="Monthly Charges", step=0.05)
+        TotalCharges = gr.Slider(label="Total Charges", step=0.05)
+
+    # Phone Service Usage part
+    gr.Markdown("**Phone Service Usage**")
+    with gr.Row():
+        PhoneService = gr.Radio(label="Phone Service", choices=["Yes", "No"])
+        MultipleLines = gr.Dropdown(label="Multiple Lines", choices=[
+                                    "Yes", "No", "No phone service"])
+
+    # Internet Service Usage part
+    gr.Markdown("**Internet Service Usage**")
+    with gr.Row():
+        InternetService = gr.Dropdown(label="Internet Service", choices=["DSL", "Fiber optic", "No"])
+        OnlineSecurity = gr.Dropdown(label="Online Security", choices=["Yes", "No", "No internet service"])
+        OnlineBackup = gr.Dropdown(label="Online Backup", choices=["Yes", "No", "No internet service"])
+        DeviceProtection = gr.Dropdown(label="Device Protection", choices=["Yes", "No", "No internet service"])
+        TechSupport = gr.Dropdown(label="Tech Support", choices=["Yes", "No", "No internet service"])
+        StreamingTV = gr.Dropdown(label="TV Streaming", choices=["Yes", "No", "No internet service"])
+        StreamingMovies = gr.Dropdown(label="Movie Streaming", choices=["Yes", "No", "No internet service"])
+
+    # Billing and Payment part
+    gr.Markdown("**Billing and Payment**")
+    with gr.Row():
+        PaperlessBilling = gr.Radio(
+            label="Paperless Billing", choices=["Yes", "No"])
+        PaymentMethod = gr.Dropdown(label="Payment Method", choices=["Electronic check", "Mailed check", "Bank transfer (automatic)", "Credit card (automatic)"])
+
+    # Output Prediction
+    output = gr.Text(label="Outcome")
+    submit_button = gr.Button("Predict")
+    
+    submit_button.click(fn= predict,
+                        outputs= output,
+                        inputs=[gender, SeniorCitizen, Partner, Dependents, Contract, tenure, MonthlyCharges, TotalCharges, PaymentMethod, PhoneService, MultipleLines, InternetService, OnlineSecurity, OnlineBackup, DeviceProtection, TechSupport, StreamingTV, StreamingMovies, PaperlessBilling],
+    
+    ),
+    
+    # Add the reset and flag buttons
+    
+    def clear():
+        output.value = ""
+        return None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None, None
+         
+    clear_btn = gr.Button("Reset", variant="primary")
+    clear_btn.click(fn=clear, inputs=None, outputs=output)
+        
+ 
+demo.launch(inbrowser = True)