Update app.py

app.py

@@ -3,25 +3,33 @@ from prompt.bmi import calculate_bmi, bmi_category
 from prompt.prompt_builder import build_prompt
 from prompt.generator import load_model, generate_plan
 
+# -------------------------
+# Page Config
+# -------------------------
+
 st.set_page_config(page_title="FitPlan AI", page_icon="💪")
 
-st.title("💪 FitPlan AI - Personalized Workout Generator")
+st.title("💪 FitPlan AI — Personalized Workout Generator")
+
+# -------------------------
+# Load AI Model
+# -------------------------
 
-# Load model
 tokenizer, model = load_model()
 
 # -------------------------
 # User Inputs
 # -------------------------
 
-name = st.text_input("Name")
+name = st.text_input("Enter Your Name")
+
 gender = st.selectbox("Gender", ["Male", "Female"])
 
 height = st.number_input("Height (cm)", min_value=0.0)
 weight = st.number_input("Weight (kg)", min_value=0.0)
 
 goal = st.selectbox(
-    "Fitness Goal",
+    "Select Your Fitness Goal",
     ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
 )
 
@@ -36,23 +44,24 @@ fitness_level = st.radio(
 )
 
 # -------------------------
-# Generate Plan
+# Generate Plan Button
 # -------------------------
 
 if st.button("Generate Workout Plan"):
 
     if not name or height <= 0 or weight <= 0:
-        st.error("Please fill all fields properly")
+        st.error("⚠ Please fill all required fields correctly!")
     else:
 
         prompt, bmi, status = build_prompt(
             name, gender, height, weight, goal, fitness_level, equipment
         )
 
-        st.write("### 📊 BMI:", round(bmi, 2), "-", status)
+        st.write(f"### 📊 BMI: {round(bmi,2)} ({status})")
 
-        with st.spinner("Generating your personalized plan..."):
-            plan = generate_plan(prompt, tokenizer, model)
+        with st.spinner("Generating your AI workout plan..."):
+            result = generate_plan(prompt, tokenizer, model)
 
-        st.success("Your Personalized Workout Plan")
-        st.write(plan)
+        st.success("🏆 Your Personalized Workout Plan")
+        st.write(result)
+