Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 
-# Revised Function to calculate calories burned during swimming based on swim style
-def calories_swim(time_hr, weight_kg, style_mets):
-    return (time_hr * style_mets * 3.5 * weight_kg) / 200
+# Function to calculate calories burned during swimming based on swim style, now in pounds
+def calories_swim(time_hr, weight_lb, style_mets):
+    return (time_hr * style_mets * 3.5 * weight_lb) / 200
 
 # Function to calculate calories burned during pull-ups based on grip style
 def calories_pullup(reps, weight, grip_style_factor):
@@ -26,15 +26,15 @@ exercise_type = st.sidebar.selectbox(
     ["Swim Jim 🏊‍♂️", "Ring King 👑", "Both Boost 🚀"]
 )
 
-# Revised Swim Styles with METs
+# Revised Swim Styles with METs to meet your requirement
 swim_styles = {
-    "Treading Water 🌊": 6.0,
-    "Backstroke 🏊‍♂️": 9.0,
-    "Breaststroke 🐸": 10.0,
-    "Freestyle Light 🦋": 11.0,
-    "Freestyle Vigorous 🚀": 13.0,
-    "Butterfly 🦋": 12.0,
-    "Dog Paddle 🐶": 7.0
+    "Treading Water 🌊": 6,
+    "Backstroke 🏊‍♂️": 9,
+    "Breaststroke 🐸": 10,
+    "Freestyle Light 🦋": 11,
+    "Freestyle Vigorous 🚀": 14.3,
+    "Butterfly 🦋": 14.3,
+    "Dog Paddle 🐶": 7
 }
 
 # Grip Styles with factors
@@ -57,8 +57,7 @@ grip_style = st.sidebar.selectbox(
 )
 
 # Calculation
-weight_kg = weight * 0.453592
-calories_from_swimming = calories_swim(time_swim, weight_kg, swim_styles[swim_style])
+calories_from_swimming = calories_swim(time_swim, weight, swim_styles[swim_style])
 calories_from_pullups = calories_pullup(reps, weight, grip_styles[grip_style])
 
 # Display Results
@@ -84,4 +83,3 @@ st.write(f"METS for chosen style: {swim_styles[swim_style]}")
 
 st.subheader(f"Ring Style: {grip_style} 🪐")
 st.write(f"Factor for chosen grip: {grip_styles[grip_style]}")
-