instituting database.py

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
+    
+db = init_conn()

src/streamlit_app.py

@@ -1,25 +1,11 @@
 import streamlit as st
 import numpy as np
 import pandas as pd
-import pymongo
-import re
-import os
+from database import db
 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()
-
 game_format = {'Win Percentage': '{:.2%}','First Inning Lead Percentage': '{:.2%}',
               'Fifth Inning Lead Percentage': '{:.2%}', '8+ runs': '{:.2%}', 'DK LevX': '{:.2%}', 'FD LevX': '{:.2%}'}