Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from transformers import pipeline
 
 def calculate_bmi(weight, height_cm):
     # Convert height from cm to meters
@@ -20,6 +21,50 @@ 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"
+    )
+ 
+generator = load_model()
+
+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)
+    with col2:
+        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"])
+    equipment = st.multiselect("Available Equipment", ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Kettlebell", "Pull-up Bar"])
+
+    submit = st.form_submit_button("Generate Profile")
+
+if submit:
+    # Validation: Ensure name is not empty
+    if not name.strip():
+        st.error("Please enter your name.")
+    elif height <= 0 or weight <= 0:
+        st.error("Height and Weight must be positive values.")
+    else:
+        # BMI Logic
+        user_bmi = calculate_bmi(weight, height)
+        category = get_category(user_bmi)
+        
+        # Display Results
+        st.success(f"Profile Created Successfully for {name}!")
+        st.metric(label="Your Calculated BMI", value=user_bmi)
+        st.info(f"Health Category: **{category}**")
+        
+        # Summary
+        st.write(f"**Goal:** {goal} | **Level:** {level}")
+        st.write(f"**Equipment:** {', '.join(equipment) if equipment else 'None selected'}")
 if st.button("Submit Profile"):
  
     if not name:
@@ -60,41 +105,4 @@ if st.button("Submit Profile"):
             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")
-    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)
-    with col2:
-        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"])
-    equipment = st.multiselect("Available Equipment", ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Kettlebell", "Pull-up Bar"])
-
-    submit = st.form_submit_button("Generate Profile")
-
-if submit:
-    # Validation: Ensure name is not empty
-    if not name.strip():
-        st.error("Please enter your name.")
-    elif height <= 0 or weight <= 0:
-        st.error("Height and Weight must be positive values.")
-    else:
-        # BMI Logic
-        user_bmi = calculate_bmi(weight, height)
-        category = get_category(user_bmi)
-        
-        # Display Results
-        st.success(f"Profile Created Successfully for {name}!")
-        st.metric(label="Your Calculated BMI", value=user_bmi)
-        st.info(f"Health Category: **{category}**")
-        
-        # Summary
-        st.write(f"**Goal:** {goal} | **Level:** {level}")
-        st.write(f"**Equipment:** {', '.join(equipment) if equipment else 'None selected'}")
+        st.write(result)