Update src/app.py

src/app.py

@@ -8,11 +8,11 @@ st.set_page_config(
 
 st.title("🎓 ExtraaLearn Lead Conversion Prediction")
 
-st.write("Enter lead details to predict conversion likelihood.")
+st.write("Fill in the lead details below to predict conversion likelihood.")
 
 # ---------------- INPUT FIELDS ---------------- #
 
-age = st.number_input("Age", min_value=18, max_value=70, value=28)
+age = st.number_input("Age", min_value=18, max_value=80, value=57)
 
 current_occupation = st.selectbox(
     "Current Occupation",
@@ -20,38 +20,56 @@ current_occupation = st.selectbox(
 )
 
 first_interaction = st.selectbox(
-    "First Interaction Channel",
+    "First Interaction",
     ["Website", "Email", "Referral", "Social Media"]
 )
 
 profile_completed = st.selectbox(
-    "Profile Completed?",
-    ["Yes", "No"]
+    "Profile Completion Level",
+    ["Low", "Medium", "High"]
 )
 
 website_visits = st.number_input(
-    "Website Visits", min_value=0, max_value=100, value=5
+    "Website Visits", min_value=0, max_value=100, value=7
 )
 
 time_spent_on_website = st.number_input(
-    "Time Spent on Website (minutes)", min_value=0.0, value=10.0
+    "Time Spent on Website (seconds)", min_value=0, value=1639
 )
 
 page_views_per_visit = st.number_input(
-    "Page Views per Visit", min_value=0.0, value=3.5
+    "Page Views per Visit", min_value=0.0, value=1.861
 )
 
 last_activity = st.selectbox(
     "Last Activity",
-    ["Email Opened", "SMS Clicked", "Website Visit", "Form Submitted"]
+    ["Website Activity", "Email Opened", "SMS Clicked", "Form Submitted"]
+)
+
+print_media_type1 = st.selectbox(
+    "Print Media Type 1",
+    ["Yes", "No"]
 )
 
-# Binary media flags (VERY IMPORTANT)
-print_media_type1 = st.checkbox("Print Media Type 1")
-print_media_type2 = st.checkbox("Print Media Type 2")
-digital_media = st.checkbox("Digital Media")
-educational_channels = st.checkbox("Educational Channels")
-referral = st.checkbox("Referral")
+print_media_type2 = st.selectbox(
+    "Print Media Type 2",
+    ["Yes", "No"]
+)
+
+digital_media = st.selectbox(
+    "Digital Media",
+    ["Yes", "No"]
+)
+
+educational_channels = st.selectbox(
+    "Educational Channels",
+    ["Yes", "No"]
+)
+
+referral = st.selectbox(
+    "Referral",
+    ["Yes", "No"]
+)
 
 # ---------------- PREDICT ---------------- #
 
@@ -62,21 +80,20 @@ if st.button("Predict Conversion"):
                 "age": age,
                 "current_occupation": current_occupation,
                 "first_interaction": first_interaction,
-                "profile_completed": 1 if profile_completed == "Yes" else 0,
+                "profile_completed": profile_completed,
                 "website_visits": website_visits,
                 "time_spent_on_website": time_spent_on_website,
                 "page_views_per_visit": page_views_per_visit,
                 "last_activity": last_activity,
-                "print_media_type1": int(print_media_type1),
-                "print_media_type2": int(print_media_type2),
-                "digital_media": int(digital_media),
-                "educational_channels": int(educational_channels),
-                "referral": int(referral)
+                "print_media_type1": 1 if print_media_type1 == "Yes" else 0,
+                "print_media_type2": 1 if print_media_type2 == "Yes" else 0,
+                "digital_media": 1 if digital_media == "Yes" else 0,
+                "educational_channels": 1 if educational_channels == "Yes" else 0,
+                "referral": 1 if referral == "Yes" else 0
             }
         ]
     }
 
-    # 🔴 Replace with your actual backend URL
     BACKEND_URL = "https://rohitmv-extra-learn-be.hf.space/predict"
 
     try:
@@ -92,11 +109,10 @@ if st.button("Predict Conversion"):
                 f"{int(result['conversion_probability'] * 100)}%"
             )
 
-            st.write("### Prediction")
             st.write(
-                "✅ Likely to Convert"
+                "### ✅ Likely to Convert"
                 if result["prediction"] == 1
-                else "❌ Unlikely to Convert"
+                else "### ❌ Unlikely to Convert"
             )
 
             st.write(f"**Lead Category:** {result['lead_category']}")