Update src/streamlit_app.py

src/streamlit_app.py

@@ -12,14 +12,35 @@ st.set_page_config(
 )
 
 # =========================
-# Load Model
+# Load Model (Cached)
 # =========================
+@st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
     model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
     return tokenizer, model
- 
+
 tokenizer, model = load_model()
+
+# =========================
+# BMI Functions
+# =========================
+def calculate_bmi(height_cm, weight_kg):
+    if height_cm > 0 and weight_kg > 0:
+        height_m = height_cm / 100
+        return weight_kg / (height_m ** 2)
+    return None
+
+def bmi_category(bmi):
+    if bmi < 18.5:
+        return "Underweight"
+    elif bmi < 25:
+        return "Normal weight"
+    elif bmi < 30:
+        return "Overweight"
+    else:
+        return "Obese"
+
 # =========================
 # 1️⃣ Personal Information
 # =========================
@@ -45,34 +66,14 @@ weight_kg = st.number_input(
 )
 
 # =========================
-# BMI Calculation
+# BMI Display
 # =========================
-# =========================
-# BMI Calculation
-# =========================
-def calculate_bmi(height_cm, weight_kg):
-    if height_cm > 0 and weight_kg > 0:
-        height_m = height_cm / 100
-        return weight_kg / (height_m ** 2)
-    return 0   # Always return something
+bmi = None
 
-def bmi_category(bmi):
-    if bmi < 18.5:
-        return "Underweight"
-    elif 18.5 <= bmi < 25:
-        return "Normal weight"
-    elif 25 <= bmi < 30:
-        return "Overweight"
-    else:
-        return "Obese"
-
-# ALWAYS define bmi safely
-bmi = 0
 if height_cm > 0 and weight_kg > 0:
     bmi = calculate_bmi(height_cm, weight_kg)
     st.info(f"Your BMI is: {bmi:.2f} ({bmi_category(bmi)})")
 
-
 # =========================
 # 2️⃣ Fitness Details
 # =========================
@@ -99,16 +100,12 @@ st.markdown("---")
 # =========================
 # Submit Button
 # =========================
-# =========================
-# Submit Button
-# =========================
-
 if st.button("Submit Profile"):
 
     if not name:
         st.error("Please enter your name.")
 
-    elif height_cm <= 0 or weight_kg <= 0:
+    elif bmi is None:
         st.error("Please enter valid height and weight.")
 
     elif not equipment:
@@ -121,7 +118,6 @@ if st.button("Submit Profile"):
         equipment_list = ", ".join(equipment)
 
         prompt = f"""
-prompt = f"""
 You are a certified fitness trainer.
 
 Create a STRICTLY formatted 5-day workout plan.
@@ -142,22 +138,32 @@ IMPORTANT RULES:
    Sets:
    Reps:
    Rest:
-5. Keep exercises suitable for the fitness level.
-6. Do NOT add explanations, tips, introductions, or conclusions.
-7. Keep formatting clean and consistent.
+5. Do NOT add explanations or extra text.
 
-OUTPUT FORMAT (FOLLOW EXACTLY):
+FORMAT:
 
 Day 1: [Workout Focus]
 1. Exercise: Push-Ups
    Sets: 3
    Reps: 12
    Rest: 60 sec
+"""
 
-2. Exercise: Squats
-   Sets: 3
-   Reps: 15
-   Rest: 60 sec
+        with st.spinner("Generating your AI workout plan..."):
 
-Continue same format for all 5 days.
-"""
+            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+
+            outputs = model.generate(
+                **inputs,
+                max_new_tokens=700,
+                temperature=0.3,
+                do_sample=True
+            )
+
+            result = tokenizer.decode(
+                outputs[0],
+                skip_special_tokens=True
+            ).strip()
+
+        st.subheader("Your Personalized Workout Plan")
+        st.write(result)