Update functions/PitchPlotFunctions.py

functions/PitchPlotFunctions.py

@@ -158,25 +158,28 @@ class PitchPlotFunctions:
         sport_id (int): The sport ID to determine the URL format.
         """
         # Construct the URL for the player's headshot image
-        if sport_id == 1:
-            url = f'https://img.mlbstatic.com/mlb-photos/image/'\
-                  f'upload/d_people:generic:headshot:67:current.png'\
-                  f'/w_640,q_auto:best/v1/people/{pitcher_id}/headshot/silo/current.png'
-        else:
-            url = f'https://img.mlbstatic.com/mlb-photos/image/upload/c_fill,g_auto/w_640/v1/people/{pitcher_id}/headshot/milb/current.png'
-        
-        # Send a GET request to the URL
-        response = requests.get(url)
-        # Open the image from the response content
-        img = Image.open(BytesIO(response.content))
-        # Display the image on the axis
-        ax.set_xlim(0, 1)
-        ax.set_ylim(0, 1)
-        if sport_id == 1:
-            ax.imshow(img, extent=[0, 1, 0, 1], origin='upper')
-        else:
-            ax.imshow(img, extent=[1/6, 5/6, 0, 1], origin='upper')
+        try:
+            if sport_id == 1:
+                url = f'https://img.mlbstatic.com/mlb-photos/image/'\
+                      f'upload/d_people:generic:headshot:67:current.png'\
+                      f'/w_640,q_auto:best/v1/people/{pitcher_id}/headshot/silo/current.png'
+            else:
+                url = f'https://img.mlbstatic.com/mlb-photos/image/upload/c_fill,g_auto/w_640/v1/people/{pitcher_id}/headshot/milb/current.png'
+            
+            # Send a GET request to the URL
+            response = requests.get(url)
+            # Open the image from the response content
+            img = Image.open(BytesIO(response.content))
+            # Display the image on the axis
+            ax.set_xlim(0, 1)
+            ax.set_ylim(0, 1)
+            if sport_id == 1:
+                ax.imshow(img, extent=[0, 1, 0, 1], origin='upper')
+            else:
+                ax.imshow(img, extent=[1/6, 5/6, 0, 1], origin='upper')
         # Turn off the axis
+        except PIL.UnidentifiedImageError:
+            print('NA')
         ax.axis('off')
 
     # Function to display player bio information on the given axis