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,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
 
 st.set_page_config(layout="wide")
 
-@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()
-
 wrong_acro = ['WSH', 'AZ']
 right_acro = ['WAS', 'ARI']