Update app.py

app.py

@@ -752,7 +752,11 @@ def server(input, output, session):
         df = df.with_columns([pl.col(col).cast(pl.Int32) for col in columns_to_cast])
         df_2 = df_2.with_columns([pl.col(col).cast(pl.Int32) for col in columns_to_cast])
 
-        df_legend = pl.concat([df, df_2], rechunk=True) 
+        common_cols = list(set(df.columns) & set(df_2.columns))
+
+        
+        
+        df_legend = pl.concat([df.select(common_cols), df_2.select(common_cols)], rechunk=True) 
             # # Create legend for pitch types
         items_in_order = (df_legend.sort("pitch_count", descending=True)['pitch_type'].unique(maintain_order=True).to_numpy())
         colour_pitches = [dict_colour[x] for x in items_in_order]