Update functions/PitchPlotFunctions.py

functions/PitchPlotFunctions.py

@@ -180,7 +180,7 @@ class PitchPlotFunctions:
         ax.axis('off')
 
     # Function to display player bio information on the given axis
-    def player_bio(self, pitcher_id: str, ax: plt.Axes, start_date: str, end_date: str, batter_hand: list):
+    def player_bio(self, pitcher_id: str, ax: plt.Axes, start_date: str, end_date: str, batter_hand: list,game_type: list = ['R']):
         """
         Fetches and displays the player's bio information on the given axis.
 
@@ -191,6 +191,16 @@ class PitchPlotFunctions:
         end_date (str): The end date for the bio information.
         batter_hand (list): The list of batter hands (e.g., ['R'] or ['L']).
         """
+        type_dict =  {'R':'Regular Season',
+                       'S':'Spring',
+                       'P':'Playoffs' }
+    
+        split_title = {
+            'all':'',
+            'right':' vs RHH',
+            'left':' vs LHH'
+        }
+        
         # Construct the URL to fetch player data
         url = f"https://statsapi.mlb.com/api/v1/people?personIds={pitcher_id}&hydrate=currentTeam"
         # Send a GET request to the URL and parse the JSON response
@@ -211,7 +221,16 @@ class PitchPlotFunctions:
             batter_hand_text = ', vs LHH'
         else:
             batter_hand_text = ''
-        ax.text(0.5, 0.40, f'{start_date} to {end_date}{batter_hand_text}', va='top', ha='center', fontsize=12, fontstyle='italic')
+
+            # Set header text 
+        if game_type[0] in ['S','P']:
+            ax.text(0.5, 0.4, f'{start_date} to {end_date} ({type_dict[game_type[0]]}){batter_hand_text}', va='bottom', ha='center',
+                    fontsize=12, fontstyle='italic')
+
+        else:
+            ax.text(0.5, 0.4, f'{start_date} to {end_date}{batter_hand_text}', va='bottom', ha='center',
+                    fontsize=12, fontstyle='italic')        
+        # ax.text(0.5, 0.40, f'{start_date} to {end_date}{batter_hand_text}', va='top', ha='center', fontsize=12, fontstyle='italic')
         # Turn off the axis
         ax.axis('off')
 
@@ -540,7 +559,7 @@ class PitchPlotFunctions:
         )
         return df
     
-    def final_plot(self, df: pl.DataFrame, pitcher_id: str, plot_picker: str, sport_id: int):
+    def final_plot(self, df: pl.DataFrame, pitcher_id: str, plot_picker: str, sport_id: int,game_type: list = ['R']):
         """
         Creates a final plot with player headshot, bio, logo, and pitch movement plots.
 
@@ -570,7 +589,7 @@ class PitchPlotFunctions:
 
         # Plot player headshot, bio, and logo
         self.player_headshot(pitcher_id=pitcher_id, ax=ax_headshot, sport_id=sport_id)
-        self.player_bio(pitcher_id=pitcher_id, ax=ax_bio, start_date=start_date, end_date=end_date, batter_hand=batter_hand)
+        self.player_bio(pitcher_id=pitcher_id, ax=ax_bio, start_date=start_date, end_date=end_date, batter_hand=batter_hand,game_type=game_type)
         self.plot_logo(pitcher_id=pitcher_id, ax=ax_logo)
 
         # Create subplot for the main plot