Jon Solow commited on
Commit
b7f4f75
·
1 Parent(s): 86f5e84

Move app into function for better testing

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +21 -8
src/streamlit_app.py CHANGED
@@ -7,20 +7,33 @@ LEAGUE_NAME = "LOFG"
7
  DATA_URL = "../tests/mocks/2023_keepers.csv"
8
 
9
 
10
- st.title(f"{LEAGUE_NAME} Keeper Options")
11
-
12
-
13
  @st.cache_data
14
  def load_data():
15
  data = pd.read_csv(os.path.join(os.path.dirname(__file__), DATA_URL), index_col=0)
 
16
  data.columns = data.columns.str.lower()
17
  return data
18
 
19
 
20
- data_load_state = st.text("Loading data...")
21
- data = load_data()
22
- data_load_state.text("Data loaded")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
 
24
 
25
- st.subheader("Raw data")
26
- st.write(data)
 
7
  DATA_URL = "../tests/mocks/2023_keepers.csv"
8
 
9
 
 
 
 
10
  @st.cache_data
11
  def load_data():
12
  data = pd.read_csv(os.path.join(os.path.dirname(__file__), DATA_URL), index_col=0)
13
+ # data["position"] = data["eligible_positions"].fillna('[""]').apply(lambda x: json.loads(x))
14
  data.columns = data.columns.str.lower()
15
  return data
16
 
17
 
18
+ def get_app():
19
+ st.title(f"{LEAGUE_NAME} Keeper Options")
20
+ data_load_state = st.text("Loading data...")
21
+ data = load_data()
22
+ data_load_state.text("Data loaded")
23
+
24
+ raw_display_columns = [
25
+ "player_id",
26
+ "name",
27
+ "position",
28
+ "team_name",
29
+ "years_eligible",
30
+ "keeper_cost",
31
+ "eligible",
32
+ ]
33
+
34
+ st.subheader("Raw data")
35
+ st.write(data[raw_display_columns])
36
 
37
 
38
+ if __name__ == "__main__":
39
+ get_app()