inputs_page.py

import streamlit as st
from inputs.income import show_income_inputs
from inputs.assets import show_assets_inputs


def show_net_worth_breakdown(prefix):
    return show_assets_inputs(prefix)


def show_input_page():
    st.title("UK Financial Planning Tool")

    # Sidebar for inputs
    st.sidebar.header("Personal Information")
    inflation_rate = st.sidebar.number_input(
        "Inflation Rate (%)", value=st.session_state.user_inputs["inflation_rate"] * 100
    ) or 0
    st.session_state.user_inputs["inflation_rate"] = inflation_rate / 100

    has_partner = st.sidebar.checkbox(
        "Do you have a partner?",
        value=st.session_state.user_inputs.get("has_partner", False),
        key="has_partner_checkbox",
    )
    st.session_state.user_inputs["has_partner"] = has_partner

    st.session_state.age_inputs["current_age"] = st.sidebar.number_input(
        "Current Age",
        min_value=18,
        max_value=100,
        value=st.session_state.age_inputs["current_age"],
    )
    st.session_state.age_inputs["retirement_age"] = st.sidebar.number_input(
        "Retirement Age",
        min_value=st.session_state.age_inputs["current_age"] + 1,
        max_value=100,
        value=st.session_state.age_inputs["retirement_age"],
    )
    st.session_state.age_inputs["life_expectancy"] = st.sidebar.number_input(
        "Life Expectancy",
        min_value=st.session_state.age_inputs["retirement_age"],
        max_value=120,
        value=st.session_state.age_inputs["life_expectancy"],
    )

    if has_partner:
        st.session_state.age_inputs["partner_current_age"] = st.sidebar.number_input(
            "Partner's Current Age",
            min_value=18,
            max_value=100,
            value=st.session_state.age_inputs["partner_current_age"],
        )
        st.session_state.age_inputs["partner_retirement_age"] = st.sidebar.number_input(
            "Partner's Retirement Age",
            min_value=st.session_state.age_inputs["partner_current_age"] + 1,
            max_value=100,
            value=st.session_state.age_inputs["partner_retirement_age"],
        )
        st.session_state.age_inputs["partner_life_expectancy"] = (
            st.sidebar.number_input(
                "Partner's Life Expectancy",
                min_value=st.session_state.age_inputs["partner_retirement_age"],
                max_value=120,
                value=st.session_state.age_inputs["partner_life_expectancy"],
            )
        )

    show_income_inputs()
    show_assets_inputs("user")
    if has_partner:
        show_assets_inputs("partner")