Update app.py

app.py

@@ -1,11 +1,10 @@
 import streamlit as st
-import sys
-import os
 
 # ---------- ADD EXTERNAL PATHS ----------
 sys.path.append(os.path.abspath("../model"))
 sys.path.append(os.path.abspath("../prompt"))
 
+
 # ---------- IMPORT YOUR FILES ----------
 from model_api import query_model
 from prompt_builder import build_prompt, calculate_bmi, bmi_category
@@ -15,6 +14,7 @@ from prompt_builder import build_prompt, calculate_bmi, bmi_category
 st.title("🏋️ AI Personalized 5-Day Workout Planner")
 
 name = st.text_input("Name")
+age = st.number_input("Age", min_value=0, max_value=100)
 gender = st.selectbox("Gender", ["Male", "Female", "Other"])
 
 height = st.number_input("Height (cm)", min_value=0, max_value=250)
@@ -38,9 +38,9 @@ equipment = st.multiselect(
 if st.button("Generate 5-Day Plan 💪"):
 
     prompt, bmi, bmi_status = build_prompt(
-        name, gender, height, weight,
-        goal, fitness_level, equipment
-    )
+    name, age, gender, height, weight,
+    goal, fitness_level, equipment
+)
 
     st.subheader(f"Your BMI: {bmi:.2f} ({bmi_status})")