Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,36 +1,25 @@
 import streamlit as st
-import torch
-import random
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+st.set_page_config(page_title="FitPlan AI - BMI Calculator", page_icon="💪")
 
-# -------------------------
-# Page Config
-# -------------------------
-
-st.set_page_config(page_title="FitPlan AI", page_icon="💪")
+st.title("💪 FitPlan AI - Fitness Profile & BMI Calculator")
 
-st.title("💪 FitPlan AI — Personalized Workout Generator")
-
-st.write("Fill your details to generate AI workout plan.")
+st.write("Fill in your details to calculate your BMI and fitness category.")
 
 # -------------------------
-# Load Model (Small for HF Spaces)
+# 1. Personal Information
 # -------------------------
 
-@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
+name = st.text_input("Enter Your Name *")
 
-tokenizer, model = load_model()
+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")
 
 # -------------------------
-# User Inputs
+# 2. Fitness Details
 # -------------------------
 
-name = st.text_input("Enter Your Name")
+st.subheader("Fitness Details")
 
 goal = st.selectbox(
     "Select Your Fitness Goal",
@@ -38,87 +27,56 @@ goal = st.selectbox(
 )
 
 equipment = st.multiselect(
-    "Available Equipment",
+    "Available Equipment (Select multiple if available)",
     ["Dumbbells", "Resistance Band", "Yoga Mat", "Skipping Rope",
      "Weight Plates", "Cycling", "Inclined Bench", "Pullups Bar", "No Equipment"]
 )
 
 fitness_level = st.radio(
-    "Fitness Level",
+    "Select Your Fitness Level",
     ["Beginner", "Intermediate", "Advanced"]
 )
 
 # -------------------------
-# Function: Create Prompt
+# BMI Calculation Function
 # -------------------------
 
-def create_prompt(goal, equipment, fitness_level):
-
-    equipment_list = ", ".join(equipment)
-
-    # Goal instructions
-    goal_map = {
-        "Build Muscle": "Focus on muscle hypertrophy training.",
-        "Weight Loss": "Focus on fat burning and cardio workouts.",
-        "Strength Gain": "Focus on heavy compound strength exercises.",
-        "Abs Building": "Focus on core and abdominal exercises.",
-        "Flexible": "Focus on stretching and mobility exercises."
-    }
-
-    # Level instructions
-    level_map = {
-        "Beginner": "Use simple low intensity exercises.",
-        "Intermediate": "Use moderate intensity exercises.",
-        "Advanced": "Use high intensity advanced exercises."
-    }
-
-    prompt = f"""
-    Generate a 5-day workout plan.
-
-    Goal: {goal}
-    Fitness Level: {fitness_level}
-    Equipment: {equipment_list}
-
-    Goal Instruction:
-    {goal_map[goal]}
-
-    Level Instruction:
-    {level_map[fitness_level]}
-
-    Requirements:
-    - Include warm-up
-    - Exercises with sets and reps
-    - Rest time
-    - Day-wise structure
-    """
-
-    return prompt
+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:
+        return "Obese"
 
 # -------------------------
-# Generate Plan
+# Submit Button
 # -------------------------
 
-if st.button("Generate Workout Plan"):
-
-    if not name:
-        st.error("Enter your name")
-    elif not equipment:
-        st.error("Select at least one equipment")
+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:
-        prompt = create_prompt(goal, equipment, fitness_level)
-
-        with st.spinner("Generating your AI workout plan..."):
-
-            inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+        bmi = calculate_bmi(weight_kg, height_cm)
+        category = bmi_category(bmi)
 
-            outputs = model.generate(
-                **inputs,
-                max_new_tokens=400,
-                temperature=0.9,
-                do_sample=True
-            )
+        st.success("✅ Calculation Successful!")
 
-            result = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        st.write(f"### 👤 Name: {name}")
+        st.write(f"### 📊 Your BMI: {bmi}")
+        st.write(f"### 🏷 BMI Category: {category}")
 
-        st.success(f"Workout Plan for {name}")
-        st.write(result)
+        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'}")