Update src/streamlit_app.py

src/streamlit_app.py

@@ -34,6 +34,7 @@ st.markdown("Predict if a customer will purchase the Wellness Tourism Package")
 with st.form("customer_form"):
     col1, col2 = st.columns(2)
 
+    # Left column
     with col1:
         age = st.number_input("Age", min_value=18, max_value=80, value=35)
         gender = st.selectbox("Gender", ["Male", "Female"])
@@ -43,7 +44,11 @@ with st.form("customer_form"):
         occupation = st.selectbox(
             "Occupation", ["Salaried", "Employee", "Freelancer", "Small Business"]
         )
+        designation = st.selectbox(
+            "Designation", ["Executive", "Manager", "Senior Manager", "AVP", "VP"]
+        )
 
+    # Right column
     with col2:
         city_tier = st.selectbox("City Tier", [1, 2, 3])
         typeofcontact = st.selectbox(
@@ -51,8 +56,13 @@ with st.form("customer_form"):
         )
         passport = st.selectbox("Has Passport", [0, 1])
         own_car = st.selectbox("Owns Car", [0, 1])
+        product_pitched = st.selectbox(
+            "Product Pitched",
+            ["Basic", "Standard", "Deluxe", "Super Deluxe", "King"],
+        )
 
     col3, col4 = st.columns(2)
+
     with col3:
         num_person_visiting = st.number_input(
             "Number of People Visiting", min_value=1, max_value=10, value=2
@@ -82,22 +92,27 @@ with st.form("customer_form"):
 
 if submit:
     input_data = {
+        # Dummy index column expected by the model
+        "Unnamed: 0": [0],
         "Age": [age],
         "TypeofContact": [typeofcontact],
         "CityTier": [city_tier],
+        "DurationOfPitch": [duration_of_pitch],
         "Occupation": [occupation],
         "Gender": [gender],
         "NumberOfPersonVisiting": [num_person_visiting],
+        "NumberOfFollowups": [number_of_followups],
+        "ProductPitched": [product_pitched],
         "PreferredPropertyStar": [preferred_property_star],
         "MaritalStatus": [marital_status],
         "NumberOfTrips": [num_trips],
         "Passport": [passport],
+        "PitchSatisfactionScore": [pitch_satisfaction_score],
         "OwnCar": [own_car],
         "NumberOfChildrenVisiting": [num_children_visiting],
-        "PitchSatisfactionScore": [pitch_satisfaction_score],
+        "Designation": [designation],
         "MonthlyIncome": [monthly_income],
-        "DurationOfPitch": [duration_of_pitch],
-        "NumberOfFollowups": [number_of_followups],
+        # Engineered features (also used in training)
         "ChildrenRatio": [num_children_visiting / max(num_person_visiting, 1)],
         "IncomePerPerson": [monthly_income / max(num_person_visiting, 1)],
     }