Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,18 +1,17 @@
 import streamlit as st
 from transformers import pipeline
 
+# Load model only once
+@st.cache_resource
 def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/flan-t5-base"
-    )
- 
+    return pipeline("text-generation", model="google/flan-t5-base")
+
 generator = load_model()
 
 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.")
+st.write("Fill in your details to calculate your BMI and get AI workout plan.")
 
 # -------------------------
 # 1. Personal Information
@@ -20,6 +19,8 @@ st.write("Fill in your details to calculate your BMI and fitness category.")
 
 name = st.text_input("Enter Your Name *")
 
+gender = st.selectbox("Select Gender", ["Male", "Female", "Other"])
+
 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")
 
@@ -35,30 +36,30 @@ goal = st.selectbox(
 )
 
 equipment = st.multiselect(
-    "Available Equipment (Select multiple if available)",
+    "Available Equipment",
     ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
 )
 
 fitness_level = st.radio(
-    "Select Your Fitness Level",
+    "Fitness Level",
     ["Beginner", "Intermediate", "Advanced"]
 )
 
 # -------------------------
-# BMI Calculation Function
+# BMI Functions
 # -------------------------
 
 def calculate_bmi(weight, height_cm):
-    height_m = height_cm / 100  # Convert cm to meters
+    height_m = height_cm / 100
     bmi = weight / (height_m ** 2)
     return round(bmi, 2)
 
 def bmi_category(bmi):
     if bmi < 18.5:
         return "Underweight"
-    elif 18.5 <= bmi < 24.9:
+    elif bmi < 24.9:
         return "Normal"
-    elif 25 <= bmi < 29.9:
+    elif bmi < 29.9:
         return "Overweight"
     else:
         return "Obese"
@@ -68,44 +69,41 @@ def bmi_category(bmi):
 # -------------------------
 
 if st.button("Submit Profile"):
- 
+
+    # Validation
     if not name:
         st.error("Please enter your name.")
- 
+    elif height_cm <= 0 or weight_kg <= 0:
+        st.error("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.")
- 
+        st.error("Select at least one equipment option.")
     else:
         st.success("✅ Profile Submitted Successfully!")
- 
+
+        # Calculate BMI
+        bmi = calculate_bmi(weight_kg, height_cm)
         bmi_status = bmi_category(bmi)
+
+        st.write(f"### 📊 BMI: {bmi} ({bmi_status})")
+
         equipment_list = ", ".join(equipment)
- 
+
         prompt = f"""
         Generate a 5-day structured workout plan.
- 
+
         User Details:
         Name: {name}
         Gender: {gender}
-        BMI: {bmi:.2f} ({bmi_status})
+        BMI: {bmi} ({bmi_status})
         Goal: {goal}
         Fitness Level: {fitness_level}
         Available Equipment: {equipment_list}
- 
-        Requirements:
-        - Include warmup
-        - Include exercises with sets and reps
-        - Include rest time
-        - Adjust intensity based on BMI and fitness level
-        - Keep it structured day-wise
+
+        Include warmup, exercises with sets & reps, rest time.
         """
- 
+
         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)
-