Update app.py

app.py

@@ -1,78 +1,58 @@
 import streamlit as st
+from model_api import query_model
+from prompt_builder import build_prompt
 
-st.markdown(
-"""
-<style>
-
-.stApp{
-background-image: url("https://images.unsplash.com/photo-1554284126-aa88f22d8b74");
-background-size: cover;
-}
-
-h1{
-color:white;
-text-align:center;
-}
+st.title("🏋️ AI Personalized 5-Day Workout Planner")
 
-</style>
-""",
-unsafe_allow_html=True
-)
 if "logged_in" not in st.session_state:
     st.session_state.logged_in = False
 
 if not st.session_state.logged_in:
     st.warning("Please login first")
     st.stop()
-# ---------- ADD EXTERNAL PATHS ----------
-sys.path.append(os.path.abspath("../model"))
-sys.path.append(os.path.abspath("../prompt"))
-
-if st.button("Logout"):
-    st.session_state.logged_in = False
-    st.experimental_rerun()
-
-# ---------- IMPORT YOUR FILES ----------
-from model_api import query_model
-from prompt_builder import build_prompt, calculate_bmi, bmi_category
-
-
-# ---------- STREAMLIT UI ----------
-st.title("🏋️ AI Personalized 5-Day Workout Planner")
 
 name = st.text_input("Name")
-age = st.number_input("Age", min_value=0, max_value=100)
+age = st.number_input("Age", 10, 80)
+
 gender = st.selectbox("Gender", ["Male", "Female", "Other"])
 
-height = st.number_input("Height (cm)", min_value=0, max_value=250)
-weight = st.number_input("Weight (kg)", min_value=0, max_value=200)
+height = st.number_input("Height (cm)", 100, 250)
+weight = st.number_input("Weight (kg)", 30, 200)
 
-goal = st.selectbox("Fitness Goal",["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"])
+goal = st.selectbox(
+    "Fitness Goal",
+    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexibility"]
+)
 
-fitness_level = st.selectbox("Fitness Level", [
-    "Beginner",
-    "Intermediate",
-    "Advanced"
-])
+fitness_level = st.selectbox(
+    "Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"]
+)
 
 equipment = st.multiselect(
-    "Available Equipment",
-   ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope",
-     "Weight Plates", "Cycling", "Inclined Bench", "Pullups Bar", "No Equipment"]
+    "Equipment",
+    [
+        "Dumbbells",
+        "Resistance Band",
+        "Yoga Mat",
+        "Skipping Rope",
+        "Pullups Bar",
+        "Inclined Bench",
+        "Weight Plates"
+    ]
 )
 
-# ---------- GENERATE PLAN ----------
 if st.button("Generate 5-Day Plan 💪"):
 
     prompt, bmi, bmi_status = build_prompt(
-    name, age, 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})")
+    st.subheader(f"BMI: {bmi} ({bmi_status})")
 
-    with st.spinner("Creating your personalized workout plan..."):
+    with st.spinner("Generating workout plan..."):
         result = query_model(prompt)
 
     st.markdown("## 🗓️ Your 5-Day Workout Plan")
-    st.write(result)
+    st.markdown(result)