Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,18 +1,17 @@
 import streamlit as st
 from transformers import pipeline
 
-# -----------------------------
+# -------------------------
 # PAGE CONFIG (MUST BE FIRST)
-# -----------------------------
+# -------------------------
 st.set_page_config(page_title="FitPlan-AI", page_icon="💪")
 
-# -----------------------------
-# BMI Functions
-# -----------------------------
+# -------------------------
+# BMI FUNCTIONS
+# -------------------------
 def calculate_bmi(weight, height_cm):
     height_m = height_cm / 100
-    bmi = weight / (height_m ** 2)
-    return round(bmi, 2)
+    return round(weight / (height_m ** 2), 2)
 
 def get_category(bmi):
     if bmi < 18.5:
@@ -24,26 +23,26 @@ def get_category(bmi):
     else:
         return "Obese"
 
-# -----------------------------
-# Load AI Model (Correct Pipeline)
-# -----------------------------
+# -------------------------
+# LOAD MODEL (Correct Pipeline)
+# -------------------------
 @st.cache_resource
 def load_model():
     return pipeline(
-        "text2text-generation",  # ✅ Correct for FLAN-T5
+        "text2text-generation",  # Correct for FLAN-T5
         model="google/flan-t5-base"
     )
 
 generator = load_model()
 
-# -----------------------------
-# UI
-# -----------------------------
+# -------------------------
+# UI STARTS
+# -------------------------
 st.title("💪 FitPlan-AI: Personalized Fitness Profile")
 
 with st.form("fitness_form"):
 
-    st.header("Personal Information")
+    st.subheader("Personal Information")
 
     name = st.text_input("Full Name*", placeholder="Enter your name")
 
@@ -53,7 +52,7 @@ with st.form("fitness_form"):
     with col2:
         weight = st.number_input("Weight (kg)*", min_value=1.0, step=0.1)
 
-    st.header("Fitness Details")
+    st.subheader("Fitness Details")
 
     goal = st.selectbox(
         "Fitness Goal",
@@ -72,52 +71,54 @@ with st.form("fitness_form"):
 
     submit = st.form_submit_button("Generate Workout Plan")
 
-# -----------------------------
-# FORM SUBMISSION LOGIC
-# -----------------------------
+# -------------------------
+# HANDLE SUBMISSION
+# -------------------------
 if submit:
 
     # Validation
     if name.strip() == "":
         st.error("Please enter your name.")
+        st.stop()
 
-    elif height <= 0 or weight <= 0:
+    if height <= 0 or weight <= 0:
         st.error("Height and Weight must be positive values.")
+        st.stop()
 
-    elif len(equipment) == 0:
+    if len(equipment) == 0:
         st.error("Please select at least one equipment option.")
+        st.stop()
 
-    else:
-        # Calculate BMI
-        user_bmi = calculate_bmi(weight, height)
-        bmi_status = get_category(user_bmi)
-
-        st.success(f"✅ Profile Created Successfully for {name}!")
-        st.metric("Your BMI", user_bmi)
-        st.info(f"Health Category: **{bmi_status}**")
-
-        equipment_list = ", ".join(equipment)
-
-        prompt = f"""
-        Create a professional 5-day structured workout plan.
-
-        User Details:
-        Name: {name}
-        BMI: {user_bmi} ({bmi_status})
-        Goal: {goal}
-        Fitness Level: {level}
-        Equipment: {equipment_list}
-
-        Include:
-        - Warm-up
-        - Exercises with sets and reps
-        - Rest time
-        - Intensity adjustment
-        - Day-wise structure
-        """
-
-        with st.spinner("Generating your AI workout plan..."):
-            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
-
-        st.subheader("🏋️ Your Personalized Workout Plan")
-        st.write(result)
+    # Calculate BMI
+    user_bmi = calculate_bmi(weight, height)
+    bmi_status = get_category(user_bmi)
+
+    st.success(f"✅ Profile Created Successfully for {name}!")
+    st.metric("Your BMI", user_bmi)
+    st.info(f"Health Category: **{bmi_status}**")
+
+    equipment_list = ", ".join(equipment)
+
+    prompt = f"""
+    Create a professional 5-day structured workout plan.
+
+    User Details:
+    Name: {name}
+    BMI: {user_bmi} ({bmi_status})
+    Goal: {goal}
+    Fitness Level: {level}
+    Equipment: {equipment_list}
+
+    Include:
+    - Warm-up
+    - Exercises with sets and reps
+    - Rest time
+    - Intensity adjustment
+    - Day-wise structure
+    """
+
+    with st.spinner("Generating your AI workout plan..."):
+        result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
+
+    st.subheader("🏋️ Your Personalized Workout Plan")
+    st.write(result)