instituting database.py

src/database.py

@@ -0,0 +1,16 @@
+import streamlit as st
+import pymongo
+import os
+
+@st.cache_resource
+def init_conn():
+         # Try to get from environment variable first, fall back to secrets
+        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,28 +1,14 @@
 import streamlit as st
 import numpy as np
 import pandas as pd
-import pymongo
 import re
-import os
 import math
+from database import db
 
 st.set_page_config(layout="wide")
 
 print(f"Streamlit version: {st.__version__}")
 
-@st.cache_resource
-def init_conn():
-         # Try to get from environment variable first, fall back to secrets
-        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%}'}