import os

import streamlit as st
from dotenv import load_dotenv
from pymongo import MongoClient


@st.cache_resource
def init_connection():
    load_dotenv()
    uri = f"mongodb+srv://{os.getenv('MONGO_USERNAME')}:{os.getenv('MONGO_PASSWORD')}@{os.getenv('MONGO_HOST')}/?retryWrites=true&w=majority&appName=Cluster0"
    client = MongoClient(uri)
    return client.event_data