Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,52 +1,85 @@
 import streamlit as st
 from transformers import pipeline
-def load_model():
-    return pipeline(
-        "text-generation",
-        model="google/flan-t5-base"
-    )
- 
-generator = load_model()
-
-if st.button("Submit Profile"):
- 
-    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.")
- 
+import streamlit as st
+
+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 fitness category.")
+
+# -------------------------
+# 1. Personal Information
+# -------------------------
+
+name = st.text_input("Enter Your Name *")
+
+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(
+    "Select Your Fitness Goal",
+    ["Build Muscle", "Weight Loss", "Strength Gain", "Abs Building", "Flexible"]
+)
+
+equipment = st.multiselect(
+    "Available Equipment (Select multiple if available)",
+    ["Dumbbells", "Resistance Band", "Yoga Mat", "No Equipment"]
+)
+
+fitness_level = st.radio(
+    "Select Your Fitness Level",
+    ["Beginner", "Intermediate", "Advanced"]
+)
+
+# -------------------------
+# BMI Calculation Function
+# -------------------------
+
+def calculate_bmi(weight, height_cm):
+    height_m = height_cm / 100  # Convert cm to meters
+    bmi = weight / (height_m ** 2)
+    return round(bmi, 2)
+
+def bmi_category(bmi):
+    if bmi < 18.5:
+        return "Underweight"
+    elif 18.5 <= bmi < 24.9:
+        return "Normal"
+    elif 25 <= bmi < 29.9:
+        return "Overweight"
     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)
+        return "Obese"
+
+# -------------------------
+# Submit Button
+# -------------------------
+
+if st.button("Calculate BMI"):
+    
+    # Validation
+    if not name or height_cm <= 0 or weight_kg <= 0:
+        st.error("⚠ Please fill all required fields with valid values!")
+    else:
+        bmi = calculate_bmi(weight_kg, height_cm)
+        category = bmi_category(bmi)
+
+        st.success("✅ Calculation Successful!")
+
+        st.write(f"### 👤 Name: {name}")
+        st.write(f"### 📊 Your BMI: {bmi}")
+        st.write(f"### 🏷 BMI Category: {category}")
+
+        st.write("---")
+        st.write("### 🏋 Fitness Summary")
+        st.write(f"**Goal:** {goal}")
+        st.write(f"**Fitness Level:** {fitness_level}")
+        st.write(f"**Equipment Available:** {', '.join(equipment) if equipment else 'None selected'}")
+