Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,7 +4,7 @@ import torch
 
 st.set_page_config(page_title="FitPlan AI", layout="centered")
 
-# LOAD MODEL (FIXED FOR FLAN-T5)
+# LOAD MODEL 
 @st.cache_resource
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
@@ -13,15 +13,11 @@ def load_model():
 
 tokenizer, model = load_model()
 
-# ---------------------------------------------------
 # TITLE
-# ---------------------------------------------------
-st.title("💪 FitPlan AI – User Fitness Profile")
+st.title(" FitPlan AI – User Fitness Profile")
 
-# ---------------------------------------------------
 # PERSONAL INFORMATION
-# ---------------------------------------------------
-st.subheader("👤 Personal Information")
+st.subheader(" Personal Information")
 
 name = st.text_input("Enter Your Name")
 
@@ -54,9 +50,7 @@ with col2:
         step=0.1
     )
 
-# ---------------------------------------------------
-# BMI FUNCTION
-# ---------------------------------------------------
+#BMI Function
 def bmi_category(bmi):
     if bmi < 18.5:
         return "Underweight"
@@ -67,9 +61,7 @@ def bmi_category(bmi):
     else:
         return "Obese"
 
-# ---------------------------------------------------
 # BMI CALCULATION
-# ---------------------------------------------------
 bmi = None
 
 if height > 0 and weight > 0:
@@ -79,9 +71,7 @@ if height > 0 and weight > 0:
     st.metric("📊 Your BMI", f"{bmi:.2f}")
     st.info(f"BMI Category: {bmi_category(bmi)}")
 
-# ---------------------------------------------------
 # FITNESS GOAL
-# ---------------------------------------------------
 st.subheader("🎯 Fitness Goal")
 
 goal = st.selectbox(
@@ -95,10 +85,9 @@ goal = st.selectbox(
     ]
 )
 
-# ---------------------------------------------------
 # EQUIPMENT
-# ---------------------------------------------------
-st.subheader("🏋️ Available Equipment")
+
+st.subheader(" Available Equipment")
 
 equipment_map = {}
 
@@ -130,9 +119,8 @@ with col3:
 
 equipment = [item for item, selected in equipment_map.items() if selected]
 
-# ---------------------------------------------------
+
 # FITNESS LEVEL
-# ---------------------------------------------------
 st.subheader("📈 Fitness Level")
 
 fitness_level = st.radio(
@@ -141,10 +129,8 @@ fitness_level = st.radio(
     horizontal=True
 )
 
-# ---------------------------------------------------
 # SUBMIT BUTTON
-# ---------------------------------------------------
-if st.button("🚀 Submit Profile"):
+if st.button(" Submit Profile"):
 
     if not name:
         st.error("Please enter your name.")
@@ -156,12 +142,11 @@ 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)
 
-        # Improved Prompt
         prompt = f"""
 You are a certified professional fitness trainer.
 
@@ -180,14 +165,17 @@ Day 1:
 """
 
         with st.spinner("Generating your AI workout plan..."):
+
             inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
 
-outputs = model.generate(
-    **inputs,
-    max_new_tokens=400,
-    temperature=0.7,
-    do_sample=True
-)
+            outputs = model.generate(
+                **inputs,
+                max_new_tokens=300,
+                temperature=0.7,
+                do_sample=True
+            )
 
-result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
+            result = tokenizer.decode(outputs[0], skip_special_tokens=True).strip()
 
+        st.subheader(" Your Personalized Workout Plan")
+        st.write(result)