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 Configuration
@@ -98,49 +99,73 @@ st.markdown("---")
 # =========================
 # Submit Button
 # =========================
-if st.button("Submit Profile"):
+# SUBMIT BUTTON
 
+if st.button(" Submit Profile"):
+ 
     if not name:
+
         st.error("Please enter your name.")
+ 
+    elif height <= 0 or weight <= 0:
 
-    elif height_cm <= 0 or weight_kg <= 0:
         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:
-        st.success("✅ Profile Submitted Successfully!")
 
+        st.success(" Profile Submitted Successfully!")
+ 
         bmi_status = bmi_category(bmi)
-        equipment_list = ", ".join(equipment)
 
+        equipment_list = ", ".join(equipment)
+ 
         prompt = f"""
-        Generate a 5-day structured workout plan.
-
-        User Details:
-        Name: {name}
-        Gender: {gender}
-        BMI: {bmi:.2f} ({bmi_status})
-        Goal: {fitness_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..."):
-            result = generator(
-                prompt,
-                max_new_tokens=400,
-                do_sample=True,
-                temperature=0.7
-            )[0]["generated_text"]
-
-        st.subheader("🏋️ Your Personalized Workout Plan")
+ 
+            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)
+