Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from transformers import pipeline  # <--- Line 2: Added as requested
+from transformers import pipeline 
 
 # --- Page Config ---
 st.set_page_config(page_title="FitPlan AI", page_icon="🏋", layout="wide")
@@ -35,12 +35,13 @@ def bmi_category(bmi):
    if bmi < 30: return "Overweight"
    return "Obese"
 
-# --- AI Model Loading (Mam's Code) ---
+# --- AI Model Loading ---
 @st.cache_resource 
 def load_model():
-    return pipeline("text2text-generation", model="google/flan-t5-base")
+    # FIXED: Changed back to "text-generation" to match your error's allowed list
+    return pipeline("text-generation", model="google/flan-t5-base")
 
-generator = load_model() # <--- Line 48: Added as requested
+generator = load_model() 
 
 # --- Sidebar Navigation ---
 with st.sidebar:
@@ -50,9 +51,8 @@ with st.sidebar:
 # --- NAVIGATION LOGIC ---
 
 if menu == "👤 Profile":
-   st.title("👤 Profile")
+   st.title("👤 User Profile")
    
-   # Input fields
    name = st.text_input("Name", value=st.session_state.user_data['name'])
    gender = st.selectbox("Gender", ["Male", "Female", "Other"])
    col1, col2 = st.columns(2)
@@ -65,22 +65,18 @@ if menu == "👤 Profile":
 
    bmi = calculate_bmi(u_weight, u_height)
 
-   # --- MAM'S SUBMIT LOGIC START (Lines 87-119) ---
+   # --- MAM'S SUBMIT LOGIC ---
    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!")
         
-        # Save to session state
         st.session_state.user_data.update({
             'name': name, 'height': u_height, 'weight': u_weight, 
             'goal': goal, 'equip': equipment, 'level': fitness_level, 'gender': gender
@@ -91,30 +87,18 @@ if menu == "👤 Profile":
         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
+        Generate a 5-day structured workout plan for {name}.
+        Gender: {gender}, BMI: {bmi:.2f} ({bmi_status}), Goal: {goal}, Level: {fitness_level}.
+        Equipment: {equipment_list}.
+        Plan:
         """
  
         with st.spinner("Generating your AI workout plan..."):
-            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
+            # Using the generator as per mam's code
+            result = generator(prompt, max_new_tokens=200)[0]["generated_text"]
  
         st.subheader("🏋️ Your Personalized Workout Plan")
         st.write(result)
-   # --- MAM'S SUBMIT LOGIC END ---
 
 elif menu == "📊 Dashboard":
    if not st.session_state.profile_complete: