Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,16 +1,38 @@
 import streamlit as st
+from transformers import pipeline
 
-st.title("FIT Plan AI – Personalized Fitness Plan Generator")
+# Load model
+@st.cache_resource
+def load_model():
+    return pipeline(
+        "text2text-generation",
+        model="google/flan-t5-base"
+    )
 
-goal = st.selectbox(
-    "Goal",
-    [
-        "Flexibility",
-        "Weight Loss",
-        "Build Muscle",
-        "Strength Gaining",
-        "Abs Building"
-    ]
-)
+model = load_model()
+
+st.set_page_config(page_title="Fitness Plan AI", layout="centered")
+
+st.title("🏋️ Fitness Plan AI Generator")
+st.write("Generate a personalized fitness plan using AI")
+
+# User inputs
+age = st.number_input("Age", 15, 80, 22)
+gender = st.selectbox("Gender", ["Male", "Female"])
+goal = st.selectbox("Fitness Goal", ["Weight Loss", "Muscle Gain", "General Fitness"])
+level = st.selectbox("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
+days = st.slider("Workout Days per Week", 1, 7, 4)
+
+if st.button("Generate Fitness Plan"):
+    prompt = f"""
+    Create a {days}-day fitness plan for a {age}-year-old {gender}.
+    Goal: {goal}
+    Level: {level}
+    Include warm-up, main workout, and cool-down.
+    """
+
+    with st.spinner("Generating plan..."):
+        output = model(prompt, max_length=300)
+        st.success("Your Fitness Plan:")
+        st.write(output[0]['generated_text'])
 
-st.write("You selected:", goal)