Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,4 @@
 import streamlit as st
-from transformers import pipeline
 
 def calculate_bmi(weight, height_cm):
     # Convert height from cm to meters
@@ -21,13 +20,47 @@ def get_category(bmi):
 # App UI
 st.set_page_config(page_title="FitPlan-AI", page_icon="💪")
 st.title("💪 FitPlan-AI: Personalized Fitness Profile")
-def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/flan-t5-base"
-    )
+if st.button("Submit Profile"):
  
-generator = load_model()
+    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)
+        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)
 
 with st.form("fitness_form"):
     st.header("Personal Information")