reports/database.py

from motor.motor_asyncio import AsyncIOMotorClient
from dotenv import load_dotenv
import os

load_dotenv()

MONGO_URL = os.getenv("MONGO_URL", "mongodb+srv://localhost:27017")  # Fallback to local MongoDB
MONGO_DB = os.getenv("MONGO_DB", "web-maint-ai")

client = AsyncIOMotorClient(MONGO_URL)
db = client[MONGO_DB]

def get_db():
    return db