Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,101 +1,81 @@
 import streamlit as st
 
-st.set_page_config(page_title="AI Fitness Plan Generator", page_icon="💪", layout="wide")
-
-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"])
+st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
+
+st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
+
+st.write("Fill in your details to calculate your BMI and fitness category.")
+
+# -------------------------
+# 1. Personal Information
+# -------------------------
+
+name = st.text_input("Enter Your Name *")
+
+height_cm = st.number_input("Enter Height (in centimeters) *", min_value=0.0, format="%.2f")
+weight_kg = st.number_input("Enter Weight (in kilograms) *", min_value=0.0, format="%.2f")
+
+# -------------------------
+# 2. Fitness Details
+# -------------------------
+
+st.subheader("Fitness Details")
+
 goal = st.selectbox(
-    "Fitness Goal",
-    [
-        "Flexibility",
-        "Weight Loss",
-        "Build Muscle",
-        "Strength Gaining",
-        "Abs Building"
-    ]
-)
-fitness_level = st.selectbox(
-    "Fitness Level",
-    ["Beginner", "Intermediate", "Advanced"]
+    "Select Your Fitness Goal",
+    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
 )
 
 equipment = st.multiselect(
-    "Available Equipment",
-    [
-        "Dumbbells",
-        "Resistance Band",
-        "Yoga Mat",
-        "Inclined Bench",
-        "Treadmill",
-        "Cycle",
-        "Skipping Rope",
-        "Hand Gripper",
-        "Pull-ups Bar",
-        "Weight Plates",
-        "Hula Hoop Ring",
-        "Bosu Ball"
-    ]
+    "Available Equipment (Select multiple if available)",
+    ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
 )
 
-# Plan Logic
+fitness_level = st.radio(
+    "Select Your Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"]
+)
 
-def generate_plan(goal, fitness_level, equipment):
-    plan = ""
+# -------------------------
+# BMI Calculation Function
+# -------------------------
 
-    if goal == "Weight Loss":
-        plan = "🏃 Focus on calorie-burning workouts with cardio and light strength training."
-    elif goal == "Build Muscle":
-        plan = "🏋️ Progressive strength training with adequate protein intake."
-    elif goal == "Strength Gaining":
-        plan = "💪 Heavy compound lifts with longer rest periods."
-    elif goal == "Abs Building":
-        plan = "🔥 Core-focused workouts combined with fat-loss strategies."
-    else:
-        plan = "🧘 Flexibility and mobility routines with stretching and yoga."
+def calculate_bmi(weight, height_cm):
+    height_m = height_cm / 100  # Convert cm to meters
+    bmi = weight / (height_m ** 2)
+    return round(bmi, 2)
 
-    if fitness_level == "Beginner":
-        plan += " Start slow, focus on form, and train 3–4 days/week."
-    elif fitness_level == "Intermediate":
-        plan += " Increase intensity and volume, train 4–5 days/week."
+def bmi_category(bmi):
+    if bmi < 18.5:
+        return "Underweight"
+    elif 18.5 <= bmi < 24.9:
+        return "Normal"
+    elif 25 <= bmi < 29.9:
+        return "Overweight"
     else:
-        plan += " High-intensity training with structured programming, 5–6 days/week."
+        return "Obese"
 
-    if equipment:
-        plan += f" Use available equipment: {', '.join(equipment)}."
+# -------------------------
+# Submit Button
+# -------------------------
+
+if st.button("Calculate BMI"):
+    
+    # Validation
+    if not name or height_cm <= 0 or weight_kg <= 0:
+        st.error("⚠ Please fill all required fields with valid values!")
     else:
-        plan += " Bodyweight exercises will be emphasized."
-
-    return plan
-
-# Button
-if st.button("Generate Fitness Plan"):
-    plan = generate_plan(goal, fitness_level, equipment)
-    st.success("Your Personalized Fitness Plan")
-    st.markdown(f"""
-    **Age:** {age}  
-    **Gender:** {gender}  
-    **Goal:** {goal}  
-    **Fitness Level:** {fitness_level}  
-**Equipment:** {', '.join(equipment) if equipment else 'None'}
-
-    ---
-    {plan}
-    """)
-
-st.markdown("---")
-st.caption("🚀 Built with Streamlit | Ready for AI integration")
+        bmi = calculate_bmi(weight_kg, height_cm)
+        category = bmi_category(bmi)
+
+        st.success("✅ Calculation Successful!")
+
+        st.write(f"### 👤 Name: {name}")
+        st.write(f"### 📊 Your BMI: {bmi}")
+        st.write(f"### 🏷 BMI Category: {category}")
+
+        st.write("---")
+        st.write("### 🏋 Fitness Summary")
+        st.write(f"**Goal:** {goal}")
+        st.write(f"**Fitness Level:** {fitness_level}")
+        st.write(f"**Equipment Available:** {', '.join(equipment) if equipment else 'None selected'}")