Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,10 +1,11 @@
 import streamlit as st
 from transformers import pipeline
 
+# -----------------------------
+# BMI Functions
+# -----------------------------
 def calculate_bmi(weight, height_cm):
-    # Convert height from cm to meters
     height_m = height_cm / 100
-    # BMI formula: weight (kg) / (height in meters)^2
     bmi = weight / (height_m ** 2)
     return round(bmi, 2)
 
@@ -18,19 +19,27 @@ def get_category(bmi):
     else:
         return "Obese"
 
-# App UI
-st.set_page_config(page_title="FitPlan-AI", page_icon="💪")
-st.title("💪 FitPlan-AI: Personalized Fitness Profile")
+# -----------------------------
+# Load AI Model
+# -----------------------------
+@st.cache_resource
 def load_model():
     return pipeline(
         "text-generation",
         model="google/flan-t5-base"
     )
- 
+
 generator = load_model()
 
+# -----------------------------
+# UI
+# -----------------------------
+st.set_page_config(page_title="FitPlan-AI", page_icon="💪")
+st.title("💪 FitPlan-AI: Personalized Fitness Profile")
+
 with st.form("fitness_form"):
     st.header("Personal Information")
+    
     name = st.text_input("Full Name*", placeholder="Enter your name")
     
     col1, col2 = st.columns(2)
@@ -40,59 +49,56 @@ with st.form("fitness_form"):
         weight = st.number_input("Weight (kg)*", min_value=1.0, step=0.1)
 
     st.header("Fitness Details")
-    goal = st.selectbox("Fitness Goal", ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
+    
+    goal = st.selectbox(
+        "Fitness Goal",
+        ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
+    )
+    
     level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
-    equipment = st.multiselect("Available Equipment", ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Kettlebell", "Pull-up Bar"])
+    
+    equipment = st.multiselect(
+        "Available Equipment",
+        ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Kettlebell", "Pull-up Bar"]
+    )
 
-    submit = st.form_submit_button("Generate Profile")
+    submit = st.form_submit_button("Generate Workout Plan")
 
+# -----------------------------
+# When Form is Submitted
+# -----------------------------
 if submit:
-    # Validation: Ensure name is not empty
+
     if not name.strip():
         st.error("Please enter your name.")
+
+    elif not equipment:
+        st.error("Please select at least one equipment option.")
+
     elif height <= 0 or weight <= 0:
         st.error("Height and Weight must be positive values.")
+
     else:
-        # BMI Logic
+        # Calculate BMI
         user_bmi = calculate_bmi(weight, height)
-        category = get_category(user_bmi)
-        
-        # Display Results
-        st.success(f"Profile Created Successfully for {name}!")
-        st.metric(label="Your Calculated BMI", value=user_bmi)
-        st.info(f"Health Category: **{category}**")
-        
-        # Summary
-        st.write(f"**Goal:** {goal} | **Level:** {level}")
-        st.write(f"**Equipment:** {', '.join(equipment) if equipment else 'None selected'}")
-if st.button("Submit Profile"):
- 
-    if not name:
-        st.error("Please enter your name.")
- 
-    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!")
- 
-        bmi_status = bmi_category(bmi)
+        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"""
         Generate a 5-day structured workout plan.
- 
+
         User Details:
         Name: {name}
-        Gender: {gender}
-        BMI: {bmi:.2f} ({bmi_status})
+        BMI: {user_bmi} ({bmi_status})
         Goal: {goal}
-        Fitness Level: {fitness_level}
+        Fitness Level: {level}
         Available Equipment: {equipment_list}
- 
+
         Requirements:
         - Include warmup
         - Include exercises with sets and reps
@@ -100,9 +106,9 @@ if st.button("Submit Profile"):
         - Adjust intensity based on BMI and fitness level
         - Keep it structured day-wise
         """
- 
+
         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)
+        st.write(result)