import streamlit as st

# Set up the title and description
st.title("Hyper-Pump AI")
st.markdown("""
Welcome to the Workout Planner!  
Please select or input the muscle group you'd like to work out, and we'll suggest exercises for you.
""")

# Dropdown menu for muscle group selection
muscle_groups = [
    "Chest",
    "Back",
    "Legs",
    "Arms",
    "Shoulders",
    "Core",
    "Full Body",
]

selected_group = st.selectbox("Choose a muscle group:", muscle_groups)

# Option to input a custom muscle group
custom_group = st.text_input("Or enter a custom muscle group:")

# Display the selected or custom muscle group
if st.button("Submit"):
    if custom_group:
        st.success(f"You selected a custom muscle group: {custom_group}")
    else:
        st.success(f"You selected: {selected_group}")

st.markdown("---")
st.markdown("This app is powered by Streamlit and hosted on Hugging Face Spaces!")