Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,10 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import torch
+
 # 1. Basic Configuration
 st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
+
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
     model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
@@ -115,10 +118,8 @@ with st.form("fitness_form", clear_on_submit=False):
     
     level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"], horizontal=True)
 
-    # Submission Button
-    submit = st.form_submit_button("GENERATE MY PLAN")
-
-if st.button("🚀 Submit Profile"):
+    # SUBMIT BUTTON
+if st.button(" Submit Profile"):
  
     if not name:
         st.error("Please enter your name.")
@@ -130,33 +131,40 @@ if st.button("🚀 Submit Profile"):
         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)
  
         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(prompt, max_new_tokens=400)
-            result = response[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)