Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,8 +1,14 @@
 import streamlit as st
-
+from transformers import pipeline
 # 1. Basic Configuration
 st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
-
+def load_model():
+    return pipeline(
+        "text-generation",
+        model="google/flan-t5-base"
+    )
+ 
+generator = load_model()
 # 2. Enhanced CSS
 st.markdown("""
 <style>
@@ -110,34 +116,45 @@ with st.form("fitness_form", clear_on_submit=False):
     # Submission Button
     submit = st.form_submit_button("GENERATE MY PLAN")
 
-# --------------------------------------------------
-# Logic & Calculation
-# --------------------------------------------------
-if submit:
-    if not name.strip() or height <= 50 or weight <= 10:
-        st.error("🚨 Please provide a valid name, height, and weight.")
+if st.button("🚀 Submit Profile"):
+ 
+    if not name:
+        st.error("Please enter your name.")
+ 
+    elif height <= 0 or weight <= 0:
+        st.error("Please enter valid height and weight.")
+ 
+    elif not equipment:
+        st.error("Please select at least one equipment option.")
+ 
     else:
-        # BMI Calculation
-        height_m = height / 100
-        bmi = round(weight / (height_m ** 2), 2)
-
-        if bmi < 18.5:
-            cat, color = "Underweight", "blue"
-        elif 18.5 <= bmi < 25:
-            cat, color = "Normal", "green"
-        elif 25 <= bmi < 30:
-            cat, color = "Overweight", "orange"
-        else:
-            cat, color = "Obese", "red"
-
-        # Results Display
-        st.balloons()
-        st.success(f"✅ Profile created for {name}")
-        
-        c1, c2 = st.columns(2)
-        with c1:
-            st.metric("Calculated BMI", bmi)
-        with c2:
-            st.markdown(f"### Status: :{color}[{cat}]")
-        
-        st.info(f"*Goal:* {goal} | *Experience:* {level}")
+        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..."):
+            response = generator(prompt, max_new_tokens=400)
+            result = response[0]["generated_text"]
+ 
+        st.subheader("🏋️ Your Personalized Workout Plan")
+        st.write(result)