Update src/streamlit_app.py

src/streamlit_app.py

@@ -121,45 +121,43 @@ if st.button("Submit Profile"):
         equipment_list = ", ".join(equipment)
 
         prompt = f"""
-You are a certified professional fitness trainer.
+prompt = f"""
+You are a certified fitness trainer.
 
-Generate a structured 5-day workout plan based on the following user profile.
+Create a STRICTLY formatted 5-day workout plan.
 
-User Profile:
+User Details:
 - Gender: {gender}
 - BMI: {bmi:.2f} ({bmi_status})
-- Goal: {fitness_goal}
+- Fitness Goal: {fitness_goal}
 - Fitness Level: {fitness_level}
-- Available Equipment: {equipment_list}
-
-Instructions:
-1. Divide the plan clearly into Day 1, Day 2, Day 3, Day 4, and Day 5.
-2. Under each day, list 4-6 exercises.
-3. For each exercise include:
-   - Exercise Name
-   - Sets
-   - Reps
-   - Rest Time
-4. Keep exercises appropriate for the user’s fitness level.
-5. Avoid medical or dietary advice.
-6. Keep the output clean, structured, and easy to read.
-7. Do NOT include explanations outside the workout plan.
-
-Only return the workout plan.
+- Equipment Available: {equipment_list}
+
+IMPORTANT RULES:
+1. Output ONLY the workout plan.
+2. Divide into Day 1, Day 2, Day 3, Day 4, Day 5.
+3. Each day must contain EXACTLY 5 exercises.
+4. For each exercise include:
+   Exercise:
+   Sets:
+   Reps:
+   Rest:
+5. Keep exercises suitable for the fitness level.
+6. Do NOT add explanations, tips, introductions, or conclusions.
+7. Keep formatting clean and consistent.
+
+OUTPUT FORMAT (FOLLOW EXACTLY):
+
+Day 1: [Workout Focus]
+1. Exercise: Push-Ups
+   Sets: 3
+   Reps: 12
+   Rest: 60 sec
+
+2. Exercise: Squats
+   Sets: 3
+   Reps: 15
+   Rest: 60 sec
+
+Continue same format for all 5 days.
 """
-
-        with st.spinner("Generating your AI workout plan..."):
-
-            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
-
-            outputs = model.generate(
-                **inputs,
-                max_new_tokens=600,
-                temperature=0.7,
-                do_sample=True
-            )
-
-            result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
-
-        st.subheader("Your Personalized Workout Plan")
-        st.write(result)