import streamlit as st
import pandas as pd

def main():
    st.title("Data Collection Form")

    sex = st.radio("Sex", ["Female", "Male"])
    age = st.number_input("Age", min_value=0)
    hypertension = st.selectbox("Hypertension", [0, 1])
    heart_disease = st.selectbox("Heart Disease", [0, 1])
    ever_married = st.selectbox("Ever Married", [0, 1])
    work_type = st.selectbox("Work Type", ["Never_worked", "children", "Govt_job", "Self-employed", "Private"])
    residence_type = st.selectbox("Residence Type", [0, 1])
    avg_glucose_level = st.number_input("Average Glucose Level", min_value=0.0)
    bmi = st.number_input("BMI", min_value=0.0)
    smoking_status = st.selectbox("Smoking Status", ["never smoked", "smokes"])

    submit_button = st.button("Submit")

    if submit_button:
        data = {
            "sex": 1 if sex == "Male" else 0,
            "age": age,
            "hypertension": hypertension,
            "heart_disease": heart_disease,
            "ever_married": ever_married,
            "work_type": ["Never_worked", "children", "Govt_job", "Self-employed", "Private"].index(work_type),
            "residence_type": residence_type,
            "avg_glucose_level": avg_glucose_level,
            "bmi": bmi,
            "smoking_status": ["never smoked", "smokes"].index(smoking_status)
        }

        df = pd.DataFrame([data])
        st.write("Submitted Data:")
        st.write(df)

if __name__ == "__main__":
    main()