Add database connection module and refactor Streamlit app to utilize it, improving code organization and data handling for simulations.

src/database.py

@@ -0,0 +1,15 @@
+import streamlit as st
+import pymongo
+import os
+
+@st.cache_resource
+def init_conn():
+    uri = os.getenv('MONGO_URI')
+    if not uri:
+        uri = st.secrets['mongo_uri']
+    client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
+    db = client["NFL_Database"]
+    return db
+
+# Initialize the database connection
+db = init_conn()

src/sim_func_hold/regular_functions.py

@@ -2,8 +2,7 @@ import streamlit as st
 import pandas as pd
 import numpy as np
 from pymongo import MongoClient
-
-from streamlit_app import db
+from database import db
 
 @st.cache_data(ttl = 600)
 def init_DK_seed_frames(slate_var, sharp_split):

src/sim_func_hold/showdown_functions.py

@@ -2,9 +2,10 @@ import streamlit as st
 import pandas as pd
 import numpy as np
 from pymongo import MongoClient
+from database import db
 
 @st.cache_data(ttl = 599)
-def init_DK_SD_seed_frames(slate, split, db):
+def init_DK_SD_seed_frames(slate, split):
     if slate == 'Main Slate':
         collection = db[f"DK_NFL_SD_seed_frame"] 
     elif slate == 'Secondary Slate':
@@ -21,7 +22,7 @@ def init_DK_SD_seed_frames(slate, split, db):
     return DK_seed
 
 @st.cache_data(ttl = 599)
-def init_FD_SD_seed_frames(slate, split, db):
+def init_FD_SD_seed_frames(slate, split):
     if slate == 'Main Slate':
         collection = db[f"FD_NFL_SD_seed_frame"] 
     elif slate == 'Secondary Slate':
@@ -38,7 +39,7 @@ def init_FD_SD_seed_frames(slate, split, db):
     return FD_seed
 
 @st.cache_data(ttl = 599)
-def init_SD_baselines(slate_var, db):
+def init_SD_baselines(slate_var):
     collection = db['DK_SD_NFL_ROO'] 
     cursor = collection.find()
 

src/streamlit_app.py

@@ -9,18 +9,7 @@ from itertools import combinations
 
 st.set_page_config(layout="wide")
 
-@st.cache_resource
-def init_conn():
-        uri = os.getenv('MONGO_URI')
-        if not uri:
-            uri = st.secrets['mongo_uri']
-        client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
-        db = client["NFL_Database"]
-
-        return db
-    
-db = init_conn()
-
+from database import db
 from sim_func_hold.regular_functions import *
 from sim_func_hold.showdown_functions import *