File size: 2,426 Bytes
f2f04e0 3bda825 f2f04e0 bc1f6f8 f2f04e0 bc1f6f8 f2f04e0 bc1f6f8 f2f04e0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | 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'] |