comfyui-deploy/web/src/components/OutputRender.tsx
2024-01-11 23:14:06 +08:00

28 lines
769 B
TypeScript

import { DownloadButton } from "./DownloadButton";
import { getFileDownloadUrl } from "@/server/getFileDownloadUrl";
export async function OutputRender(props: {
run_id: string;
filename: string;
}) {
if (
props.filename.endsWith(".png") ||
props.filename.endsWith(".gif") ||
props.filename.endsWith(".jpg") ||
props.filename.endsWith(".jpeg")
) {
const url = await getFileDownloadUrl(
`outputs/runs/${props.run_id}/${props.filename}`
);
return <img className="max-w-[200px]" alt={props.filename} src={url} />;
} else {
const url = await getFileDownloadUrl(
`outputs/runs/${props.run_id}/${props.filename}`
);
// console.log(url);
return <DownloadButton filename={props.filename} href={url} />;
}
}