Update app.py

app.py

@@ -1,42 +1,44 @@
 import streamlit as st
+import sys
+import os
 
-# IMPORT your existing functions
+# ---------- ADD EXTERNAL PATHS ----------
+sys.path.append(os.path.abspath("../model"))
+sys.path.append(os.path.abspath("../prompt"))
+
+# ---------- IMPORT YOUR FILES ----------
 from query_model import query_model
-from prompt_builder import build_prompt, calculate_bmi
+from prompt_builder import build_prompt, calculate_bmi, bmi_category
 
-st.set_page_config(page_title="AI Workout Planner", layout="centered")
 
+# ---------- STREAMLIT UI ----------
 st.title("🏋️ AI Personalized 5-Day Workout Planner")
 
-# ---------- USER INPUT ----------
 name = st.text_input("Name")
-age = st.number_input("Age", min_value=10, max_value=100)
-
 gender = st.selectbox("Gender", ["Male", "Female", "Other"])
 
-height = st.number_input("Height (cm)", min_value=100, max_value=250)
-weight = st.number_input("Weight (kg)", min_value=30, max_value=200)
+height = st.number_input("Height (cm)", min_value=0, max_value=250)
+weight = st.number_input("Weight (kg)", min_value=0, max_value=200)
 
-goal = st.selectbox(
-    "Fitness Goal",
-    ["Weight Loss", "Muscle Gain", "General Fitness", "Strength"]
-)
+goal = st.selectbox("Fitness Goal",["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
 
-fitness_level = st.selectbox(
-    "Fitness Level",
-    ["Beginner", "Intermediate", "Advanced"]
-)
+fitness_level = st.selectbox("Fitness Level", [
+    "Beginner",
+    "Intermediate",
+    "Advanced"
+])
 
 equipment = st.multiselect(
     "Available Equipment",
-    ["Dumbbells", "Barbell", "Resistance Bands", "Kettlebell", "No Equipment"]
+   ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope",
+     "Weight Plates", "Cycling", "Inclined Bench", "Pullups Bar", "No Equipment"]
 )
 
 # ---------- GENERATE PLAN ----------
 if st.button("Generate 5-Day Plan 💪"):
 
     prompt, bmi, bmi_status = build_prompt(
-        name, age, gender, height, weight,
+        name, gender, height, weight,
         goal, fitness_level, equipment
     )
 
@@ -45,5 +47,5 @@ if st.button("Generate 5-Day Plan 💪"):
     with st.spinner("Creating your personalized workout plan..."):
         result = query_model(prompt)
 
-    st.markdown("## 🏋️ Your Personalized Workout Plan")
-    st.markdown(result)
+    st.markdown("## 🗓️ Your 5-Day Workout Plan")
+    st.write(result)