Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,10 +1,20 @@
 import streamlit as st
+from transformers import pipeline
 
 # ---------------------------------------------------
 # PAGE CONFIG
 # ---------------------------------------------------
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
+@st.cache_resource
+def load_model():
+    return pipeline(
+        "text-generation",
+        model="google/flan-t5-base"
+    )
+
+generator = load_model()
+
 st.title(" FitPlan AI – User Fitness Profile")
 
 # ---------------------------------------------------
@@ -40,6 +50,16 @@ if height > 0 and weight > 0:
     height_m = height / 100
     bmi = weight / (height_m ** 2)
     st.write(f"### 📊 Your BMI: {bmi:.2f}")
+    
+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"
 
 # ---------------------------------------------------
 # FITNESS GOAL
@@ -102,23 +122,45 @@ fitness_level = st.radio(
     horizontal=True
 )
 
-# ---------------------------------------------------
-# SUBMIT BUTTON
-# ---------------------------------------------------
+
 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!")
 
-        st.json({
-            "Name": name,
-            "Gender": gender,
-            "BMI": round(bmi, 2) if bmi else None,
-            "Goal": goal,
-            "Fitness Level": fitness_level,
-            "Equipment": equipment
-        })
+        bmi_status = bmi_category(bmi)
+        equipment_list = ", ".join(equipment)
+
+        prompt = f"""
+        Generate a 5-day structured workout plan.
+
+        User Details:
+        Name: {name}
+        Gender: {gender}
+        BMI: {bmi:.2f} ({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
+        """
+
+        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)