Update src/streamlit_app.py

src/streamlit_app.py

@@ -13,6 +13,7 @@ def load_model():
 
 tokenizer, model = load_model()
 
+
 # TITLE
 st.title(" FitPlan AI – User Fitness Profile")
 
@@ -129,63 +130,70 @@ fitness_level = st.radio(
     horizontal=True
 )
 
-# SUBMIT BUTTON
-if st.button(" Submit Profile"):
-
-    if not name:
-        st.error("Please enter your name.")
+def create_prompt(gender, bmi, bmi_status, goal, level, equipment):
 
-    elif height <= 0 or weight <= 0:
-        st.error("Please enter valid height and weight.")
+    equipment_list = ", ".join(equipment)
 
-    elif not equipment:
-        st.error("Please select at least one equipment option.")
-
-    else:
-        st.success(" Profile Submitted Successfully!")
-
-        bmi_status = bmi_category(bmi)
-        equipment_list = ", ".join(equipment)
-
-        prompt = f"""
+    prompt = f"""
 Instruction:
+You are a certified professional fitness trainer.
+
 Generate a structured 5-day workout plan.
 
-User:
+User Information:
 Gender: {gender}
 BMI: {bmi:.2f} ({bmi_status})
 Goal: {goal}
-Level: {fitness_level}
-Equipment: {equipment_list}
+Fitness Level: {level}
+Equipment Available: {equipment_list}
 
 Rules:
-- Only 5 days.
-- Each day must include:
-  Focus
-  4 Exercises
-  Sets x Reps
-  Rest
-- No repetition.
-- No extra explanation.
+1. Only generate Day 1 to Day 5.
+2. Each day must include:
+   - Focus
+   - 4 exercises
+   - Sets x Reps
+   - Rest time
+3. Do not repeat days.
+4. Do not generate explanations.
+5. Keep output clean and professional.
 
 Start directly with:
 Day 1:
 """
 
+    return prompt
 
-        with st.spinner("Generating your AI workout plan..."):
+def generate_workout_plan(prompt):
 
-            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
 
-            outputs = model.generate(
-                **inputs,
-                max_new_tokens=400,
-                num_beams=4,
-                early_stopping=True,
-                no_repeat_ngram_size=3
-            )
+    outputs = model.generate(
+        **inputs,
+        max_new_tokens=400,
+        num_beams=4,
+        early_stopping=True,
+        no_repeat_ngram_size=3
+    )
+
+    result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+    return result
+
+
+if st.button("Submit Profile"):
+
+    prompt = create_prompt(
+        gender,
+        bmi,
+        bmi_status,
+        goal,
+        fitness_level,
+        equipment
+    )
 
-            result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+    with st.spinner("Generating AI workout plan..."):
+        result = generate_workout_plan(prompt)
 
-        st.subheader(" Your Personalized Workout Plan")
-        st.write(result)
+    st.subheader("Your Personalized Workout Plan")
+    st.write(result)