Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,13 @@
 import streamlit as st
 from transformers import pipeline
 st.set_page_config(page_title="Fitness Profile", page_icon="🏋️", layout="centered")
-
+def load_model():
+    return pipeline(
+        "text-generation",
+        model="google/flan-t5-base"
+    )
+ 
+generator = load_model()
 st.title("🏋️ Personalized Fitness Profile")
 
 st.markdown("---")
@@ -40,50 +46,44 @@ fitness_level = st.radio(
 st.markdown("---")
 
 # =========================
-# 3️⃣ Submit Button
-# =========================
-if st.button("Generate Profile"):
-
-    # =========================
-    # Validation
-    # =========================
-    if name.strip() == "":
-        st.error("❌ Name is required.")
-    elif height_cm <= 0:
-        st.error("❌ Height must be greater than zero.")
-    elif weight_kg <= 0:
-        st.error("❌ Weight must be greater than zero.")
+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:
-        # =========================
-        # BMI Calculation
-        # =========================
-        height_m = height_cm / 100  # Convert cm to meters
-        bmi = weight_kg / (height_m ** 2)
-        bmi = round(bmi, 2)
-
-        # =========================
-        # BMI Category
-        # =========================
-        if bmi < 18.5:
-            category = "Underweight"
-        elif 18.5 <= bmi < 24.9:
-            category = "Normal"
-        elif 25 <= bmi < 29.9:
-            category = "Overweight"
-        else:
-            category = "Obese"
-
-        # =========================
-        # Display Results
-        # =========================
-        st.success("✅ Profile Generated Successfully!")
-
-        st.subheader(f"👤 {name}'s Fitness Summary")
-
-        st.write(f"**Height (m):** {round(height_m, 2)} m")
-        st.write(f"**Weight (kg):** {weight_kg} kg")
-        st.write(f"**BMI:** {bmi}")
-        st.write(f"**BMI Category:** {category}")
-        st.write(f"**Fitness Goal:** {fitness_goal}")
-        st.write(f"**Fitness Level:** {fitness_level}")
-        st.write(f"**Available Equipment:** {', '.join(equipment) if equipment else 'None Selected'}")
+        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)