Update app.py

app.py

@@ -1,82 +1,134 @@
 import streamlit as st
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
+# -------------------------
+# Page Config
+# -------------------------
+
+st.set_page_config(page_title="FitPlan AI", page_icon="💪")
+
+st.title("💪 FitPlan AI — Personalized Workout Generator")
+
+# -------------------------
+# BMI Functions
+# -------------------------
+
+def calculate_bmi(weight, height):
+    height_m = height / 100
+    return weight / (height_m ** 2)
+
+def bmi_category(bmi):
+    if bmi < 18.5:
+        return "Underweight"
+    elif bmi < 25:
+        return "Normal Weight"
+    elif bmi < 30:
+        return "Overweight"
+    else:
+        return "Obese"
+
+# -------------------------
+# Prompt Builder Function
+# -------------------------
+
+def build_prompt(name, gender, height, weight, goal, fitness_level, equipment):
+
+    bmi = calculate_bmi(weight, height)
+    bmi_status = bmi_category(bmi)
+
+    equipment_list = ", ".join(equipment) if equipment else "No Equipment"
+
+    prompt = f"""
+You are a certified professional fitness trainer.
 
-st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
+Create a STRICTLY FORMATTED 5-day personalized workout plan.
 
-st.write("Fill in your details to calculate your BMI and fitness category.")
+User Profile:
+Name: {name}
+Gender: {gender}
+Height: {height} cm
+Weight: {weight} kg
+BMI: {bmi:.2f} ({bmi_status})
+Goal: {goal}
+Fitness Level: {fitness_level}
+Available Equipment: {equipment_list}
+
+Instructions:
+• Divide into Day 1 to Day 5
+• Include Warm-up
+• Include Exercises with Sets × Reps
+• Include Rest Time
+• Adjust intensity based on BMI
+"""
+
+    return prompt, bmi, bmi_status
 
 # -------------------------
-# 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
 
-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")
+tokenizer, model = load_model()
 
 # -------------------------
-# 2. Fitness Details
+# User Inputs
 # -------------------------
 
-st.subheader("Fitness Details")
+name = st.text_input("Enter Your Name")
+gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+height = st.number_input("Height (cm)", min_value=0.0)
+weight = st.number_input("Weight (kg)", min_value=0.0)
 
 goal = st.selectbox(
-    "Select Your Fitness Goal",
+    "Fitness Goal",
     ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
 )
 
 equipment = st.multiselect(
-    "Available Equipment (Select multiple if available)",
+    "Available Equipment",
     ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope",
      "Weight Plates", "Cycling", "Inclined Bench", "Pullups Bar", "No Equipment"]
 )
 
 fitness_level = st.radio(
-    "Select Your Fitness Level",
+    "Fitness Level",
     ["Beginner", "Intermediate", "Advanced"]
 )
 
 # -------------------------
-# BMI Calculation Function
+# Generate Plan
 # -------------------------
 
-def calculate_bmi(weight, height_cm):
-    height_m = height_cm / 100  # Convert cm to meters
-    bmi = weight / (height_m ** 2)
-    return round(bmi, 2)
+if st.button("Generate Workout Plan"):
 
-def bmi_category(bmi):
-    if bmi < 18.5:
-        return "Underweight"
-    elif 18.5 <= bmi < 24.9:
-        return "Normal"
-    elif 25 <= bmi < 29.9:
-        return "Overweight"
+    if not name or height <= 0 or weight <= 0:
+        st.error("Please fill all required fields")
     else:
-        return "Obese"
 
-# -------------------------
-# Submit Button
-# -------------------------
+        prompt, bmi, bmi_status = build_prompt(
+            name, gender, height, weight,
+            goal, fitness_level, equipment
+        )
 
-if st.button("Calculate BMI"):
-    
-    # Validation
-    if not name or height_cm <= 0 or weight_kg <= 0:
-        st.error("⚠ Please fill all required fields with valid values!")
-    else:
-        bmi = calculate_bmi(weight_kg, height_cm)
-        category = bmi_category(bmi)
+        st.success(f"BMI: {bmi:.2f} ({bmi_status})")
+
+        with st.spinner("Generating your personalized workout plan..."):
+
+            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
 
-        st.success("✅ Calculation Successful!")
+            outputs = model.generate(
+                **inputs,
+                max_new_tokens=500,
+                temperature=0.3,
+                do_sample=True
+            )
 
-        st.write(f"### 👤 Name: {name}")
-        st.write(f"### 📊 Your BMI: {bmi}")
-        st.write(f"### 🏷 BMI Category: {category}")
+            result = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-        st.write("---")
-        st.write("### 🏋 Fitness Summary")
-        st.write(f"**Goal:** {goal}")
-        st.write(f"**Fitness Level:** {fitness_level}")
-        st.write(f"**Equipment Available:** {', '.join(equipment) if equipment else 'None selected'}")
+        st.subheader("🏋️ Your Workout Plan")
+        st.write(result)