supermemory/apps/web/components/document-modal/content/image-preview.tsx

"use client"

import { useCallback, useState } from "react"
import { cn } from "@lib/utils"

interface ImagePreviewProps {
	url: string
	title?: string | null
}

export function ImagePreview({ url, title }: ImagePreviewProps) {
	const [imageError, setImageError] = useState(false)
	const [isLoading, setIsLoading] = useState(true)
	const [retryKey, setRetryKey] = useState(0)

	// On first failure, wait briefly then force a re-render with a new key to
	// retry the fetch (covers transient R2 timing issues).
	// On second failure, give up and show the error state.
	const handleImageError = useCallback(() => {
		if (retryKey === 0) {
			setTimeout(() => setRetryKey(1), 500)
			return
		}
		setImageError(true)
		setIsLoading(false)
	}, [retryKey])

	if (imageError || !url) {
		return (
			<div className="flex items-center justify-center h-full text-[#737373]">
				<p>Failed to load image</p>
			</div>
		)
	}

	return (
		<div className="relative size-full overflow-hidden flex items-center justify-center bg-[#0B1017]">
			{isLoading && (
				<div className="absolute inset-0 bg-cover bg-center animate-pulse">
					<div className="size-full bg-[#1B1F24]" />
				</div>
			)}
			<div
				className="absolute inset-0 bg-cover bg-center"
				style={{
					backgroundImage: `url(${url})`,
					filter: "blur(100px)",
					transform: "scale(1.1)",
					opacity: isLoading ? 0.5 : 1,
				}}
			/>
			<div className="absolute inset-0 bg-black/30" />
			<img
				key={retryKey}
				src={url}
				alt={title || "Image preview"}
				className={cn(
					"relative max-w-full max-h-full size-auto object-contain z-10",
					isLoading && "opacity-0",
				)}
				onError={handleImageError}
				onLoad={() => setIsLoading(false)}
				loading="lazy"
			/>
		</div>
	)
}