Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,6 +6,7 @@ from transformers import pipeline
 # ---------------------------------------------------
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
+
 @st.cache_resource
 def load_model():
     return pipeline(
@@ -15,7 +16,10 @@ def load_model():
 
 generator = load_model()
 
-st.title(" FitPlan AI – User Fitness Profile")
+# ---------------------------------------------------
+# TITLE
+# ---------------------------------------------------
+st.title("💪 FitPlan AI – User Fitness Profile")
 
 # ---------------------------------------------------
 # PERSONAL INFORMATION
@@ -24,50 +28,62 @@ st.subheader("👤 Personal Information")
 
 name = st.text_input("Enter Your Name")
 
-st.subheader("Gender")
-
 gender = st.radio(
-    "",
+    "Gender",
     ["Male", "Female"],
-    horizontal=True,index=0
+    horizontal=True
 )
 
-
+# ---------------------------------------------------
+# HEIGHT & WEIGHT
+# ---------------------------------------------------
 col1, col2 = st.columns(2)
 
 with col1:
-    height = st.number_input("Height (in cm)", min_value=100, max_value=250)
+    height = st.number_input(
+        "Height (in cm)",
+        min_value=0.0,
+        max_value=250.0,
+        value=0.0,
+        step=0.1
+    )
 
 with col2:
-    weight = st.number_input("Weight (in kg)", min_value=30, max_value=200)
-
-# ---------------------------------------------------
-# BMI CALCULATION
-# ---------------------------------------------------
-bmi = None
+    weight = st.number_input(
+        "Weight (in kg)",
+        min_value=0.0,
+        max_value=200.0,
+        value=0.0,
+        step=0.1
+    )
 
-if height > 0 and weight > 0:
-    height_m = height / 100
-    bmi = weight / (height_m ** 2)
-    st.write(f"### 📊 Your BMI: {bmi:.2f}")
-    
 def bmi_category(bmi):
     if bmi < 18.5:
         return "Underweight"
-    elif 18.5 <= bmi < 25:
+    elif bmi < 25:
         return "Normal weight"
-    elif 25 <= bmi < 30:
+    elif bmi < 30:
         return "Overweight"
     else:
         return "Obese"
 
+
+bmi = None
+
+if height > 0 and weight > 0:
+    height_m = height / 100
+    bmi = weight / (height_m ** 2)
+
+    st.metric("📊 Your BMI", f"{bmi:.2f}")
+    st.info(f"BMI Category: {bmi_category(bmi)}")
+
 # ---------------------------------------------------
 # FITNESS GOAL
 # ---------------------------------------------------
 st.subheader("🎯 Fitness Goal")
 
 goal = st.selectbox(
-    "",
+    "Select Your Goal",
     [
         "Flexible",
         "Weight Loss",
@@ -77,6 +93,9 @@ goal = st.selectbox(
     ]
 )
 
+# ---------------------------------------------------
+# EQUIPMENT
+# ---------------------------------------------------
 st.subheader("🏋️ Available Equipment")
 
 equipment_map = {}
@@ -107,33 +126,33 @@ with col3:
     equipment_map["Exercise Bike"] = st.checkbox("Exercise Bike")
     equipment_map["Skipping Rope"] = st.checkbox("Skipping Rope")
 
-# Convert selected to list
 equipment = [item for item, selected in equipment_map.items() if selected]
 
-
 # ---------------------------------------------------
 # FITNESS LEVEL
 # ---------------------------------------------------
 st.subheader("📈 Fitness Level")
 
 fitness_level = st.radio(
-    "",
+    "Select Fitness Level",
     ["Beginner", "Intermediate", "Advanced"],
     horizontal=True
 )
 
-
-if st.button("Submit Profile"):
+# ---------------------------------------------------
+# SUBMIT BUTTON
+# ---------------------------------------------------
+if st.button("🚀 Submit Profile"):
 
     if not name:
         st.error("Please enter your name.")
 
+    elif height <= 0 or weight <= 0:
+        st.error("Please enter valid height and weight.")
+
     elif not equipment:
         st.error("Please select at least one equipment option.")
 
-    elif bmi is None:
-        st.error("Please enter valid height and weight.")
-
     else:
         st.success("✅ Profile Submitted Successfully!")
 
@@ -163,4 +182,4 @@ if st.button("Submit Profile"):
             result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
 
         st.subheader("🏋️ Your Personalized Workout Plan")
-        st.write(result)
+        st.write(result)