r2_uploader.py

import boto3, os
from uuid import uuid4

def upload_image_to_r2(image_bytes, folder_name="search_arb", app_type="template"):
    s3 = boto3.client(
        "s3",
        endpoint_url=os.getenv("R2_ENDPOINT"),
        aws_access_key_id=os.getenv("R2_ACCESS_KEY"),
        aws_secret_access_key=os.getenv("R2_SECRET_KEY")
    )
    
    filename = f"{uuid4().hex}.png"
    
    if folder_name:
        file_key = f"hug_face/{app_type.strip('/')}/{folder_name.strip('/')}/{filename}"
    else:
        file_key = f"hug_face/{app_type.strip('/')}/{filename}"
    
    s3.put_object(
        Bucket=os.getenv("R2_BUCKET_NAME"),
        Key=file_key,
        Body=image_bytes,
        ContentType="image/png"
    )
    
    return f"{os.getenv('R2_ENDPOINT')}/{os.getenv('R2_BUCKET_NAME')}/{file_key}"