Update app.py

app.py

@@ -96,8 +96,7 @@ 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
-AVERAGE_CALORIES_MALE = 2500  # Average daily calorie burn for males
-AVERAGE_CALORIES_FEMALE = 2000  # Average daily calorie burn for females
+CALORIES_PER_KG_MUSCLE = 13  # Average additional calories burned per day per kg of muscle gained
 
 # Convert pounds to kilograms
 def pounds_to_kg(pounds):
@@ -108,49 +107,39 @@ 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):
+def calculate_calories_burned(mass_kg, height_diff, sets, reps, duration_hours):
     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(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
+    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
 
-# Load history
-def load_history():
-    files = [f for f in os.listdir('.') if f.startswith('data_')]
-    return files
+# 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
 
 # UI
-st.title('🏋️ Personalized Calorie Counter for Weightlifting 🏋️')
+st.title('🏋️ Advanced 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)
-    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}')
+    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')
+
 
 # Display history
 st.sidebar.header('📚 History')