Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,11 @@
 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
 # -----------------------------
@@ -12,20 +17,20 @@ def calculate_bmi(weight, height_cm):
 def get_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"
 
 # -----------------------------
-# Load AI Model
+# Load AI Model (Correct Pipeline)
 # -----------------------------
 @st.cache_resource
 def load_model():
     return pipeline(
-        "text-generation",
+        "text2text-generation",  # ✅ Correct for FLAN-T5
         model="google/flan-t5-base"
     )
 
@@ -34,14 +39,14 @@ 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)
     with col1:
         height = st.number_input("Height (cm)*", min_value=1.0, step=0.1)
@@ -49,14 +54,17 @@ 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"]
     )
-    
-    level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
-    
+
+    level = st.radio(
+        "Fitness Level",
+        ["Beginner", "Intermediate", "Advanced"]
+    )
+
     equipment = st.multiselect(
         "Available Equipment",
         ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Kettlebell", "Pull-up Bar"]
@@ -65,19 +73,20 @@ with st.form("fitness_form"):
     submit = st.form_submit_button("Generate Workout Plan")
 
 # -----------------------------
-# When Form is Submitted
+# FORM SUBMISSION LOGIC
 # -----------------------------
 if submit:
 
-    if not name.strip():
+    # Validation
+    if 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.")
 
+    elif len(equipment) == 0:
+        st.error("Please select at least one equipment option.")
+
     else:
         # Calculate BMI
         user_bmi = calculate_bmi(weight, height)
@@ -90,21 +99,21 @@ if submit:
         equipment_list = ", ".join(equipment)
 
         prompt = f"""
-        Generate a 5-day structured workout plan.
+        Create a professional 5-day structured workout plan.
 
         User Details:
         Name: {name}
         BMI: {user_bmi} ({bmi_status})
         Goal: {goal}
         Fitness Level: {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
+        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..."):