Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,16 +1,61 @@
 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"
-    ]
-)
-
-st.write("You selected:", goal)
+st.set_page_config(page_title="AI Fitness Plan Generator", page_icon="💪", layout="centered")
+
+st.markdown("""
+<style>
+.main {
+    background: linear-gradient(135deg, #0f2027, #203a43, #2c5364);
+}
+label, .stTextInput label, .stSelectbox label {
+    color: white !important;
+}
+</style>
+""", unsafe_allow_html=True)
+
+st.title("💪 AI Fitness Plan Generator")
+st.write("Get a personalized fitness plan based on your details")
+
+# User Inputs
+age = st.number_input("Age", min_value=10, max_value=100, step=1)
+gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+goal = st.selectbox("Fitness Goal", ["Weight Loss", "Muscle Gain", "General Fitness"])
+activity = st.selectbox("Activity Level", ["Low", "Medium", "High"])
+
+# Plan Logic
+
+def generate_plan(age, goal, activity):
+    plan = ""
+
+    if goal == "Weight Loss":
+        plan = "🏃 Cardio 4–5 days/week, strength training 2–3 days/week, calorie-controlled diet."
+    elif goal == "Muscle Gain":
+        plan = "🏋️ Strength training 4–5 days/week, high-protein diet, proper sleep and recovery."
+    else:
+        plan = "🤸 Combination of cardio & strength training 3–4 days/week, balanced nutrition."
+
+    if activity == "Low":
+        plan += " Start with light intensity and gradually increase."
+    elif activity == "Medium":
+        plan += " Maintain moderate intensity workouts."
+    else:
+        plan += " Include high-intensity workouts with proper rest days."
+
+    return plan
+
+# Button
+if st.button("Generate Fitness Plan"):
+    plan = generate_plan(age, goal, activity)
+    st.success("Your Personalized Fitness Plan")
+    st.markdown(f"""
+    **Age:** {age}  
+    **Gender:** {gender}  
+    **Goal:** {goal}  
+    **Activity Level:** {activity}
+
+    ---
+    {plan}
+    """)
+
+st.markdown("---")
+st.caption("🚀 Built with Streamlit | Ready for AI integration")