Daniel Rochetti c3a3c3d21a
chore: upgrade dependencies and tooling (#81)
* chore: remove uuid-random dependency

* chore: upgrade prettier

* chore(client): bump version for release
2024-08-07 15:45:15 -07:00

187 lines
5.6 KiB
TypeScript

"use client";
import * as fal from "@fal-ai/serverless-client";
import { useMemo, useState } from "react";
// @snippet:start(client.config)
fal.config({
// credentials: 'FAL_KEY_ID:FAL_KEY_SECRET',
proxyUrl: "/api/fal/proxy", // the built-int nextjs proxy
// proxyUrl: 'http://localhost:3333/api/fal/proxy', // or your own external proxy
});
// @snippet:end
// @snippet:start(client.result.type)
type Image = {
url: string;
file_name: string;
file_size: number;
};
type Result = {
image: Image;
};
// @snippet:end
type ErrorProps = {
error: any;
};
function Error(props: ErrorProps) {
if (!props.error) {
return null;
}
return (
<div
className="mb-4 rounded bg-red-50 p-4 text-sm text-red-800 dark:bg-gray-800 dark:text-red-400"
role="alert"
>
<span className="font-medium">Error</span> {props.error.message}
</div>
);
}
const DEFAULT_PROMPT =
"(masterpiece:1.4), (best quality), (detailed), Medieval village scene with busy streets and castle in the distance";
export default function Home() {
// @snippet:start("client.ui.state")
// Input state
const [prompt, setPrompt] = useState<string>(DEFAULT_PROMPT);
const [imageFile, setImageFile] = useState<File | null>(null);
// Result state
const [loading, setLoading] = useState(false);
const [error, setError] = useState<Error | null>(null);
const [result, setResult] = useState<Result | null>(null);
const [logs, setLogs] = useState<string[]>([]);
const [elapsedTime, setElapsedTime] = useState<number>(0);
// @snippet:end
const image = useMemo(() => {
if (!result) {
return null;
}
if (result.image) {
return result.image;
}
return null;
}, [result]);
const reset = () => {
setLoading(false);
setError(null);
setResult(null);
setLogs([]);
setElapsedTime(0);
};
const generateImage = async () => {
reset();
// @snippet:start("client.queue.subscribe")
setLoading(true);
const start = Date.now();
try {
const result: Result = await fal.subscribe("fal-ai/illusion-diffusion", {
input: {
prompt,
image_url: imageFile,
image_size: "square_hd",
},
logs: true,
onQueueUpdate(update) {
setElapsedTime(Date.now() - start);
if (
update.status === "IN_PROGRESS" ||
update.status === "COMPLETED"
) {
setLogs((update.logs || []).map((log) => log.message));
}
},
});
setResult(result);
} catch (error: any) {
setError(error);
} finally {
setLoading(false);
setElapsedTime(Date.now() - start);
}
// @snippet:end
};
return (
<div className="min-h-screen bg-gray-100 dark:bg-gray-900">
<main className="container flex w-full flex-1 flex-col items-center justify-center space-y-8 py-10 text-gray-900 dark:text-gray-50">
<h1 className="mb-8 text-4xl font-bold">
Hello <code className="font-light text-pink-600">fal</code>
</h1>
<div className="w-full text-lg">
<label htmlFor="prompt" className="mb-2 block text-current">
Image
</label>
<input
className="w-full rounded border border-black/20 bg-black/10 p-2 text-lg dark:border-white/10 dark:bg-white/5"
id="image_url"
name="image_url"
type="file"
placeholder="Choose a file"
accept="image/*"
onChange={(e) => setImageFile(e.target.files?.[0] ?? null)}
/>
</div>
<div className="w-full text-lg">
<label htmlFor="prompt" className="mb-2 block text-current">
Prompt
</label>
<input
className="w-full rounded border border-black/20 bg-black/10 p-2 text-lg dark:border-white/10 dark:bg-white/5"
id="prompt"
name="prompt"
placeholder="Imagine..."
value={prompt}
autoComplete="off"
onChange={(e) => setPrompt(e.target.value)}
onBlur={(e) => setPrompt(e.target.value.trim())}
/>
</div>
<button
onClick={(e) => {
e.preventDefault();
generateImage();
}}
className="focus:shadow-outline mx-auto rounded bg-indigo-600 py-3 px-6 text-lg font-bold text-white hover:bg-indigo-700 focus:outline-none"
disabled={loading}
>
{loading ? "Generating..." : "Generate Image"}
</button>
<Error error={error} />
<div className="flex w-full flex-col space-y-4">
<div className="mx-auto">
{image && (
// eslint-disable-next-line @next/next/no-img-element
<img src={image.url} alt="" />
)}
</div>
<div className="space-y-2">
<h3 className="text-xl font-light">JSON Result</h3>
<p className="text-current/80 text-sm">
{`Elapsed Time (seconds): ${(elapsedTime / 1000).toFixed(2)}`}
</p>
<pre className="h-60 w-full overflow-auto whitespace-pre rounded bg-black/70 font-mono text-sm text-white/80">
{result
? JSON.stringify(result, null, 2)
: "// result pending..."}
</pre>
</div>
<div className="space-y-2">
<h3 className="text-xl font-light">Logs</h3>
<pre className="h-60 w-full overflow-auto whitespace-pre rounded bg-black/70 font-mono text-sm text-white/80">
{logs.filter(Boolean).join("\n")}
</pre>
</div>
</div>
</main>
</div>
);
}