Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,29 +1,16 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
+import time
 
 # -------------------------
-# Page Config FIRST
+# Page Config
 # -------------------------
 
 st.set_page_config(page_title="FitPlan AI", page_icon="💪")
 
-# 🎨 Background Style
-
-st.markdown(
-    """
-    <style>
-    .stApp {
-        background-image: url("https://images.unsplash.com/photo-1554284126-aa88f22d8b74");
-        background-size: cover;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
 # -------------------------
-# Load Model (Small for HF)
+# Load Model
 # -------------------------
 
 @st.cache_resource
@@ -35,11 +22,10 @@ def load_model():
 tokenizer, model = load_model()
 
 # -------------------------
-# App Title
+# Title
 # -------------------------
 
 st.title("💪 FitPlan AI - BMI Calculator & Workout Planner")
-st.write("Enter your details to get AI workout plan.")
 
 # -------------------------
 # Personal Info
@@ -94,6 +80,7 @@ def bmi_category(bmi):
 
 if st.button("Generate Workout Plan"):
 
+    # Validation
     if not name:
         st.error("Enter your name")
     elif height_cm <= 0 or weight_kg <= 0:
@@ -101,31 +88,64 @@ if st.button("Generate Workout Plan"):
     elif not equipment:
         st.error("Select equipment")
     else:
+
+        # BMI
         bmi = calculate_bmi(weight_kg, height_cm)
         bmi_status = bmi_category(bmi)
 
-        st.success(f"BMI: {bmi} ({bmi_status})")
+        st.success(f"✅ Welcome {name}! BMI: {bmi} ({bmi_status})")
 
-        equipment_list = ", ".join(equipment)
+        # -------------------------
+        # Progress Bar Animation
+        # -------------------------
 
-        prompt = f"""
-        Create a 5-day workout plan.
+        progress = st.progress(0)
+        status = st.empty()
+
+        for i in range(100):
+            time.sleep(0.01)
+            progress.progress(i + 1)
+            status.text("Generating personalized fitness plan...")
+
+        st.divider()
+
+        # -------------------------
+        # Core AI Module Layout
+        # -------------------------
+
+        st.subheader("🧠 Core AI Inference Module")
+
+        col1, col2, col3 = st.columns(3)
 
-        Include warmup, exercises, sets, reps, rest time.
-        
-Day 1 – Chest & Triceps: Bench Press 4×8–10, Incline Press 3×10, Dips 3×12, Tricep Extensions 3×12.
+        with col1:
+            st.info("🔽 Input Layer\nUser goals, equipment, fitness level")
 
-Day 2 – Back & Biceps: Pull-ups 4×8–10, Rows 4×10, Band Rows 3×12, Bicep Curls 3×12.
+        with col2:
+            st.info("⚙ Processing\nAI model analysis")
 
-Day 3 – Legs: Squats 4×8–10, Lunges 3×12/leg, Romanian Deadlift 3×10, Calf Raises 4×15. 
+        with col3:
+            st.info("📤 Output Layer\nWorkout plan")
 
-Day 4 – Shoulders & Core: Overhead Press 4×8–10, Lateral Raises 3×12, Plank 3×60s, Leg Raises 3×12.
- 
-Day 5 – Cardio & Conditioning: 25-min Run, Skipping 3×2min, Burpees 3×12, Mountain Climbers 3×30s.
-        
+        st.divider()
+
+        # -------------------------
+        # Model Input & Output
+        # -------------------------
+
+        equipment_list = ", ".join(equipment)
+
+        prompt = f"""
+        Create a structured 5-day workout plan.
+
+        User:
+        Goal: {goal}
+        Fitness Level: {fitness_level}
+        Equipment: {equipment_list}
+
+        Include exercises with sets, reps, and rest time.
         """
 
-        with st.spinner("Generating workout plan..."):
+        with st.spinner("AI is creating your plan..."):
             inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
             outputs = model.generate(
                 **inputs,
@@ -135,5 +155,14 @@ Day 5 – Cardio & Conditioning: 25-min Run, Skipping 3×2min, Burpees 3×12, Mo
             )
             result = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-        st.subheader("🏋️ Your Workout Plan")
-        st.write(result)
+        col4, col5 = st.columns(2)
+
+        with col4:
+            st.subheader("📥 Model Input")
+            st.write(prompt)
+
+        with col5:
+            st.subheader("📤 Model Output")
+            st.write(result)
+
+        st.success("🏋️ Your Personalized Workout Plan Ready!")