Update app.py

app.py

@@ -96,7 +96,8 @@ from datetime import datetime
 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
-CALORIES_PER_KG_MUSCLE = 13  # Average additional calories burned per day per kg of muscle gained
+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):
@@ -107,45 +108,50 @@ 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(gender):
+    return AVERAGE_CALORIES_MALE if gender == 'Male' else AVERAGE_CALORIES_FEMALE
+
 # Calculate Calories Burned Lifting Weights
-def calculate_calories_burned(mass_kg, height_diff, sets, reps, duration_hours):
+def calculate_calories_burned(mass_kg, height_diff, sets, reps):
     total_mass_lifted = mass_kg * sets * reps
-    joules_per_rep = total_mass_lifted * GRAVITATIONAL_FORCE * height_diff
-    joules_total = joules_per_rep * duration_hours * 3600 / (sets * reps)  # Assuming continuous lifting for the duration
-    return joules_total / 4184  # Convert joules to kilocalories
+    joules = total_mass_lifted * GRAVITATIONAL_FORCE * height_diff
+    return joules / 4184  # Convert joules to kilocalories
+
+# Save to file
+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()], 'Gender': [gender], 'Height': [height], 'Weight Lifted (lbs)': [weight], 'Calories Burned': [calories]})
+    df.to_csv(filename, index=False)
+    return filename
 
-# Calculate Additional Daily Calorie Burn from Muscle Gain
-def estimate_additional_calorie_burn(weight_kg, duration_months):
-    muscle_gain = weight_kg * duration_months * 0.025  # Estimate muscle gain: 2.5% of lifted weight per month
-    return muscle_gain * CALORIES_PER_KG_MUSCLE
+# Load history
+def load_history():
+    files = [f for f in os.listdir('.') if f.startswith('data_')]
+    return files
 
 # UI
-st.title('🏋️ Advanced Calorie Counter for Weightlifting 🏋️')
+st.title('🏋️ Personalized Calorie Counter for Weightlifting 🏋️')
 
 with st.form('input_form'):
+    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)
-    duration_hours = st.number_input('Duration of session (hours):', min_value=0.1, value=1.0)
-    duration_months = st.number_input('Duration of regular training (months):', min_value=1, value=1)
     submitted = st.form_submit_button('Calculate')
 
 if submitted:
     height_meters = feet_inches_to_meters(feet, inches)
     weight_kg = pounds_to_kg(weight)
-    calories_burned = calculate_calories_burned(weight_kg, height_meters, sets, reps, duration_hours)
-    additional_calories = estimate_additional_calorie_burn(weight_kg, duration_months)
-    st.success(f'🔥 Calories Burned in Session: {calories_burned:.2f} kcal')
-    st.success(f'📈 Additional Daily Calorie Burn from Muscle Gain: {additional_calories:.2f} kcal/day')
+    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}')
 
-
-# Load history
-def load_history():
-    files = [f for f in os.listdir('.') if f.startswith('data_')]
-    return files
-    
 # Display history
 st.sidebar.header('📚 History')
 for file in load_history():