move to npm package

package.json

@@ -17,6 +17,7 @@
     "@radix-ui/react-tabs": "^1.0.4",
     "class-variance-authority": "^0.7.0",
     "clsx": "^2.1.0",
+    "comfydeploy": "^0.0.11",
     "lucide-react": "^0.309.0",
     "next": "14.0.3",
     "react": "^18",

src/app/page.tsx

@@ -5,7 +5,6 @@ import { Button } from "@/components/ui/button";
 import { Card, CardContent, CardHeader } from "@/components/ui/card";
 import { Input } from "@/components/ui/input";
 import { Label } from "@/components/ui/label";
-import { Skeleton } from "@/components/ui/skeleton";
 import {
   checkStatus,
   generate,
@@ -24,7 +23,6 @@ import {
   SelectTrigger,
   SelectValue,
 } from "@/components/ui/select";
-import { Separator } from "@/components/ui/separator";
 
 import { Tabs, TabsContent, TabsList, TabsTrigger } from "@/components/ui/tabs";
 import { ImageGenerationResult } from "@/components/ImageGenerationResult";

src/lib/comfy-deploy.ts

@@ -1,133 +0,0 @@
-import { z } from "zod";
-
-const runTypes = z.object({
-  run_id: z.string(),
-});
-
-const runOutputTypes = z.object({
-  id: z.string(),
-  status: z.enum(["success", "failed", "running", "uploading", "not-started"]),
-  outputs: z.array(
-    z.object({
-      data: z.any(),
-    }),
-  ),
-});
-
-const uploadFileTypes = z.object({
-  upload_url: z.string(),
-  file_id: z.string(),
-  download_url: z.string(),
-});
-
-export class ComfyDeployClient {
-  apiBase: string = "https://www.comfydeploy.com/api";
-  apiToken: string;
-
-  constructor({ apiBase, apiToken }: { apiBase?: string; apiToken: string }) {
-    if (apiBase) {
-      this.apiBase = `${apiBase}/api`;
-    }
-    this.apiToken = apiToken;
-  }
-
-  async run({
-    deployment_id,
-    inputs,
-  }: {
-    deployment_id: string;
-    inputs?: Record<string, string>;
-  }) {
-    return fetch(`${this.apiBase}/run`, {
-      method: "POST",
-      headers: {
-        "Content-Type": "application/json",
-        authorization: `Bearer ${this.apiToken}`,
-      },
-      body: JSON.stringify({
-        deployment_id: deployment_id,
-        inputs: inputs,
-      }),
-      cache: "no-store",
-    })
-      .then((response) => {
-        console.log('response', response)
-        return response.json()})
-      .then((json) => {
-        console.log('json', json)
-        return runTypes.parse(json)})
-      .catch((err) => {
-        console.error('err', err);
-        return null;
-      });
-  }
-
-  async getRun(run_id: string) {
-    return await fetch(`${this.apiBase}/run?run_id=${run_id}`, {
-      method: "GET",
-      headers: {
-        "Content-Type": "application/json",
-        authorization: `Bearer ${this.apiToken}`,
-      },
-      cache: "no-store",
-    })
-      .then((response) => response.json())
-      .then((json) => runOutputTypes.parse(json))
-      .catch((err) => {
-        console.error(err);
-        return null;
-      });
-  }
-
-  async runSync(props: {
-    deployment_id: string;
-    inputs?: Record<string, string>;
-  }) {
-    const runResult = await this.run(props);
-    if (!runResult) return null;
-
-    // 5 minutes
-    const timeout = 60 * 5;
-    const interval = 1000;
-
-    let run: Awaited<ReturnType<typeof this.getRun>> = null;
-    for (let i = 0; i < timeout; i++) {
-      run = await this.getRun(runResult.run_id);
-      if (run && run.status == "success") {
-        break;
-      }
-      await new Promise((resolve) => setTimeout(resolve, interval));
-    }
-
-    if (!run) {
-      return {
-        id: runResult.run_id,
-      };
-    }
-
-    return run;
-  }
-
-  async getUploadUrl(type: string, file_size: number) {
-    const obj = {
-      type: type,
-      file_size: file_size.toString(),
-    };
-    const url = new URL(`${this.apiBase}/upload-url`);
-    url.search = new URLSearchParams(obj).toString();
-
-    return await fetch(url.href, {
-      method: "GET",
-      headers: {
-        authorization: `Bearer ${this.apiToken}`,
-      },
-      cache: "no-store",
-    })
-      .then((response) => response.json())
-      .then((json) => uploadFileTypes.parse(json))
-      .catch((err) => {
-        console.error(err);
-        return null;
-      });
-  }
-}

src/server/generate.tsx

@@ -1,6 +1,6 @@
 "use server"
 
-import { ComfyDeployClient } from "@/lib/comfy-deploy"
+import { ComfyDeployClient } from "comfydeploy"
 
 const client = new ComfyDeployClient({
     apiBase: process.env.COMFY_API_URL,