Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,6 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torch
 # -------------------------
 # PAGE CONFIG
 # -------------------------
@@ -69,54 +70,73 @@ with st.form("fitness_form"):
 # -------------------------
 # HANDLE SUBMISSION
 # -------------------------
-if submit:
+# SUBMIT BUTTON
 
-    if not name.strip():
-        st.error("Please enter your name.")
+if st.button(" Submit Profile"):
+ 
+    if not name:
 
+        st.error("Please enter your name.")
+ 
     elif height <= 0 or weight <= 0:
-        st.error("Height and Weight must be positive values.")
 
+        st.error("Please enter valid height and weight.")
+ 
     elif not equipment:
-        st.error("Please select at least one equipment option.")
 
+        st.error("Please select at least one equipment option.")
+ 
     else:
-        # Calculate BMI
-        user_bmi = calculate_bmi(weight, height)
-        bmi_status = get_category(user_bmi)
 
-        st.success(f"✅ Profile Created Successfully for {name}!")
-        st.metric("Your BMI", user_bmi)
-        st.info(f"Health Category: **{bmi_status}**")
+        st.success(" Profile Submitted Successfully!")
+ 
+        bmi_status = bmi_category(bmi)
 
         equipment_list = ", ".join(equipment)
-
-        # FLAN-T5 works best with instruction-style prompts
+ 
         prompt = f"""
-        Create a structured 5-day workout plan.
-
-        User Details:
-        Name: {name}
-        BMI: {user_bmi} ({bmi_status})
-        Goal: {goal}
-        Fitness Level: {level}
-        Equipment: {equipment_list}
-
-        Include:
-        - Warm-up
-        - Exercises with sets and reps
-        - Rest time
-        - Day-wise breakdown
-        - Safety advice
-        """
 
+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..."):
-            result = generator(
-                prompt,
-                max_new_tokens=400,
-                do_sample=True,
-                temperature=0.7
-            )[0]["generated_text"]
-
-        st.subheader("🏋️ Your Personalized Workout Plan")
-        st.write(result)
+ 
+            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+ 
+            outputs = model.generate(
+
+                **inputs,
+
+                max_new_tokens=900,
+
+                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)
+