web/removebg: convert to a proper web worker

no more hanging ui :3
This commit is contained in:
wukko
2025-01-15 17:22:34 +06:00
parent 2f2d39dc4c
commit 28d8927c08
3 changed files with 66 additions and 42 deletions

View File

@@ -1,36 +1,49 @@
<script lang="ts">
import settings from "$lib/state/settings";
import RemoveBgWorker from '$lib/workers/removebg?worker';
import { onMount } from "svelte";
import { goto } from "$app/navigation";
import { downloadFile } from "$lib/download";
import { removeImageBackground } from "$lib/workers/removebg";
import { maskImage } from "$lib/workers/removebg";
import DropReceiver from "$components/misc/DropReceiver.svelte";
import FileReceiver from "$components/misc/FileReceiver.svelte";
import Skeleton from "$components/misc/Skeleton.svelte";
let imageContainer: HTMLElement;
let draggedOver = false;
let file: File | undefined;
let thinking = false;
let done = false;
let result: HTMLCanvasElement;
let imageContainer: HTMLElement;
let state: "empty" | "busy" | "done" = "empty";
const worker = new RemoveBgWorker();
const processImage = async () => {
if (file) {
thinking = true;
const removedBackground = await removeImageBackground(file);
if (removedBackground) {
thinking = false;
done = true;
result = removedBackground;
imageContainer.append(removedBackground);
}
if (!file) return;
state = "busy";
worker.postMessage({ file });
worker.onmessage = async (event) => {
const maskedCanvas = await maskImage(event.data.source, event.data.mask);
if (!maskedCanvas) {
state = "empty";
return;
};
state = "done";
result = maskedCanvas;
imageContainer.append(maskedCanvas);
};
worker.onerror = (e) => {
console.error("bg removal worker exploded:", e)
worker.terminate();
}
};
@@ -53,7 +66,7 @@
</script>
<DropReceiver bind:file bind:draggedOver id="cutout-container">
{#if !thinking && !done}
{#if state === "empty"}
<FileReceiver
bind:draggedOver
bind:file
@@ -68,27 +81,27 @@
{/if}
{/if}
{#if thinking}
{#if state === "busy"}
<div>thinking very hard rn...</div>
{/if}
{#if done}
{#if state === "done"}
<div>thought a lot, here's what i got:</div>
{/if}
{#if thinking || done}
{#if ["busy", "done"].includes(state)}
<div id="image-preview" bind:this={imageContainer}>
{#if !done}
{#if state === "busy"}
<Skeleton width="100%" height="100%" class="big" />
{/if}
</div>
{/if}
{#if done}
{#if state === "done"}
<div id="finished-actions">
<button
on:click={() => {
done = false;
state = "empty"
file = undefined;
}}
>

View File

@@ -0,0 +1 @@
export const ssr = false;