Update app.py

app.py

@@ -91,6 +91,39 @@ pitch_colours = {
     'UN': {'colour': '#9C8975', 'name': 'Unknown'},
 }
 
+
+# Dictionary to map pitch types to their corresponding colors and names
+pitch_colours_tp = {
+    ## Fastballs ##
+    'FF': {'colour': '#FF007D50', 'name': '4-Seam Fastball'},
+    'FA': {'colour': '#FF007D50', 'name': 'Fastball'},
+    'SI': {'colour': '#98165D50', 'name': 'Sinker'},
+    'FC': {'colour': '#BE5FA050', 'name': 'Cutter'},
+
+    ## Offspeed ##
+    'CH': {'colour': '#F79E7050', 'name': 'Changeup'},
+    'FS': {'colour': '#FE610050', 'name': 'Splitter'},
+    'SC': {'colour': '#F0822350', 'name': 'Screwball'},
+    'FO': {'colour': '#FFB00050', 'name': 'Forkball'},
+
+    ## Sliders ##
+    'SL': {'colour': '#67E18D50', 'name': 'Slider'},
+    'ST': {'colour': '#1BB99950', 'name': 'Sweeper'},
+    'SV': {'colour': '#37674850', 'name': 'Slurve'},
+
+    ## Curveballs ##
+    'KC': {'colour': '#311D8B50', 'name': 'Knuckle Curve'},
+    'CU': {'colour': '#3025CE50', 'name': 'Curveball'},
+    'CS': {'colour': '#274BFC50', 'name': 'Slow Curve'},
+    'EP': {'colour': '#648FFF50', 'name': 'Eephus'},
+
+    ## Others ##
+    'KN': {'colour': '#867A0850', 'name': 'Knuckleball'},
+    'KN': {'colour': '#867A0850', 'name': 'Knuckle Ball'},
+    'PO': {'colour': '#472C3050', 'name': 'Pitch Out'},
+    'UN': {'colour': '#9C897550', 'name': 'Unknown'},
+}
+
 # Create dictionaries for pitch types and their attributes
 dict_colour = {key: value['colour'] for key, value in pitch_colours.items()}
 dict_pitch = {key: value['name'] for key, value in pitch_colours.items()}
@@ -103,6 +136,10 @@ dict_pitch_name.update({'4-Seam':'#FF007D'})
 dict_pitch.update({'FF':'Four-Seam Fastball'})
 
 
+dict_pitch_name_tp = {value['name']: value['colour'] for key, value in pitch_colours_tp.items()}
+dict_pitch_name_tp.update({'Four-Seam Fastball':'#FF007D50'})
+dict_pitch_name_tp.update({'4-Seam':'#FF007D50'})
+
 # Sort dict_pitch alphabetically by pitch name
 dict_pitch_alpha = dict(sorted(dict_pitch.items(), key=lambda item: item[1]))
 
@@ -692,7 +729,7 @@ def server(input, output, session):
                 .set_table_styles([{'selector': 'thead th:nth-child(9)', 'props': [('min-width', '40px')]}], overwrite=False)
                 .background_gradient(cmap=cmap_sum,subset = (brushed_df_final.columns[-2]),vmin=80,vmax=120)
                 .background_gradient(cmap=cmap_sum,subset = (brushed_df_final.columns[-1]),vmin=20,vmax=80)
-                .applymap(lambda x: f'background-color: {dict_pitch_name.get(x, "")}', subset=['Pitch Type'])
+                .applymap(lambda x: f'background-color: {dict_pitch_name_tp.get(x, "")}', subset=['Pitch Type'])
                 .applymap(lambda x: f'background-color: black' if x == 0 else '', subset=['Spin'])