Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,16 +1,53 @@
 import streamlit as st
 
-st.title("FIT Plan AI – Personalized Fitness Plan Generator")
-
-goal = st.selectbox(
-    "Goal",
-    [
-        "Flexibility",
-        "Weight Loss",
-        "Build Muscle",
-        "Strength Gaining",
-        "Abs Building"
-    ]
+# Page config
+st.set_page_config(page_title="Fitness Profile", page_icon="🏋️", layout="centered")
+
+# Title
+st.markdown("## 🏋️ Your Fitness Profile")
+
+st.markdown("---")
+
+# Fitness Goal
+fitness_goal = st.selectbox(
+    "Fitness Goal",
+    ["Build Muscle", "Lose Weight", "Improve Endurance", "General Fitness"]
 )
 
-st.write("You selected:", goal)
+# Available Equipment
+st.write("Available Equipment")
+col1, col2 = st.columns(2)
+
+with col1:
+    dumbbells = st.checkbox("Dumbbells")
+    no_equipment = st.checkbox("No Equipment")
+
+with col2:
+    resistance_bands = st.checkbox("Resistance Bands")
+
+# Fitness Level
+st.write("Fitness Level")
+fitness_level = st.radio(
+    "",
+    ["Beginner", "Intermediate", "Advanced"],
+    horizontal=True
+)
+
+st.markdown("")
+
+# Generate Button
+if st.button("Generate Personalized Plan", use_container_width=True):
+    st.success("Generating your personalized workout plan...")
+    
+    st.write("### Your Selected Preferences:")
+    st.write("**Goal:**", fitness_goal)
+    st.write("**Equipment:**",
+             ", ".join(
+                 [item for item, selected in {
+                     "Dumbbells": dumbbells,
+                     "Resistance Bands": resistance_bands,
+                     "No Equipment": no_equipment
+                 }.items() if selected]
+             ) or "None Selected"
+             )
+    st.write("**Fitness Level:**", fitness_level)