Update functions/PitchPlotFunctions.py

functions/PitchPlotFunctions.py

@@ -562,7 +562,7 @@ class PitchPlotFunctions:
         self.plot_logo(pitcher_id=pitcher_id, ax=ax_logo)
 
         # Create subplot for the main plot
-        ax_main_plot = fig.add_subplot(gs[2, :])
+        ax_main_plot = fig.add_subplot(gs[2, 1:-1])
 
         # Create subplot for the legend
         ax_legend = fig.add_subplot(gs[3, :])
@@ -571,13 +571,13 @@ class PitchPlotFunctions:
         # Create subplot for the footer
         ax_footer = fig.add_subplot(gs[-2, :])
 
-        # Plot the selected pitch movement plot
-        # if plot_picker == 'short_form_movement':
-        #     self.break_plot_big(df, ax_main_plot, sport_id=sport_id)
-        # elif plot_picker == 'long_form_movement':
-        #     self.break_plot_big_long(df, ax_main_plot, sport_id=sport_id)
-        # elif plot_picker == 'release_point':
-        #     self.release_point_plot(df, ax_main_plot, sport_id=sport_id)
+        Plot the selected pitch movement plot
+        if plot_picker == 'short_form_movement':
+            self.break_plot_big(df, ax_main_plot, sport_id=sport_id)
+        elif plot_picker == 'long_form_movement':
+            self.break_plot_big_long(df, ax_main_plot, sport_id=sport_id)
+        elif plot_picker == 'release_point':
+            self.release_point_plot(df, ax_main_plot, sport_id=sport_id)
 
         # Sort the DataFrame and get unique pitch types
         items_in_order = list(df.sort(by=['prop', 'pitch_type'], descending=[True, True])['pitch_type'].unique(maintain_order=True))