import streamlit as st
from model_api import query_model
from prompt_builder import build_prompt

st.set_page_config(page_title="FitPlan AI", layout="centered")

st.title("🏋️ FitPlan AI – Personalized Workout Generator")

# ---------------- INPUT ---------------- #

name = st.text_input("Enter Your Name")

gender = st.radio("Gender", ["Male", "Female"])

height = st.number_input("Height (cm)", min_value=100.0, max_value=250.0)
weight = st.number_input("Weight (kg)", min_value=30.0, max_value=200.0)

goal = st.selectbox(
    "Fitness Goal",
    ["Build Muscle", "Lose Weight", "Improve Endurance", "General Fitness"]
)

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

equipment = st.multiselect(
    "Available Equipment",
    [
        "No Equipment", "Dumbbells", "Barbell",
        "Pull-up Bar", "Resistance Bands",
        "Treadmill", "Kettlebells", "Full Gym"
    ]
)

# ---------------- GENERATE ---------------- #

if st.button("Generate Workout Plan"):

    if height > 0 and weight > 0:

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

        with st.spinner("Generating your personalized plan..."):
            response = query_model(prompt)

        st.subheader("📋 Your Personalized Workout Plan")
        st.write(response)

        st.info(f"Calculated BMI: {bmi:.2f} ({bmi_status})")

    else:
        st.warning("Please enter valid height and weight.")