Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,141 @@
+import streamlit as st
+
+from transformers import pipeline
+
+
+
+# -------------------------
+
+# PAGE CONFIG
+
+# -------------------------
+
+st.set_page_config(page_title="FitPlan-AI", page_icon="💪")
+
+
+
+# -------------------------
+
+# BMI FUNCTIONS
+
+# -------------------------
+
+def calculate_bmi(weight, height_cm):
+
+    height_m = height_cm / 100
+
+    return round(weight / (height_m ** 2), 2)
+
+
+
+def get_category(bmi):
+
+    if bmi < 18.5:
+
+        return "Underweight"
+
+    elif bmi < 24.9:
+
+        return "Normal"
+
+    elif bmi < 29.9:
+
+        return "Overweight"
+
+    else:
+
+        return "Obese"
+
+
+
+# -------------------------
+
+# LOAD AI MODEL
+
+# -------------------------
+
+@st.cache_resource
+
+def load_model():
+
+    return pipeline(
+
+        "text2text-generation",  # Correct for FLAN-T5
+
+        model="google/flan-t5-base"
+
+    )
+
+
+
+generator = load_model()
+
+
+
+# -------------------------
+
+# UI STARTS
+
 # -------------------------
+
+st.title("💪 FitPlan-AI: Personalized Fitness Profile")
+
+
+
+# Form to collect all inputs
+
+with st.form("fitness_form"):
+
+
+
+    st.subheader("Personal Information")
+
+    name = st.text_input("Full Name*", placeholder="Enter your name")
+
+
+
+    col1, col2 = st.columns(2)
+
+    with col1:
+
+        height = st.number_input("Height (cm)*", min_value=1.0, step=0.1)
+
+    with col2:
+
+        weight = st.number_input("Weight (kg)*", min_value=1.0, step=0.1)
+
+
+
+    st.subheader("Fitness Details")
+
+    goal = st.selectbox(
+
+        "Fitness Goal",
+
+        ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
+
+    )
+
+    level = st.radio("Fitness Level", ["Beginner", "Intermediate", "Advanced"])
+
+    equipment = st.multiselect(
+
+        "Available Equipment",
+
+        ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment", "Kettlebell", "Pull-up Bar"]
+
+    )
+
+
+
+    # -------------------------
+
+    # SUBMIT BUTTON AT BOTTOM
+
+    # -------------------------
+
+    submit = st.form_submit_button("Submit Profile")
+    # -------------------------
     # HANDLE SUBMISSION
     # -------------------------
     if submit:
@@ -41,4 +178,5 @@
                 result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
 
             st.subheader("🏋️ Your Personalized Workout Plan")
-            st.write(result)
+            st.write(result)
+