src/database.py

import streamlit as st
import pymongo
import os

@st.cache_resource
def init_conn():
        uri = os.getenv('mongo_uri')
        client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
        db = client["MLB_Database"]

        return db
    
db = init_conn()