VibecoderMcSwaggins's picture
feat: Gradio Custom Component for NiiVue (#29)
227ab66 unverified
raw
history blame
3.52 kB
<script lang="ts">
import { createEventDispatcher, onMount } from "svelte";
import type { SelectData } from "@gradio/utils";
import { uploadToHuggingFace } from "@gradio/utils";
import {
BlockLabel,
Empty,
IconButton,
ShareButton,
IconButtonWrapper,
FullscreenButton,
DownloadLink
} from "@gradio/atoms";
import { Download, Image as ImageIcon } from "@gradio/icons";
import { get_coordinates_of_clicked_image } from "./utils";
import Image from "./Image.svelte";
import type { I18nFormatter } from "@gradio/utils";
import type { FileData } from "@gradio/client";
export let value: null | FileData;
export let label: string | undefined = undefined;
export let show_label: boolean;
export let buttons: string[] | null = null;
export let selectable = false;
export let i18n: I18nFormatter;
export let display_icon_button_wrapper_top_corner = false;
export let fullscreen = false;
export let show_button_background = true;
const dispatch = createEventDispatcher<{
change: string;
select: SelectData;
fullscreen: boolean;
}>();
const handle_click = (evt: MouseEvent): void => {
let coordinates = get_coordinates_of_clicked_image(evt);
if (coordinates) {
dispatch("select", { index: coordinates, value: null });
}
};
let image_container: HTMLElement;
</script>
<BlockLabel
{show_label}
Icon={ImageIcon}
label={!show_label ? "" : label || i18n("image.image")}
/>
{#if value == null || !value?.url}
<Empty unpadded_box={true} size="large"><ImageIcon /></Empty>
{:else}
<div class="image-container" bind:this={image_container}>
<IconButtonWrapper
display_top_corner={display_icon_button_wrapper_top_corner}
show_background={show_button_background}
>
{#if buttons === null ? true : buttons.includes("fullscreen")}
<FullscreenButton {fullscreen} on:fullscreen />
{/if}
{#if buttons === null ? true : buttons.includes("download")}
<DownloadLink href={value.url} download={value.orig_name || "image"}>
<IconButton Icon={Download} label={i18n("common.download")} />
</DownloadLink>
{/if}
{#if buttons === null ? true : buttons.includes("share")}
<ShareButton
{i18n}
on:share
on:error
formatter={async (value) => {
if (!value) return "";
let url = await uploadToHuggingFace(value, "url");
return `<img src="${url}" />`;
}}
{value}
/>
{/if}
</IconButtonWrapper>
<button on:click={handle_click}>
<div class:selectable class="image-frame">
<Image
src={value.url}
restProps={{ loading: "lazy", alt: "" }}
on:load
/>
</div>
</button>
</div>
{/if}
<style>
.image-container {
height: 100%;
position: relative;
min-width: var(--size-20);
}
.image-container button {
width: var(--size-full);
height: var(--size-full);
border-radius: var(--radius-lg);
display: flex;
align-items: center;
justify-content: center;
}
.image-frame :global(img) {
width: var(--size-full);
height: var(--size-full);
object-fit: scale-down;
}
.selectable {
cursor: crosshair;
}
:global(.fullscreen-controls svg) {
position: relative;
top: 0px;
}
:global(.image-container:fullscreen) {
background-color: black;
display: flex;
justify-content: center;
align-items: center;
}
:global(.image-container:fullscreen img) {
max-width: 90vw;
max-height: 90vh;
object-fit: scale-down;
}
.image-frame {
width: auto;
height: 100%;
display: flex;
align-items: center;
justify-content: center;
}
</style>