backend/database.py

import os
import cloudinary
import cloudinary.uploader
import cloudinary.api
from motor.motor_asyncio import AsyncIOMotorClient
from dotenv import load_dotenv
import logging


# Load environment variables from .env file
load_dotenv()

# Configure logging
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
logging.info('Environment variables loaded successfully.')


def _get_env_var(*names: str) -> str:
    """
    Returns the first configured environment variable from a list of candidates.

    Hugging Face Spaces exposes secrets as environment variables, so this helper
    accepts both the original project names and common aliases.
    """
    for name in names:
        value = os.getenv(name)
        if value:
            return value
    raise RuntimeError(f"Missing required environment variable. Tried: {', '.join(names)}")

# MongoDB configuration
MONGODB_URI = _get_env_var('MONGODB_URI', 'MONGO_URI')
DB_NAME = _get_env_var('DATABASE_NAME', 'DB_NAME')

# Cloudinary Configuration
cloudinary.config(
    cloud_name=_get_env_var('CLOUDINARY_CLOUD_NAME'),
    api_key=_get_env_var('CLOUDINARY_API_KEY'),
    api_secret=_get_env_var('CLOUDINARY_API_SECRET'),
    secure=True
)

def upload_image_to_cloud(file_byte: bytes, folder_name: str='xrays') -> str:
    """
    Uploads an image to Cloudinary and returns the URL of the uploaded image.

    Args:
        file_byte (bytes): The byte content of the image to be uploaded.
        folder_name (str): The folder in Cloudinary where the image will be stored.

    Returns:
        str: The URL of the uploaded image.
        
    Raises:
        Exception: If there is an error during the upload process.
        
    """
    try:
        logging.info('Uploading image to Cloudinary...')
        response = cloudinary.uploader.upload(file_byte,
                                        folder=f'precision_diagnostics/{folder_name}')
        logging.info('Image uploaded successfully.')
        return response.get('secure_url')
    except Exception as e:
        logging.error(f'Error uploading image: {e}')
        raise e
    
    
    
# Initialize MongoDB client
try:
    client = AsyncIOMotorClient(MONGODB_URI)
    db = client[DB_NAME]
    logging.info('Connected to MongoDB successfully.')
except Exception as e:
    logging.error(f'Error connecting to MongoDB: {e}')
    raise e

diagnostic_records = db['records']