src/Database/connections.py

from pymongo import MongoClient
from core.logger import logger
from core.setting import cnf

def get_mongo_client():
    try:
        uri = cnf.MONGO_URI
        dbname = cnf.MONGO_DB
    
        client = MongoClient(uri)
        db = client[dbname]
        db.command("ping")
        logger.info("MongoDB connection established successfully")
    except Exception as e:
        logger.error(e)
    return db


def get_img_gen_collection():
    try:
        db = get_mongo_client()
        if db is None:
            return None
        return db["img_gen"]

    except Exception as e:
        st.error(f"Connections File: {e}")