Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,92 +1,58 @@
-import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-
-# -----------------------------
-# PAGE CONFIG
-# -----------------------------
-st.set_page_config(page_title="FitPlan AI", page_icon="💪", layout="centered")
-
-# -----------------------------
-# BACKGROUND + STYLES
-# -----------------------------
-st.markdown("""
-<style>
-.stApp {
-    background: linear-gradient(to right, #ff512f, #dd2476);
-}
-
-/* Title */
-.main-title {
-    font-size: 40px;
-    font-weight: bold;
-    text-align: center;
-    color: white;
-}
-
-/* Workout Output Box */
-.workout-box {
-    background-color: #111;
-    padding: 25px;
-    border-radius: 15px;
-    border: 1px solid #444;
-    color: white;
-    font-size: 18px;
-    line-height: 1.7;
-}
-
-/* Section Title */
-.title-box {
-    font-size: 28px;
-    font-weight: bold;
-    color: #ff7b00;
-    margin-top: 20px;
-}
-</style>
-""", unsafe_allow_html=True)
-
-st.markdown('<div class="main-title">💪 FitPlan AI</div>', unsafe_allow_html=True)
-st.write("### Fitness Profile & BMI Calculator")
-
-# -----------------------------
-# LOAD MODEL (HUGGING FACE)
-# -----------------------------
+  import streamlit as st
+from transformers import pipeline
+
+# Load model only once
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
-    model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-large")
-    return tokenizer, model
+    return pipeline("text-generation", model="google/flan-t5-base")
+
+generator = load_model()
+
+st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
+st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
+
+st.write("Fill in your details to calculate your BMI and get AI workout plan.")
 
-tokenizer, model = load_model()
+# -------------------------
+# 1. Personal Information
+# -------------------------
 
-# -----------------------------
-# USER INPUT
-# -----------------------------
 name = st.text_input("Enter Your Name *")
-gender = st.selectbox("Gender", ["Male", "Female", "Other"])
-height_cm = st.number_input("Height (cm)", min_value=0.0)
-weight_kg = st.number_input("Weight (kg)", min_value=0.0)
+
+gender = st.selectbox("Select Gender", ["Male", "Female", "Other"])
+
+height_cm = st.number_input("Enter Height (in centimeters) *", min_value=0.0, format="%.2f")
+weight_kg = st.number_input("Enter Weight (in kilograms) *", min_value=0.0, format="%.2f")
+
+# -------------------------
+# 2. Fitness Details
+# -------------------------
 
 st.subheader("Fitness Details")
 
 goal = st.selectbox(
-    "Fitness Goal",
-    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexibility"]
+    "Select Your Fitness Goal",
+    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
 )
 
 equipment = st.multiselect(
     "Available Equipment",
-    ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope",
-     "Weight Plates", "Cycling", "Inclined Bench", "Pull-up Bar", "No Equipment"]
+    ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
 )
 
-fitness_level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
+fitness_level = st.radio(
+    "Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"]
+)
+
+# -------------------------
+# BMI Functions
+# -------------------------
 
-# -----------------------------
-# BMI FUNCTIONS
-# -----------------------------
-def calculate_bmi(weight, height):
-    height_m = height / 100
-    return round(weight / (height_m ** 2), 2)
+def calculate_bmi(weight, height_cm):
+    height_m = height_cm / 100
+    bmi = weight / (height_m ** 2)
+    return round(bmi, 2)
 
 def bmi_category(bmi):
     if bmi < 18.5:
@@ -98,20 +64,23 @@ def bmi_category(bmi):
     else:
         return "Obese"
 
-# -----------------------------
-# SUBMIT BUTTON
-# -----------------------------
-if st.button("Generate Fitness Plan 💥"):
+# -------------------------
+# Submit Button
+# -------------------------
 
+if st.button("Submit Profile"):
+
+    # Validation
     if not name:
-        st.error("Enter your name")
+        st.error("Please enter your name.")
     elif height_cm <= 0 or weight_kg <= 0:
-        st.error("Enter valid height & weight")
+        st.error("Enter valid height and weight.")
     elif not equipment:
-        st.error("Select equipment")
+        st.error("Select at least one equipment option.")
     else:
-        st.success("Profile Submitted Successfully!")
+        st.success("✅ Profile Submitted Successfully!")
 
+        # Calculate BMI
         bmi = calculate_bmi(weight_kg, height_cm)
         bmi_status = bmi_category(bmi)
 
@@ -119,55 +88,22 @@ if st.button("Generate Fitness Plan 💥"):
 
         equipment_list = ", ".join(equipment)
 
-        # -----------------------------
-        # PROMPT FOR AI
-        # -----------------------------
         prompt = f"""
-Create a STRICT 5-day structured workout plan.
-
-Format EXACTLY like:
-
-Day 1:
-Warm-up:
-Exercises:
-Sets & Reps:
-Rest:
-
-Day 2:
-...
-
-User Details:
-Name: {name}
-Gender: {gender}
-BMI: {bmi} ({bmi_status})
-Goal: {goal}
-Fitness Level: {fitness_level}
-Equipment: {equipment_list}
-"""
-
-        # -----------------------------
-        # GENERATE PLAN
-        # -----------------------------
+        Generate a 5-day structured workout plan.
+
+        User Details:
+        Name: {name}
+        Gender: {gender}
+        BMI: {bmi} ({bmi_status})
+        Goal: {goal}
+        Fitness Level: {fitness_level}
+        Available Equipment: {equipment_list}
+
+        Include warmup, exercises with sets & reps, rest time.
+        """
+
         with st.spinner("Generating your AI workout plan..."):
-            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
-            outputs = model.generate(
-                **inputs,
-                max_new_tokens=600,
-                temperature=0.7,
-                do_sample=True
-            )
-            result = tokenizer.decode(outputs[0], skip_special_tokens=True)
-
-        # -----------------------------
-        # DISPLAY OUTPUT (Styled)
-        # -----------------------------
-        st.markdown(
-            '<div class="title-box">🏋️ Your Personalized Workout Plan</div>',
-            unsafe_allow_html=True
-        )
-
-        st.markdown(f"""
-        <div class="workout-box">
-        {result.replace("Day", "<br><br><b>Day").replace("\n", "<br>")}
-        </div>
-        """, unsafe_allow_html=True)
+            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
+
+        st.subheader("🏋️ Your Personalized Workout Plan")
+        st.write(result)