Update app.py

app.py

@@ -1,75 +1,58 @@
 import streamlit as st
+from prompt.bmi import calculate_bmi, bmi_category
+from prompt.prompt_builder import build_prompt
+from prompt.generator import load_model, generate_plan
 
-st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
+st.set_page_config(page_title="FitPlan AI", page_icon="💪")
 
-st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
+st.title("💪 FitPlan AI - Personalized Workout Generator")
 
-st.write("Fill in your details to calculate your BMI and fitness category.")
+# Load model
+tokenizer, model = load_model()
 
 # -------------------------
-# 1. Personal Information
+# User Inputs
 # -------------------------
 
-name = st.text_input("Enter Your Name *")
+name = st.text_input("Name")
+gender = st.selectbox("Gender", ["Male", "Female"])
 
-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")
-
-# -------------------------
-# 2. Fitness Details
-# -------------------------
-
-st.subheader("Fitness Details")
+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)",
-    ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope",
-     "Weight Plates", "Cycling", "Inclined Bench", "Pullups Bar", "No Equipment"]
+    "Available Equipment",
+    ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
 )
 
 fitness_level = st.radio(
-    "Select Your Fitness Level",
+    "Fitness Level",
     ["Beginner", "Intermediate", "Advanced"]
 )
 
 # -------------------------
-# BMI Calculation Function
-# -------------------------
-
-def calculate_bmi(weight, height_cm):
-    height_m = height_cm / 100  # Convert cm to meters
-    bmi = weight / (height_m ** 2)
-    return round(bmi, 2)
-
-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"
-    else:
-        return "Obese"
-
-# -------------------------
-# Submit Button
+# Generate Plan
 # -------------------------
 
 if st.button("Generate Workout Plan"):
 
-    bmi = calculate_bmi(weight_kg, height_cm)
-    category = bmi_category(bmi)
+    if not name or height <= 0 or weight <= 0:
+        st.error("Please fill all fields properly")
+    else:
+
+        prompt, bmi, status = build_prompt(
+            name, gender, height, weight, goal, fitness_level, equipment
+        )
 
-    st.success(f"BMI: {bmi} ({category})")
+        st.write("### 📊 BMI:", round(bmi, 2), "-", status)
 
-    workout_plan = generate_5day_plan(
-        name, bmi, category, fitness_level, equipment
-    )
+        with st.spinner("Generating your personalized plan..."):
+            plan = generate_plan(prompt, tokenizer, model)
 
-    st.subheader("🏋️ Your Personalized Workout Plan")
-    st.text(workout_plan)
+        st.success("Your Personalized Workout Plan")
+        st.write(plan)