Update src/streamlit_app.py

src/streamlit_app.py

@@ -166,24 +166,23 @@ if st.button("🚀 Submit Profile"):
         bmi_status = bmi_category(bmi)
         equipment_list = ", ".join(equipment)
 
+       # Improved Prompt
         prompt = f"""
-        Generate a 5-day structured workout plan.
-
-        User Details:
-        Name: {name}
-        Gender: {gender}
-        BMI: {bmi:.2f} ({bmi_status})
-        Goal: {goal}
-        Fitness Level: {fitness_level}
-        Available Equipment: {equipment_list}
-
-        Requirements:
-        - Include warmup
-        - Include exercises with sets and reps
-        - Include rest time
-        - Adjust intensity based on BMI and fitness level
-        - Keep it structured day-wise
-        """
+You are a certified professional fitness trainer.
+
+Create a detailed 5-day workout plan.
+
+User Information:
+- Gender: {gender}
+- BMI: {bmi:.2f} ({bmi_status})
+- Goal: {goal}
+- Fitness Level: {fitness_level}
+- Equipment Available: {equipment_list}
+
+Start directly with:
+
+Day 1:
+"""
 
         with st.spinner("Generating your AI workout plan..."):
             response = generator(
@@ -195,5 +194,9 @@ if st.button("🚀 Submit Profile"):
 
             result = response[0]["generated_text"]
 
+            # Remove prompt if model repeats it
+            if result.startswith(prompt):
+                result = result[len(prompt):].strip()
+
         st.subheader("🏋️ Your Personalized Workout Plan")
         st.write(result)