import streamlit as st

# ---------- IMPORT YOUR FILES ----------
from model_api import query_model
from prompt_builder import build_prompt, calculate_bmi, bmi_category

# ---------- STREAMLIT UI ----------
st.title("🏋️ AI Personalized 5-Day Workout Planner")

name = st.text_input("Name")
age = st.number_input("Age", min_value=0, max_value=100)
gender = st.selectbox("Gender", ["Male", "Female", "Other"])

height = st.number_input("Height (cm)", min_value=0, max_value=250)
weight = st.number_input("Weight (kg)", min_value=0, max_value=200)

goal = st.selectbox(
    "Fitness Goal",
    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
)

fitness_level = st.selectbox(
    "Fitness Level",
    ["Beginner", "Intermediate", "Advanced"]
)

equipment = st.multiselect(
    "Available Equipment",
    [
        "Dumbbells",
        "Resistance Band",
        "Yoga Mat",
        "Skipping Rope",
        "Weight Plates",
        "Cycling",
        "Inclined Bench",
        "Pullups Bar",
        "No Equipment"
    ]
)

# ---------- GENERATE PLAN ----------
if st.button("Generate 5-Day Plan 💪"):

    prompt, bmi, bmi_status = build_prompt(
        name,
        age,
        gender,
        height,
        weight,
        goal,
        fitness_level,
        equipment
    )

    st.subheader(f"Your BMI: {bmi:.2f} ({bmi_status})")

    with st.spinner("Creating your personalized workout plan..."):
        result = query_model(prompt)

    st.markdown("## 🗓️ Your 5-Day Workout Plan")
    st.write(result)