Update app.py

app.py

@@ -75,13 +75,13 @@ def init_baselines():
     return hitter_rhp, hitter_lhp, pitcher_rhh, pitcher_lhh
 
 @st.cache_resource
-def calc_poisson(hitter_val, sp_val):
+def calc_poisson(hitter_val, sp_val, sp_count, bp_count):
     base_val = hitter_val
     opp_val = sp_val
     sp_combo_val = sum([base_val, opp_val]) / 2
     bp_combo_val = sum([base_val, .085]) / 2
-    sp_instances = 1
-    bp_instances = 0
+    sp_instances = sp_count
+    bp_instances = bp_count
     sp_mean = sp_combo_val * sp_instances
     bp_mean = bp_combo_val * bp_instances
     
@@ -118,10 +118,14 @@ with col1:
         hitter_var1 = st.selectbox("What hitter are you looking at?", options = hitter_lhp['Player'].unique())
         working_hitters = hitter_lhp.copy()
         hitter_check = working_hitters[working_hitters['Player'] == hitter_var1]
+    sp_count = st.number_input("How many PA against the Pitcher?")
+    bp_count = st.number_input("How many PA against the Bullpen?")
 
 with col2:
-    st.write(pitcher_hand)
-    hitter_val = hitter_check['BB%'].iloc[0]
-    sp_val = pitcher_check['BB%'].iloc[0]
-    value = calc_poisson(hitter_val, sp_val)
-    st.write(value)
+    if st.button('calculate theoretical means'):
+        hitter_val = hitter_check['BB%'].iloc[0]
+        sp_val = pitcher_check['BB%'].iloc[0]
+        value = calc_poisson(hitter_val, sp_val, sp_count, bp_count)
+        st.write(f"Theoretical mean of the SP instances: {value[0]}")
+        st.write(f"Theoretical mean of the BP instances: {value[1]}")
+        st.write(f"Sample mean from generated data: {value[2]}")