Update app.py

app.py

@@ -87,7 +87,6 @@ st.write(f"Factor for chosen grip: {grip_styles[grip_style]}")
 
 
 # BMR Calculator
-
 import streamlit as st
 import pandas as pd
 import os
@@ -95,20 +94,34 @@ from datetime import datetime
 
 # Constants
 GRAVITATIONAL_FORCE = 9.8  # Earth's gravitational force in m/s^2
+POUND_TO_KG = 0.453592  # Conversion factor from pounds to kilograms
+INCH_TO_METER = 0.0254  # Conversion factor from inches to meters
+AVERAGE_CALORIES_MALE = 2500  # Average daily calorie burn for males
+AVERAGE_CALORIES_FEMALE = 2000  # Average daily calorie burn for females
+
+# Convert pounds to kilograms
+def pounds_to_kg(pounds):
+    return pounds * POUND_TO_KG
+
+# Convert feet and inches to meters
+def feet_inches_to_meters(feet, inches):
+    total_inches = feet * 12 + inches
+    return total_inches * INCH_TO_METER
 
 # Calculate Basal Metabolic Rate
-def calculate_bmr(lbm):
-    return 500 + 22 * lbm
+def calculate_bmr(gender):
+    return AVERAGE_CALORIES_MALE if gender == 'Male' else AVERAGE_CALORIES_FEMALE
 
 # Calculate Calories Burned Lifting Weights
-def calculate_calories_burned(mass, height_diff):
-    joules = mass * GRAVITATIONAL_FORCE * height_diff
+def calculate_calories_burned(mass_kg, height_diff, sets, reps):
+    total_mass_lifted = mass_kg * sets * reps
+    joules = total_mass_lifted * GRAVITATIONAL_FORCE * height_diff
     return joules / 4184  # Convert joules to kilocalories
 
 # Save to file
-def save_data(lbm, calories):
+def save_data(gender, height, weight, calories):
     filename = f"data_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv"
-    df = pd.DataFrame({'Date': [datetime.now()], 'LBM': [lbm], 'Calories Burned': [calories]})
+    df = pd.DataFrame({'Date': [datetime.now()], 'Gender': [gender], 'Height': [height], 'Weight Lifted (lbs)': [weight], 'Calories Burned': [calories]})
     df.to_csv(filename, index=False)
     return filename
 
@@ -118,20 +131,25 @@ def load_history():
     return files
 
 # UI
-st.title('🏋️ Calorie Counter for Weightlifting 🏋️')
+st.title('🏋️ Personalized Calorie Counter for Weightlifting 🏋️')
 
 with st.form('input_form'):
-    lbm = st.number_input('Enter your Lean Body Mass (kg):', min_value=0.0, value=60.0)
-    mass = st.number_input('Enter the weight lifted (kg):', min_value=0.0, value=30.0)
-    height_diff = st.number_input('Enter the height difference (m):', min_value=0.0, value=1.0)
+    gender = st.radio('Select your Gender:', ('Male', 'Female'))
+    feet = st.number_input('Height (feet):', min_value=0, value=5)
+    inches = st.number_input('Height (inches):', min_value=0, value=11)
+    weight = st.number_input('Enter the weight lifted (lbs):', min_value=0.0, value=30.0)
+    sets = st.number_input('Number of sets:', min_value=1, value=10)
+    reps = st.number_input('Repetitions per set:', min_value=1, value=10)
     submitted = st.form_submit_button('Calculate')
 
 if submitted:
-    bmr = calculate_bmr(lbm)
-    calories_burned = calculate_calories_burned(mass, height_diff)
-    save_file = save_data(lbm, calories_burned)
-    st.success(f'🔥 BMR: {bmr:.2f} kcal/day')
-    st.success(f'🔥 Calories Burned: {calories_burned:.4f} kcal')
+    height_meters = feet_inches_to_meters(feet, inches)
+    weight_kg = pounds_to_kg(weight)
+    bmr = calculate_bmr(gender)
+    calories_burned = calculate_calories_burned(weight_kg, height_meters, sets, reps)
+    save_file = save_data(gender, f'{feet}\'{inches}"', weight, calories_burned)
+    st.success(f'🔥 Estimated Daily Calorie Burn (BMR): {bmr} kcal/day')
+    st.success(f'🔥 Calories Burned in Session: {calories_burned:.4f} kcal')
     st.sidebar.success(f'📁 Saved as {save_file}')
 
 # Display history
@@ -141,5 +159,3 @@ for file in load_history():
         df = pd.read_csv(file)
         st.sidebar.write(df)
 
-
-