ejfox's picture
Duplicate from huggingface-projects/stable-diffusion-multiplayer
bb88c4d
import { PUBLIC_API_BASE } from '$env/static/public';
import { GRID_SIZE, FRAME_SIZE } from '$lib/constants';
export function base64ToBlob(base64image: string): Promise<Blob> {
return new Promise((resolve) => {
const img = new Image();
img.onload = async () => {
const w = img.width;
const h = img.height;
const canvas = document.createElement('canvas');
canvas.width = w;
canvas.height = h;
const ctx = canvas.getContext('2d') as CanvasRenderingContext2D;
ctx.drawImage(img, 0, 0, w, h);
const imgBlob: Blob = await new Promise((r) =>
canvas.toBlob(r as BlobCallback, 'image/jpeg', 0.95)
);
resolve(imgBlob);
};
img.src = base64image;
});
}
export async function uploadImage(imagBlob: Blob, params: {
prompt: string;
position: { x: number; y: number };
date: number;
id: string;
room: string;
}): Promise<{
url: string;
filename: string;
}> {
// simple regex slugify string for file name
const promptSlug = slugify(params.prompt);
const key = `${params.position.x}_${params.position.y}`;
const fileName = `sd-${params.room}-${params.id}-${promptSlug}-${key}.jpeg`;
const file = new File([imagBlob], fileName, { type: 'image/jpeg' });
const formData = new FormData()
formData.append('file', file)
const response = await fetch(PUBLIC_API_BASE + "/uploadfile", {
method: 'POST',
body: formData
});
if (response.status === 200) {
const data = await response.json();
return data;
}
throw new Error('Failed to upload image');
}
export function round(pos: number, canvasSize: {
width: number;
height: number;
}) {
const max = canvasSize.width - FRAME_SIZE
const value = pos % GRID_SIZE < GRID_SIZE / 2 ? pos - (pos % GRID_SIZE) : pos + GRID_SIZE - (pos % GRID_SIZE);
return Math.max(0, Math.min(Math.round(value), max))
}
function slugify(text: string) {
if (!text) return '';
return text
.toString()
.toLowerCase()
.replace(/\s+/g, '-')
.replace(/[^\w\-]+/g, '')
.replace(/\-\-+/g, '-')
.replace(/^-+/, '')
.replace(/-+$/, '');
}