Update app.py

app.py

@@ -102,26 +102,34 @@ if submit:
         # -------------------------
         with st.spinner("Generating your 5-day workout plan..."):
 
-            full_plan = ""
-
-            for day in range(1, 6):
-
-                prompt = f"""
-You are a certified professional fitness trainer.
-Create a detailed workout plan for Day {day}.
-Include:
-- Warm-up
-- 4 to 6 exercises
-- Sets and reps
-- Cool down
-User Details:
-BMI: {bmi} ({bmi_status})
-Goal: {goal}
-Fitness Level: {level}
-Equipment Available: {equipment_list}
-Start with:
-Day {day}:
-"""
+            from prompt_builder import build_prompt
+
+with st.spinner("Generating your 5-day workout plan..."):
+
+    prompt, bmi, bmi_status = build_prompt(
+        name=name,
+        gender="Not specified",  # or add gender field in form
+        height=height,
+        weight=weight,
+        goal=goal,
+        fitness_level=level,
+        equipment=equipment
+    )
+
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+
+    outputs = model.generate(
+        **inputs,
+        max_new_tokens=800,   # increase token limit
+        temperature=0.7,
+        do_sample=True,
+        repetition_penalty=1.1
+    )
+
+    full_plan = tokenizer.decode(
+        outputs[0],
+        skip_special_tokens=True
+    ).strip()
 
                 inputs = tokenizer(prompt, return_tensors="pt", truncation=True)