Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,43 +1,68 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torch
 
-# Load model only once
-@st.cache_resource
-def load_model():
-    return pipeline("text-generation", model="google/flan-t5-base")
+# -------------------------
+# Page Config FIRST
+# -------------------------
 
-generator = load_model()
+st.set_page_config(page_title="FitPlan AI", page_icon="💪")
 
-st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
-st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
+# 🎨 Background Style
 
-st.write("Fill in your details to calculate your BMI and get AI workout plan.")
+st.markdown(
+    """
+    <style>
+    .stApp {
+        background-image: url("https://images.unsplash.com/photo-1554284126-aa88f22d8b74");
+        background-size: cover;
+    }
+    </style>
+    """,
+    unsafe_allow_html=True
+)
 
 # -------------------------
-# 1. Personal Information
+# Load Model (Small for HF)
 # -------------------------
 
-name = st.text_input("Enter Your Name *")
+@st.cache_resource
+def load_model():
+    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
+    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
+    return tokenizer, model
 
-gender = st.selectbox("Select Gender", ["Male", "Female", "Other"])
+tokenizer, model = load_model()
 
-height_cm = st.number_input("Enter Height (in centimeters) *", min_value=0.0, format="%.2f")
-weight_kg = st.number_input("Enter Weight (in kilograms) *", min_value=0.0, format="%.2f")
+# -------------------------
+# App Title
+# -------------------------
+
+st.title("💪 FitPlan AI - BMI Calculator & Workout Planner")
+st.write("Enter your details to get AI workout plan.")
 
 # -------------------------
-# 2. Fitness Details
+# Personal Info
 # -------------------------
 
-st.subheader("Fitness Details")
+name = st.text_input("Enter Your Name *")
+gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+height_cm = st.number_input("Height (cm)", min_value=0.0)
+weight_kg = st.number_input("Weight (kg)", min_value=0.0)
+
+# -------------------------
+# Fitness Details
+# -------------------------
 
 goal = st.selectbox(
-    "Select Your Fitness Goal",
+    "Fitness Goal",
     ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
 )
 
 equipment = st.multiselect(
     "Available Equipment",
-    ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
+    ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope",
+     "Weight Plates", "Cycling", "Inclined Bench", "Pullups Bar", "No Equipment"]
 )
 
 fitness_level = st.radio(
@@ -49,10 +74,9 @@ fitness_level = st.radio(
 # BMI Functions
 # -------------------------
 
-def calculate_bmi(weight, height_cm):
-    height_m = height_cm / 100
-    bmi = weight / (height_m ** 2)
-    return round(bmi, 2)
+def calculate_bmi(weight, height):
+    height_m = height / 100
+    return round(weight / (height_m ** 2), 2)
 
 def bmi_category(bmi):
     if bmi < 18.5:
@@ -65,45 +89,51 @@ def bmi_category(bmi):
         return "Obese"
 
 # -------------------------
-# Submit Button
+# Generate Plan
 # -------------------------
 
-if st.button("Submit Profile"):
+if st.button("Generate Workout Plan"):
 
-    # Validation
     if not name:
-        st.error("Please enter your name.")
+        st.error("Enter your name")
     elif height_cm <= 0 or weight_kg <= 0:
-        st.error("Enter valid height and weight.")
+        st.error("Enter valid height & weight")
     elif not equipment:
-        st.error("Select at least one equipment option.")
+        st.error("Select equipment")
     else:
-        st.success("✅ Profile Submitted Successfully!")
-
-        # Calculate BMI
         bmi = calculate_bmi(weight_kg, height_cm)
         bmi_status = bmi_category(bmi)
 
-        st.write(f"### 📊 BMI: {bmi} ({bmi_status})")
+        st.success(f"BMI: {bmi} ({bmi_status})")
 
         equipment_list = ", ".join(equipment)
 
         prompt = f"""
-        Generate a 5-day structured workout plan.
+        Create a 5-day workout plan.
 
-        User Details:
-        Name: {name}
-        Gender: {gender}
-        BMI: {bmi} ({bmi_status})
-        Goal: {goal}
-        Fitness Level: {fitness_level}
-        Available Equipment: {equipment_list}
+        Include warmup, exercises, sets, reps, rest time.
+        
+Day 1 – Chest & Triceps: Bench Press 4×8–10, Incline Press 3×10, Dips 3×12, Tricep Extensions 3×12.
 
-        Include warmup, exercises with sets & reps, rest time.
-        """
+Day 2 – Back & Biceps: Pull-ups 4×8–10, Rows 4×10, Band Rows 3×12, Bicep Curls 3×12.
 
-        with st.spinner("Generating your AI workout plan..."):
-            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
+Day 3 – Legs: Squats 4×8–10, Lunges 3×12/leg, Romanian Deadlift 3×10, Calf Raises 4×15. 
+
+Day 4 – Shoulders & Core: Overhead Press 4×8–10, Lateral Raises 3×12, Plank 3×60s, Leg Raises 3×12.
+ 
+Day 5 – Cardio & Conditioning: 25-min Run, Skipping 3×2min, Burpees 3×12, Mountain Climbers 3×30s.
+        
+        """
 
-        st.subheader("🏋️ Your Personalized Workout Plan")
+        with st.spinner("Generating workout plan..."):
+            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+            outputs = model.generate(
+                **inputs,
+                max_new_tokens=400,
+                temperature=0.7,
+                do_sample=True
+            )
+            result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+        st.subheader("🏋️ Your Workout Plan")
         st.write(result)