import os
from motor.motor_asyncio import AsyncIOMotorClient

# MongoDB URI (must be set in Hugging Face Secrets)
MONGO_URI = os.environ.get("MONGODB_URI")

if not MONGO_URI:
    raise ValueError("MONGODB_URI environment variable is not set.")

# Initialize MongoDB client
client = AsyncIOMotorClient(MONGO_URI)

# Use the correct database name explicitly
db = client["cps_db"]

# Collections
patients_collection = db["patients"]
results_collection = db["patient_analysis_results"]

# Optional utility function to reuse the client
def get_mongo_client():
    return client