Update app.py

app.py

@@ -43,6 +43,9 @@ st.markdown('<div class="main-container">', unsafe_allow_html=True)
 markdown_text = """
 ## Catch Probability Lookup Tool
 
+##### By: Thomas Nestico ([@TJStats](https://x.com/TJStats))
+##### Data: [MLB](https://baseballsavant.mlb.com/)
+
 This Streamlit app retrieves catch probability data for a selected fielder from [Baseball Savant](https://baseballsavant.mlb.com/leaderboard/catch_probability).
 The app displays the fielder's data in a table and allows the user to select a 
 row to view the corresponding catch video.
@@ -123,48 +126,55 @@ df_merge['catch_rate'] = df_merge['catch_rate'].astype(float).apply(lambda x: f"
 df_merge['pos'] = df_merge['pos'].astype(int)
 df_merge['Position'] = df_merge['pos'].map(pos_dict)
 
-column_names = ['game_date','batter_name', 'pitcher_name', 'name_display_first_last', 'Position','event', 'out', 'wall', 'back', 'stars', 'distance', 'hang_time', 'catch_rate']
+df_merge = df_merge[df_merge['batter_id'] != df_merge['player_id']]
+
+
+if df_merge.empty:
+    st.write("No data available for the selected fielder.")
+    st.stop()
+else: 
+    column_names = ['game_date','batter_name', 'pitcher_name', 'name_display_first_last', 'Position','event', 'out', 'wall', 'back', 'stars', 'distance', 'hang_time', 'catch_rate']
 
-column_names_display = ['Game Date','Batter Name', 'Pitcher Name', 'Fielder Name', 'Position','Event', 'Out', 'Wall', 'Back', 'Stars', 'Distance', 'Hang Time', 'Catch Rate']
+    column_names_display = ['Game Date','Batter Name', 'Pitcher Name', 'Fielder Name', 'Position','Event', 'Out', 'Wall', 'Back', 'Stars', 'Distance', 'Hang Time', 'Catch Rate']
 
 
 
 
 
-# Use a container to control the width of the AgGrid display
-with st.container():
-    st.write("### Fielder Data")
-    # Configure the AgGrid options
-    gb = GridOptionsBuilder.from_dataframe(df_merge[column_names])
-    # Set display names for columns
-    for col, display_name in zip(column_names, column_names_display):
-        gb.configure_column(col, headerName=display_name)
+    # Use a container to control the width of the AgGrid display
+    with st.container():
+        st.write("### Fielder Data")
+        # Configure the AgGrid options
+        gb = GridOptionsBuilder.from_dataframe(df_merge[column_names])
+        # Set display names for columns
+        for col, display_name in zip(column_names, column_names_display):
+            gb.configure_column(col, headerName=display_name)
 
 
-    gb.configure_selection('single', use_checkbox=True)
-    grid_options = gb.build()
+        gb.configure_selection('single', use_checkbox=True)
+        grid_options = gb.build()
 
-    # Display the dataframe using AgGrid
-    grid_response = AgGrid(
-        df_merge[column_names],
-        gridOptions=grid_options,
-        update_mode=GridUpdateMode.SELECTION_CHANGED,
-        height=300,
-        allow_unsafe_jscode=True,
-    )
+        # Display the dataframe using AgGrid
+        grid_response = AgGrid(
+            df_merge[column_names],
+            gridOptions=grid_options,
+            update_mode=GridUpdateMode.SELECTION_CHANGED,
+            height=300,
+            allow_unsafe_jscode=True,
+        )
 
-# Get the selected row index
+    # Get the selected row index
 
 
-try:
-    # Update the video URL based on the selected row
-    selected_row_index = int(grid_response['selected_rows'].index.values[0])
-    a = requests.get(f'https://baseballsavant.mlb.com/sporty-videos?playId={df_merge["play_id"].values[selected_row_index]}')
-    soup = BeautifulSoup(a.content, 'lxml')
-    video_url = str(soup).split('<source src="')[1].split('" ')[0]
-    # Share the video through Streamlit
-    st.video(video_url)
-except AttributeError:
-    st.write("Select Row to Display Video")
+    try:
+        # Update the video URL based on the selected row
+        selected_row_index = int(grid_response['selected_rows'].index.values[0])
+        a = requests.get(f'https://baseballsavant.mlb.com/sporty-videos?playId={df_merge["play_id"].values[selected_row_index]}')
+        soup = BeautifulSoup(a.content, 'lxml')
+        video_url = str(soup).split('<source src="')[1].split('" ')[0]
+        # Share the video through Streamlit
+        st.video(video_url)
+    except AttributeError:
+        st.write("Select Row to Display Video")
 
 st.markdown('</div>', unsafe_allow_html=True)