feat: Improve SD list data and API integration

2025-08-09 16:53:04 +08:00 · 2024-07-03 15:37:34 +08:00
parent 54401162bd
commit a16725ac17
14 changed files with 508 additions and 33 deletions
--- a/app/store/sd.ts
+++ b/app/store/sd.ts
@@ -0,0 +1,78 @@
+import { initDB, useIndexedDB } from "react-indexed-db-hook";
+import { StoreKey } from "@/app/constant";
+import { create, StoreApi } from "zustand";
+
+export const SdDbConfig = {
+  name: "@chatgpt-next-web/sd",
+  version: 1,
+  objectStoresMeta: [
+    {
+      store: StoreKey.SdList,
+      storeConfig: { keyPath: "id", autoIncrement: true },
+      storeSchema: [
+        { name: "model", keypath: "model", options: { unique: false } },
+        {
+          name: "model_name",
+          keypath: "model_name",
+          options: { unique: false },
+        },
+        { name: "status", keypath: "status", options: { unique: false } },
+        { name: "params", keypath: "params", options: { unique: false } },
+        { name: "img_data", keypath: "img_data", options: { unique: false } },
+        { name: "error", keypath: "error", options: { unique: false } },
+        {
+          name: "created_at",
+          keypath: "created_at",
+          options: { unique: false },
+        },
+      ],
+    },
+  ],
+};
+
+export function SdDbInit() {
+  initDB(SdDbConfig);
+}
+
+type SdStore = {
+  execCount: number;
+  execCountInc: () => void;
+};
+
+export const useSdStore = create<SdStore>()((set) => ({
+  execCount: 1,
+  execCountInc: () => set((state) => ({ execCount: state.execCount + 1 })),
+}));
+
+export function sendSdTask(data: any, db: any, inc: any) {
+  const formData = new FormData();
+  for (let paramsKey in data.params) {
+    formData.append(paramsKey, data.params[paramsKey]);
+  }
+  fetch("https://api.stability.ai/v2beta/stable-image/generate/" + data.model, {
+    method: "POST",
+    headers: {
+      Accept: "application/json",
+    },
+    body: formData,
+  })
+    .then((response) => response.json())
+    .then((resData) => {
+      if (resData.errors && resData.errors.length > 0) {
+        db.update({ ...data, status: "error", error: resData.errors[0] });
+        inc();
+        return;
+      }
+      if (resData.finish_reason === "SUCCESS") {
+        db.update({ ...data, status: "success", img_data: resData.image });
+      } else {
+        db.update({ ...data, status: "error", error: JSON.stringify(resData) });
+      }
+      inc();
+    })
+    .catch((error) => {
+      db.update({ ...data, status: "error", error: error.message });
+      console.error("Error:", error);
+      inc();
+    });
+}