src/persistence/db.py

import streamlit as st
from pymongo import MongoClient
from dotenv import load_dotenv, dotenv_values
import os


def init_db():
    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

class CollectionNames:
   EVENT_URLS = "event_urls"
   UNSORTED_URLS = "unsorted_urls"