mirror of https://github.com/kortix-ai/suna.git
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
|
"""
|
||
|
|
Utility functions for handling image operations.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import base64
|
||
|
|
import uuid
|
||
|
|
from datetime import datetime
|
||
|
|
from utils.logger import logger
|
||
|
|
from services.supabase import DBConnection
|
||
|
|
|
||
|
|
async def upload_base64_image(base64_data: str, bucket_name: str = "browser-screenshots") -> str:
|
||
|
|
"""Upload a base64 encoded image to Supabase storage and return the URL.
|
||
|
|
|
||
|
|
Args:
|
||
|
|
base64_data (str): Base64 encoded image data (with or without data URL prefix)
|
||
|
|
bucket_name (str): Name of the storage bucket to upload to
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
str: Public URL of the uploaded image
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
# Remove data URL prefix if present
|
||
|
|
if base64_data.startswith('data:'):
|
||
|
|
base64_data = base64_data.split(',')[1]
|
||
|
|
|
||
|
|
# Decode base64 data
|
||
|
|
image_data = base64.b64decode(base64_data)
|
||
|
|
|
||
|
|
# Generate unique filename
|
||
|
|
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
|
||
|
|
unique_id = str(uuid.uuid4())[:8]
|
||
|
|
filename = f"image_{timestamp}_{unique_id}.png"
|
||
|
|
|
||
|
|
# Upload to Supabase storage
|
||
|
|
db = DBConnection()
|
||
|
|
client = await db.client
|
||
|
|
storage_response = await client.storage.from_(bucket_name).upload(
|
||
|
|
filename,
|
||
|
|
image_data,
|
||
|
|
{"content-type": "image/png"}
|
||
|
|
)
|
||
|
|
|
||
|
|
# Get public URL
|
||
|
|
public_url = await client.storage.from_(bucket_name).get_public_url(filename)
|
||
|
|
|
||
|
|
logger.debug(f"Successfully uploaded image to {public_url}")
|
||
|
|
return public_url
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Error uploading base64 image: {e}")
|
||
|
|
raise RuntimeError(f"Failed to upload image: {str(e)}")
|