30 lines
598 B
TypeScript
30 lines
598 B
TypeScript
import * as fal from '@fal-ai/serverless-client';
|
|
|
|
export type GenerateImageInput = {
|
|
prompt: string;
|
|
};
|
|
|
|
type ImageType = 'gif' | 'png' | 'jpg' | 'jpeg';
|
|
type ImageDataUri = `data:image/${ImageType};base64,${string}`;
|
|
|
|
fal.config({
|
|
credentials: {
|
|
userId: '',
|
|
keyId: '',
|
|
keySecret: '',
|
|
},
|
|
});
|
|
|
|
export async function generateImage(
|
|
input: GenerateImageInput
|
|
): Promise<ImageDataUri> {
|
|
const result = await fal.run(
|
|
'a51c0ca0-9011-4ff0-8dc1-2ac0b42a9fd0/generate',
|
|
{
|
|
input,
|
|
}
|
|
);
|
|
const data = result['raw_data'];
|
|
return `data:image/jpg;base64,${data}`;
|
|
}
|