Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,12 +1,33 @@
 import streamlit as st
+from transformers import pipeline
 
+# ---------------------------------------------------
+# PAGE CONFIG
+# ---------------------------------------------------
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-st.title(" Your Fitness Profile")
+st.title("🏋️ FitPlan AI – Personalized Fitness Plan")
 
-# --------------------------
+# ---------------------------------------------------
+# LOAD MODEL (Cached)
+# ---------------------------------------------------
+@st.cache_resource
+def load_model():
+    return pipeline(
+        "text-generation",
+        model="google/flan-t5-base"
+    )
+
+generator = load_model()
+
+# ---------------------------------------------------
+# USER NAME
+# ---------------------------------------------------
+name = st.text_input("Enter Your Name")
+
+# ---------------------------------------------------
 # GOALS
-# --------------------------
+# ---------------------------------------------------
 goal = st.selectbox(
     "Fitness Goal",
     [
@@ -18,9 +39,9 @@ goal = st.selectbox(
     ]
 )
 
-# --------------------------
+# ---------------------------------------------------
 # EQUIPMENT
-# --------------------------
+# ---------------------------------------------------
 st.subheader("Available Equipment")
 
 col1, col2 = st.columns(2)
@@ -42,7 +63,6 @@ with col2:
     hula_hoop = st.checkbox("Hula Hoop Ring")
     bosu_ball = st.checkbox("Bosu Ball")
 
-# Collect equipment into list
 equipment = []
 
 equipment_map = {
@@ -65,9 +85,9 @@ for item, selected in equipment_map.items():
     if selected:
         equipment.append(item)
 
-# --------------------------
+# ---------------------------------------------------
 # FITNESS LEVEL
-# --------------------------
+# ---------------------------------------------------
 st.subheader("Fitness Level")
 
 fitness_level = st.radio(
@@ -76,17 +96,51 @@ fitness_level = st.radio(
     horizontal=True
 )
 
-# --------------------------
+# ---------------------------------------------------
+# PROMPT BUILDER
+# ---------------------------------------------------
+def build_prompt(name, goal, fitness_level, equipment):
+    equipment_text = ", ".join(equipment)
+
+    prompt = f"""
+    Generate a detailed 5-day workout plan for {name}.
+
+    Goal: {goal}
+    Fitness Level: {fitness_level}
+    Available Equipment: {equipment_text}
+
+    Include:
+    - Day-wise workout split
+    - Exercises
+    - Sets and reps
+    - Rest time
+    - Warm-up and cool-down
+    """
+
+    return prompt
+
+
+# ---------------------------------------------------
 # SUBMIT BUTTON
-# --------------------------
+# ---------------------------------------------------
 if st.button("Generate Personalised Plan"):
-    
-    if not equipment:
+
+    if not name:
+        st.error("Please enter your name.")
+    elif not equipment:
         st.error("Please select at least one equipment option.")
     else:
-        st.success("Profile Submitted ✅")
-        st.json({
-            "goal": goal,
-            "fitness_level": fitness_level,
-            "equipment": equipment
-        })
+        with st.spinner("Generating your AI workout plan..."):
+            prompt = build_prompt(name, goal, fitness_level, equipment)
+
+            result = generator(
+                prompt,
+                max_length=500,
+                do_sample=True,
+                temperature=0.7
+            )
+
+            plan = result[0]["generated_text"]
+
+        st.success(f"Plan Generated Successfully, {name}! 💪")
+        st.write(plan)