Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,102 +1,89 @@
 import streamlit as st
 
-# Page configuration
-st.set_page_config(
-    page_title="Fitness Plan AI Generator",
-    page_icon="💪",
-    layout="centered"
-)
+# Page Configuration
+st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
 
-# Custom CSS for better UI
+# Custom Styling
 st.markdown("""
-<style>
+    <style>
+    .main {
+        background-color: #f5f7f9;
+    }
     .stButton>button {
         width: 100%;
-        height: 50px;
-        font-size: 18px;
-        background: linear-gradient(to right, #1f4e79, #2fa4a9);
+        border-radius: 5px;
+        height: 3em;
+        background-color: #FF4B4B;
         color: white;
-        border-radius: 10px;
     }
-</style>
-""", unsafe_allow_html=True)
-
-# Title
-st.markdown("## 🧍 Your Fitness Profile")
-
-# Fitness Goal
-goal = st.selectbox(
-    "Fitness Goal",
-    ["Build Muscle", "Lose Weight", "Improve Endurance", "General Fitness"]
-)
-
-# Equipment Selection
-st.markdown("### Available Equipment")
-col1, col2, col3 = st.columns(3)
+    </style>
+    """, unsafe_allow_html=True)
 
-with col1:
-    dumbbells = st.checkbox("Dumbbells")
-with col2:
-    bands = st.checkbox("Resistance Bands")
-with col3:
-    no_equipment = st.checkbox("No Equipment")
+st.title("💪 FitPlan AI")
+st.subheader("Milestone 1: Fitness Profile & BMI Analysis")
+st.write("Complete the form below to begin your personalized fitness journey.")
 
-# Fitness Level
-st.markdown("### Fitness Level")
-level = st.radio(
-    "",
-    ["Beginner", "Intermediate", "Advanced"],
-    horizontal=True
-)
+# --- Form Section ---
+with st.form("fitness_form"):
+    st.header("1. Personal Information")
+    name = st.text_input("Full Name*", placeholder="Enter your name")
+    
+    col1, col2 = st.columns(2)
+    with col1:
+        height = st.number_input("Height (cm)*", min_value=0.0, step=0.1, help="e.g., 175")
+    with col2:
+        weight = st.number_input("Weight (kg)*", min_value=0.0, step=0.1, help="e.g., 70")
 
-# Generate Plan Button
-if st.button("Generate Personalised Plan"):
-    st.markdown("---")
-    st.markdown("## 🏋️ Your Personalised Fitness Plan")
+    st.divider()
+    
+    st.header("2. Fitness Details")
+    goal = st.selectbox("Fitness Goal", 
+                        ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
+    
+    level = st.select_slider("Fitness Level", 
+                             options=["Beginner", "Intermediate", "Advanced"])
+    
+    equipment = st.multiselect("Available Equipment", 
+                               ["Dumbbells", "Resistance Band", "Yoga Mat", "Kettlebell", "Pull-up Bar", "No Equipment"],
+                               default=["No Equipment"])
 
-    equipment = []
-    if dumbbells:
-        equipment.append("Dumbbells")
-    if bands:
-        equipment.append("Resistance Bands")
-    if no_equipment or not equipment:
-        equipment.append("Bodyweight")
+    submit_button = st.form_submit_button("Generate Profile")
 
-    st.write(f"**Goal:** {goal}")
-    st.write(f"**Level:** {level}")
-    st.write(f"**Equipment:** {', '.join(equipment)}")
-
-    st.markdown("### 📅 Weekly Workout Plan")
-
-    if goal == "Build Muscle":
-        st.write("""
-        - **Day 1:** Chest & Triceps  
-        - **Day 2:** Back & Biceps  
-        - **Day 3:** Legs  
-        - **Day 4:** Shoulders & Core  
-        - **Day 5:** Full Body  
-        - **Day 6:** Active Recovery  
-        - **Day 7:** Rest
-        """)
-    elif goal == "Lose Weight":
-        st.write("""
-        - **Day 1:** Full Body HIIT  
-        - **Day 2:** Cardio + Core  
-        - **Day 3:** Strength Training  
-        - **Day 4:** HIIT  
-        - **Day 5:** Cardio  
-        - **Day 6:** Yoga / Stretching  
-        - **Day 7:** Rest
-        """)
+# --- Logic & Results ---
+if submit_button:
+    # 5. Input Validation
+    if not name.strip():
+        st.error("Please enter your name.")
+    elif height <= 0 or weight <= 0:
+        st.error("Height and weight must be greater than zero.")
     else:
-        st.write("""
-        - **Day 1:** Full Body  
-        - **Day 2:** Cardio  
-        - **Day 3:** Strength  
-        - **Day 4:** Mobility  
-        - **Day 5:** Mixed Training  
-        - **Day 6:** Light Cardio  
-        - **Day 7:** Rest
-        """)
+        # 3. Functional Requirements
+        height_m = height / 100  # Convert cm to meters
+        bmi = weight / (height_m ** 2)
+        bmi_rounded = round(bmi, 2)
+
+        # Classify BMI
+        if bmi < 18.5:
+            category = "Underweight"
+            color = "blue"
+        elif 18.5 <= bmi < 24.9:
+            category = "Normal"
+            color = "green"
+        elif 25.0 <= bmi < 29.9:
+            category = "Overweight"
+            color = "orange"
+        else:
+            category = "Obese"
+            color = "red"
 
-    st.success("✅ Plan generated successfully!")
+        # 4. Display Results
+        st.success(f"Profile Created Successfully for **{name}**!")
+        
+        res_col1, res_col2 = st.columns(2)
+        with res_col1:
+            st.metric("Your BMI", bmi_rounded)
+        with res_col2:
+            st.markdown(f"**Category:** :{color}[{category}]")
+        
+        st.info(f"**Goal:** {goal} | **Level:** {level}")
+        st.write(f"**Equipment:** {', '.join(equipment)}")