Make team filter

src/streamlit_app.py

@@ -1,7 +1,6 @@
 import os
 import streamlit as st
 import pandas as pd
-from streamlit_filter import filter_dataframe
 
 
 LEAGUE_NAME = "LOFG"
@@ -14,13 +13,14 @@ def load_data():
     # Hack to get position, replace with better position from yahoo api in future
     data["position"] = data["eligible_positions"].apply(lambda x: eval(x)[0])
     data.columns = data.columns.str.lower()
-    return data
+    teams_list = sorted(list(data["team_name"].unique()))
+    return data, teams_list
 
 
 def get_app():
     st.title(f"{LEAGUE_NAME} Keeper Options")
     data_load_state = st.text("Loading data...")
-    data = load_data()
+    data, teams_list = load_data()
     data_load_state.text("Data loaded")
 
     raw_display_columns = [
@@ -33,7 +33,10 @@ def get_app():
         "eligible",
     ]
 
-    filtered_data = filter_dataframe(data[raw_display_columns])
+    teams_selected = st.multiselect("Team:", teams_list)
+    teams_filter = data["team_name"].isin(teams_selected) if teams_selected else data["team_name"].isin(teams_list)
+
+    filtered_data = data.loc[teams_filter, raw_display_columns]
     st.write(filtered_data)