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

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

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

# User Inputs
goal = st.selectbox("Select Your Goal", 
                    ["Build Muscle", "Lose Weight", "Improve Endurance"])

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

equipment = st.selectbox("Available Equipment",
                         ["No Equipment", "Dumbbells", "Full Gym"])

if st.button("Generate Plan"):

    with st.spinner("Generating your personalized workout plan..."):
        prompt = build_prompt(goal, fitness_level, equipment)
        response = query_model(prompt)

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