main.py

"""
main.py — Streamlit UI for the SchemeEligibilityAgent.

Run with:
    streamlit run main.py
"""

import streamlit as st
from agent import SchemeEligibilityAgent
from logger import AgentLogger


# ---------------------------------------------------------------------------
# Dropdown option constants
# ---------------------------------------------------------------------------

GENDER_OPTIONS = ["Male", "Female", "Other"]

INCOME_OPTIONS = [
    "Below 1 Lakh",
    "1-2.5 Lakh",
    "2.5-5 Lakh",
    "5-8 Lakh",
    "8-10 Lakh",
    "Above 10 Lakh",
]

EDUCATION_OPTIONS = [
    "Below 10th",
    "10th Pass",
    "12th Pass",
    "Diploma/ITI",
    "Graduate",
    "Post Graduate",
    "PhD",
]

CATEGORY_OPTIONS = ["General", "OBC", "SC", "ST", "EWS", "Minority"]

EMPLOYMENT_OPTIONS = [
    "Unemployed",
    "Student",
    "Self-Employed",
    "Salaried",
    "Daily Wage",
    "Farmer",
]

STATE_OPTIONS = [
    "Any",
    "Andhra Pradesh",
    "Arunachal Pradesh",
    "Assam",
    "Bihar",
    "Chhattisgarh",
    "Delhi",
    "Goa",
    "Gujarat",
    "Haryana",
    "Himachal Pradesh",
    "Jharkhand",
    "Karnataka",
    "Kerala",
    "Madhya Pradesh",
    "Maharashtra",
    "Manipur",
    "Meghalaya",
    "Mizoram",
    "Nagaland",
    "Odisha",
    "Punjab",
    "Rajasthan",
    "Sikkim",
    "Tamil Nadu",
    "Telangana",
    "Tripura",
    "Uttar Pradesh",
    "Uttarakhand",
    "West Bengal",
]


# ---------------------------------------------------------------------------
# Page config
# ---------------------------------------------------------------------------

st.set_page_config(
    page_title="Gov Scheme Eligibility Agent",
    layout="centered",
)


# ---------------------------------------------------------------------------
# Sidebar
# ---------------------------------------------------------------------------

with st.sidebar:
    st.header("About")
    st.write(
        "**Gov Scheme Eligibility Agent** checks your eligibility "
        "for Indian government schemes based on your profile. "
        "It is a deterministic AI agent — not a chatbot."
    )
    st.divider()
    st.subheader("How it works")
    st.markdown(
        "1. **Enter** your details\n"
        "2. Agent **plans** the analysis\n"
        "3. Agent **loads** scheme database\n"
        "4. Agent **checks** each scheme's rules\n"
        "5. Agent **records** decisions with reasons\n"
        "6. Results displayed with **full transparency**"
    )
    st.divider()
    st.caption(
        "Fully deterministic. No LLM calls. No internet required."
    )


# ---------------------------------------------------------------------------
# Main area
# ---------------------------------------------------------------------------

st.title("Gov Scheme Eligibility Agent")

# --- User input form ---
st.subheader("Enter Your Details")

col1, col2 = st.columns(2)

with col1:
    age = st.slider("Age", min_value=0, max_value=100, value=25)
    income_range = st.selectbox("Annual Income Range", INCOME_OPTIONS, index=1)
    category = st.selectbox("Social Category", CATEGORY_OPTIONS)

with col2:
    gender = st.selectbox("Gender", GENDER_OPTIONS)
    education = st.selectbox("Education Level", EDUCATION_OPTIONS, index=2)
    employment_status = st.selectbox("Employment Status", EMPLOYMENT_OPTIONS)

state = st.selectbox("State", STATE_OPTIONS)


# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------

if st.button("Check Eligibility", type="primary"):

    user_profile = {
        "age": age,
        "gender": gender,
        "income_range": income_range,
        "education": education,
        "category": category,
        "employment_status": employment_status,
        "state": state,
    }

    # --- Create logger and agent ---
    logger = AgentLogger()
    agent = SchemeEligibilityAgent(
        user_profile=user_profile,
        logger=logger,
    )

    # --- Execute ---
    with st.spinner("Agent is analyzing your eligibility..."):
        results = agent.run()

    # --- Agent Plan (expanded) ---
    plan_logs = [l for l in logger.get_logs() if l.startswith("[PLAN]")]
    with st.expander("Agent Plan", expanded=True):
        for log_line in plan_logs:
            st.markdown(f"`{log_line}`")

    # --- Full Execution Log (collapsed) ---
    with st.expander("Agent Execution Log", expanded=False):
        for log_line in logger.get_logs():
            if log_line.startswith("[PLAN]"):
                st.markdown(f"`{log_line}`")
            elif log_line.startswith("[TOOL CALL]"):
                st.markdown(f"`{log_line}`")
            elif log_line.startswith("[OBSERVATION]"):
                st.markdown(f"`{log_line}`")
            else:
                st.text(log_line)

    st.divider()

    # --- Summary metrics ---
    st.subheader("Results Summary")
    total = results.get("total_schemes", 0)
    eligible_list = results.get("eligible", [])
    rejected_list = results.get("rejected", [])

    m1, m2, m3 = st.columns(3)
    m1.metric("Total Schemes", total)
    m2.metric("Eligible", len(eligible_list))
    m3.metric("Not Eligible", len(rejected_list))

    st.divider()

    # --- Eligible schemes ---
    if eligible_list:
        st.subheader("Eligible Schemes")
        for scheme in eligible_list:
            st.success(f"**{scheme['name']}** ({scheme['category']})")
            st.write(scheme["description"])
            st.markdown("**Benefits:**")
            for benefit in scheme["benefits"]:
                st.markdown(f"- {benefit}")
            with st.expander(f"Why you qualify for {scheme['name']}"):
                for reason in scheme["reasons"]:
                    if reason.startswith("Passed:"):
                        st.markdown(f"- {reason}")
    else:
        st.info("No eligible schemes found for your profile.")

    st.divider()

    # --- Rejected schemes (collapsed) ---
    with st.expander("Rejected Schemes (click to expand)", expanded=False):
        if rejected_list:
            for scheme in rejected_list:
                st.warning(
                    f"**{scheme['name']}** ({scheme['category']})"
                )
                for reason in scheme["rejection_reasons"]:
                    st.markdown(f"- {reason}")
        else:
            st.write("All schemes matched!")