Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,48 +1,89 @@
-import streamlit st
+import streamlit as st
 from transformers import pipeline
-def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/flan-t5-base"
-    )
- 
-generator = load_model()
-  if not name:
-        st.error("Please enter your name.")
- 
-    elif not equipment:
-        st.error("Please select at least one equipment option.")
- 
-    elif bmi is None:
-        st.error("Please enter valid height and weight.")
- 
+st.set_page_config(page_title="Fitness Profile", page_icon="🏋️", layout="centered")
+
+st.title("🏋️ Personalized Fitness Profile")
+
+st.markdown("---")
+
+# =========================
+# 1️⃣ Personal Information
+# =========================
+st.header("1. Personal Information")
+
+name = st.text_input("Name *")
+
+height_cm = st.number_input("Height (in centimeters) *", min_value=0.0, format="%.2f")
+weight_kg = st.number_input("Weight (in kilograms) *", min_value=0.0, format="%.2f")
+
+# =========================
+# 2️⃣ Fitness Details
+# =========================
+st.header("2. Fitness Details")
+
+fitness_goal = st.selectbox(
+    "Fitness Goal",
+    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
+)
+
+equipment = st.multiselect(
+    "Available Equipment (Multiple selection allowed)",
+    ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
+)
+
+fitness_level = st.radio(
+    "Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"],
+    horizontal=True
+)
+
+st.markdown("---")
+
+# =========================
+# 3️⃣ Submit Button
+# =========================
+if st.button("Generate Profile"):
+
+    # =========================
+    # Validation
+    # =========================
+    if name.strip() == "":
+        st.error("❌ Name is required.")
+    elif height_cm <= 0:
+        st.error("❌ Height must be greater than zero.")
+    elif weight_kg <= 0:
+        st.error("❌ Weight must be greater than zero.")
     else:
-        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
-        """
- 
-        with st.spinner("Generating your AI workout plan..."):
-            result = generator(prompt, max_new_tokens=400)[0]["generated_text"]
- 
-        st.subheader("🏋️ Your Personalized Workout Plan")
-        st.write(result)
+        # =========================
+        # BMI Calculation
+        # =========================
+        height_m = height_cm / 100  # Convert cm to meters
+        bmi = weight_kg / (height_m ** 2)
+        bmi = round(bmi, 2)
+
+        # =========================
+        # BMI Category
+        # =========================
+        if bmi < 18.5:
+            category = "Underweight"
+        elif 18.5 <= bmi < 24.9:
+            category = "Normal"
+        elif 25 <= bmi < 29.9:
+            category = "Overweight"
+        else:
+            category = "Obese"
+
+        # =========================
+        # Display Results
+        # =========================
+        st.success("✅ Profile Generated Successfully!")
+
+        st.subheader(f"👤 {name}'s Fitness Summary")
+
+        st.write(f"**Height (m):** {round(height_m, 2)} m")
+        st.write(f"**Weight (kg):** {weight_kg} kg")
+        st.write(f"**BMI:** {bmi}")
+        st.write(f"**BMI Category:** {category}")
+        st.write(f"**Fitness Goal:** {fitness_goal}")
+        st.write(f"**Fitness Level:** {fitness_level}")
+        st.write(f"**Available Equipment:** {', '.join(equipment) if equipment else 'None Selected'}")