import streamlit as st
import requests

st.title("Lead Conversion Prediction")

st.write("Enter lead details:")

age = st.number_input("Age", min_value=18, max_value=100, value=25)
website_visits = st.number_input("Website Visits", min_value=0, value=3)
time_spent = st.number_input("Time Spent on Website (seconds)", min_value=0, value=200)

if st.button("Predict"):
    
    url = "https://killswitch009-lead-conversion-api.hf.space/predict"
    
    data = {
        "data": [age, website_visits, time_spent]
    }

    response = requests.post(url, json=data)

    if response.status_code == 200:
        result = response.json()["prediction"]

        if result == 1:
            st.success("This lead is likely to convert ✅")
        else:
            st.error("This lead is unlikely to convert ❌")
    else:
        st.error("Error connecting to API")