From adeadd3586ef478e0524ab7e13234b17d0d33d9c Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Fri, 10 Nov 2023 22:10:16 +0800
Subject: [PATCH] add image gen api

---
 src/addImage.tsx | 317 +++++++++++++++++++++++++++++++++++++++++++++++
 src/app.tsx      |   8 +-
 src/chatbox.tsx  | 143 ++-------------------
 src/settings.tsx |  14 ++-
 4 files changed, 345 insertions(+), 137 deletions(-)
 create mode 100644 src/addImage.tsx

diff --git a/src/addImage.tsx b/src/addImage.tsx
new file mode 100644
index 0000000..8cbc4bb
--- /dev/null
+++ b/src/addImage.tsx
@@ -0,0 +1,317 @@
+import { useState } from "preact/hooks";
+import { ChatStore } from "./app";
+import { MessageDetail } from "./chatgpt";
+import { Tr } from "./translate";
+
+interface Props {
+  chatStore: ChatStore;
+  setChatStore: (cs: ChatStore) => void;
+  images: MessageDetail[];
+  setShowAddImage: (se: boolean) => void;
+  setImages: (images: MessageDetail[]) => void;
+}
+interface ImageResponse {
+  url?: string;
+  b64_json?: string;
+  revised_prompt: string;
+}
+export function AddImage({
+  chatStore,
+  setChatStore,
+  setShowAddImage,
+  setImages,
+  images,
+}: Props) {
+  const [enableHighResolution, setEnableHighResolution] = useState(true);
+  const [imageGenPrompt, setImageGenPrompt] = useState("");
+  const [imageGenModel, setImageGenModel] = useState("dall-e-2");
+  const [imageGenN, setImageGenN] = useState(1);
+  const [imageGenQuality, setImageGEnQuality] = useState("standard");
+  const [imageGenResponseFormat, setImageGenResponseFormat] =
+    useState("b64_json");
+  const [imageGenSize, setImageGenSize] = useState("1024x1024");
+  const [imageGenStyle, setImageGenStyle] = useState("vivid");
+  const [imageGenGenerating, setImageGenGenerating] = useState(false);
+  useState("b64_json");
+  return (
+    <div
+      className="absolute z-10 bg-black bg-opacity-50 w-full h-full flex justify-center items-center left-0 top-0 overflow-scroll"
+      onClick={() => {
+        setShowAddImage(false);
+      }}
+    >
+      <div
+        className="bg-white rounded p-2 z-20"
+        onClick={(event) => {
+          event.stopPropagation();
+        }}
+      >
+        <h2>Add Images</h2>
+        <span>
+          <button
+            className="disabled:line-through disabled:bg-slate-500 rounded m-1 p-1 border-2 bg-cyan-400 hover:bg-cyan-600"
+            onClick={() => {
+              const image_url = prompt("Image URL");
+              if (!image_url) {
+                return;
+              }
+              setImages([
+                ...images,
+                {
+                  type: "image_url",
+                  image_url: {
+                    url: image_url,
+                    detail: enableHighResolution ? "high" : "low",
+                  },
+                },
+              ]);
+            }}
+          >
+            Add from URL
+          </button>
+          <button
+            className="disabled:line-through disabled:bg-slate-500 rounded m-1 p-1 border-2 bg-cyan-400 hover:bg-cyan-600"
+            onClick={() => {
+              // select file and load it to base64 image URL format
+              const input = document.createElement("input");
+              input.type = "file";
+              input.accept = "image/*";
+              input.onchange = (event) => {
+                const file = (event.target as HTMLInputElement).files?.[0];
+                if (!file) {
+                  return;
+                }
+                const reader = new FileReader();
+                reader.readAsDataURL(file);
+                reader.onloadend = () => {
+                  const base64data = reader.result;
+                  setImages([
+                    ...images,
+                    {
+                      type: "image_url",
+                      image_url: {
+                        url: String(base64data),
+                        detail: enableHighResolution ? "high" : "low",
+                      },
+                    },
+                  ]);
+                };
+              };
+              input.click();
+            }}
+          >
+            Add from local file
+          </button>
+          <span
+            onClick={() => {
+              setEnableHighResolution(!enableHighResolution);
+            }}
+          >
+            <label>High resolution</label>
+            <input type="checkbox" checked={enableHighResolution} />
+          </span>
+        </span>
+        {chatStore.image_gen_api && chatStore.image_gen_key && (
+          <div className="flex flex-col">
+            <hr className="my-2" />
+            <h3>Generate Image</h3>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <label>Prompt: </label>
+              <textarea
+                className="border rounded border-gray-400"
+                value={imageGenPrompt}
+                onChange={(e: any) => {
+                  setImageGenPrompt(e.target.value);
+                }}
+              />
+            </span>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <label>Model: </label>
+              <select
+                value={imageGenModel}
+                onChange={(e: any) => {
+                  setImageGenModel(e.target.value);
+                }}
+              >
+                <option value="dall-e-3">DALL-E 3</option>
+                <option value="dall-e-2">DALL-E 2</option>
+              </select>
+            </span>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <label>n: </label>
+              <input
+                value={imageGenN}
+                type="number"
+                min={1}
+                max={10}
+                onChange={(e: any) => setImageGenN(parseInt(e.target.value))}
+              />
+            </span>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <label>Quality: </label>
+              <select
+                value={imageGenQuality}
+                onChange={(e: any) => setImageGEnQuality(e.target.value)}
+              >
+                <option value="hd">HD</option>
+                <option value="standard">Standard</option>
+              </select>
+            </span>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <label>Response Format: </label>
+              <select
+                value={imageGenResponseFormat}
+                onChange={(e: any) => setImageGenResponseFormat(e.target.value)}
+              >
+                <option value="b64_json">b64_json</option>
+                <option value="url">url</option>
+              </select>
+            </span>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <label>Size: </label>
+              <select
+                value={imageGenSize}
+                onChange={(e: any) => setImageGenSize(e.target.value)}
+              >
+                <option value="256x256">256x256 (dall-e-2)</option>
+                <option value="512x512">512x512 (dall-e-2)</option>
+                <option value="1024x1024">1024x1024 (dall-e-2/3)</option>
+                <option value="1792x1024">1792x1024 (dall-e-3)</option>
+                <option value="1024x1792">1024x1792 (dall-e-3)</option>
+              </select>
+            </span>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <label>Style (only dall-e-3): </label>
+              <select
+                value={imageGenStyle}
+                onChange={(e: any) => setImageGenStyle(e.target.value)}
+              >
+                <option value="vivid">vivid</option>
+                <option value="natural">natural</option>
+              </select>
+            </span>
+            <span className="flex flex-row justify-between m-1 p-1">
+              <button
+                className="bg-sky-400 m-1 p-1 rounded disabled:bg-slate-500"
+                disabled={imageGenGenerating}
+                onClick={async () => {
+                  setImageGenGenerating(true);
+                  const body: any = {
+                    prompt: imageGenPrompt,
+                    model: imageGenModel,
+                    n: imageGenN,
+                    quality: imageGenQuality,
+                    response_format: imageGenResponseFormat,
+                    size: imageGenSize,
+                  };
+                  if (imageGenModel === "dall-e-3") {
+                    body.style = imageGenStyle;
+                  }
+                  const resp: ImageResponse[] = (
+                    await fetch(chatStore.image_gen_api, {
+                      method: "POST",
+                      headers: {
+                        "Content-Type": "application/json",
+                        Authorization: `Bearer ${chatStore.image_gen_key}`,
+                      },
+                      body: JSON.stringify(body),
+                    }).then((resp) => resp.json())
+                  ).data;
+
+                  for (const image of resp) {
+                    let url = "";
+                    if (image.url) url = image.url;
+                    if (image.b64_json)
+                      url = "data:image/png;base64," + image.b64_json;
+                    if (!url) continue;
+
+                    chatStore.history.push({
+                      role: "user",
+                      content: [
+                        {
+                          type: "image_url",
+                          image_url: {
+                            url,
+                            detail: "low",
+                          },
+                        },
+                        {
+                          type: "text",
+                          text: image.revised_prompt,
+                        },
+                      ],
+                      hide: false,
+                      token: 65,
+                      example: false,
+                    });
+
+                    setChatStore({ ...chatStore });
+                  }
+
+                  setImageGenGenerating(false);
+                  console.log("image resp", resp);
+                }}
+              >
+                {Tr("Generate")}
+              </button>
+            </span>
+          </div>
+        )}
+        <div className="flex flex-wrap">
+          {images.map((image, index) => (
+            <div className="flex flex-col">
+              {image.type === "image_url" && (
+                <img
+                  className="rounded m-1 p-1 border-2 border-gray-400 w-32"
+                  src={image.image_url?.url}
+                />
+              )}
+              <span className="flex justify-between">
+                <button
+                  onClick={() => {
+                    const image_url = prompt("Image URL");
+                    if (!image_url) {
+                      return;
+                    }
+                    images[index].image_url = {
+                      url: image_url,
+                      detail: enableHighResolution ? "high" : "low",
+                    };
+                    setImages([...images]);
+                  }}
+                >
+                  🖋
+                </button>
+                <span
+                  onClick={() => {
+                    if (image.image_url === undefined) return;
+                    image.image_url.detail =
+                      image.image_url?.detail === "low" ? "high" : "low";
+                    setImages([...images]);
+                  }}
+                >
+                  <label>HiRes</label>
+                  <input
+                    type="checkbox"
+                    checked={image.image_url?.detail === "high"}
+                  />
+                </span>
+                <button
+                  onClick={() => {
+                    if (!confirm("Are you sure to delete this image?")) {
+                      return;
+                    }
+                    images.splice(index, 1);
+                    setImages([...images]);
+                  }}
+                >
+                  ❌
+                </button>
+              </span>
+            </div>
+          ))}
+        </div>
+      </div>
+    </div>
+  );
+}
diff --git a/src/app.tsx b/src/app.tsx
index f54320b..d75a80c 100644
--- a/src/app.tsx
+++ b/src/app.tsx
@@ -53,6 +53,8 @@ export interface ChatStore {
   tts_voice: string;
   tts_speed: number;
   tts_speed_enabled: boolean;
+  image_gen_api: string;
+  image_gen_key: string;
 }
 
 const _defaultAPIEndpoint = "https://api.openai.com/v1/chat/completions";
@@ -70,7 +72,9 @@ export const newChatStore = (
   tts_key = "",
   tts_speed = 1.0,
   tts_speed_enabled = false,
-  toolsString = ""
+  toolsString = "",
+  image_gen_api = "",
+  image_gen_key = ""
 ): ChatStore => {
   return {
     chatgpt_api_web_version: CHATGPT_API_WEB_VERSION,
@@ -103,6 +107,8 @@ export const newChatStore = (
     tts_voice: "alloy",
     tts_speed: tts_speed,
     tts_speed_enabled: tts_speed_enabled,
+    image_gen_api: image_gen_api,
+    image_gen_key: image_gen_key,
   };
 };
 
diff --git a/src/chatbox.tsx b/src/chatbox.tsx
index 0a53ef2..3137d57 100644
--- a/src/chatbox.tsx
+++ b/src/chatbox.tsx
@@ -22,6 +22,7 @@ import Message from "./message";
 import models from "./models";
 import Settings from "./settings";
 import getDefaultParams from "./getDefaultParam";
+import { AddImage } from "./addImage";
 
 export interface TemplateChatStore extends ChatStore {
   name: string;
@@ -38,7 +39,6 @@ export default function ChatBOX(props: {
   if (chatStore === undefined) return <div></div>;
   const [inputMsg, setInputMsg] = useState("");
   const [images, setImages] = useState<MessageDetail[]>([]);
-  const [enableHighResolution, setEnableHighResolution] = useState(true);
   const [showAddImage, setShowAddImage] = useState(false);
   const [showGenerating, setShowGenerating] = useState(false);
   const [generatingMessage, setGeneratingMessage] = useState("");
@@ -687,140 +687,13 @@ export default function ChatBOX(props: {
           </button>
         )}
         {showAddImage && (
-          <div
-            className="absolute z-10 bg-black bg-opacity-50 w-full h-full flex justify-center items-center left-0 top-0 overflow-scroll"
-            onClick={() => {
-              setShowAddImage(false);
-            }}
-          >
-            <div
-              className="bg-white rounded p-2 z-20"
-              onClick={(event) => {
-                event.stopPropagation();
-              }}
-            >
-              <h2>Add Images</h2>
-              <span>
-                <button
-                  className="disabled:line-through disabled:bg-slate-500 rounded m-1 p-1 border-2 bg-cyan-400 hover:bg-cyan-600"
-                  onClick={() => {
-                    const image_url = prompt("Image URL");
-                    if (!image_url) {
-                      return;
-                    }
-                    setImages([
-                      ...images,
-                      {
-                        type: "image_url",
-                        image_url: {
-                          url: image_url,
-                          detail: enableHighResolution ? "high" : "low",
-                        },
-                      },
-                    ]);
-                  }}
-                >
-                  Add from URL
-                </button>
-                <button
-                  className="disabled:line-through disabled:bg-slate-500 rounded m-1 p-1 border-2 bg-cyan-400 hover:bg-cyan-600"
-                  onClick={() => {
-                    // select file and load it to base64 image URL format
-                    const input = document.createElement("input");
-                    input.type = "file";
-                    input.accept = "image/*";
-                    input.onchange = (event) => {
-                      const file = (event.target as HTMLInputElement)
-                        .files?.[0];
-                      if (!file) {
-                        return;
-                      }
-                      const reader = new FileReader();
-                      reader.readAsDataURL(file);
-                      reader.onloadend = () => {
-                        const base64data = reader.result;
-                        setImages([
-                          ...images,
-                          {
-                            type: "image_url",
-                            image_url: {
-                              url: String(base64data),
-                              detail: enableHighResolution ? "high" : "low",
-                            },
-                          },
-                        ]);
-                      };
-                    };
-                    input.click();
-                  }}
-                >
-                  Add from local file
-                </button>
-                <span
-                  onClick={() => {
-                    setEnableHighResolution(!enableHighResolution);
-                  }}
-                >
-                  <label>High resolution</label>
-                  <input type="checkbox" checked={enableHighResolution} />
-                </span>
-              </span>
-              <div className="flex flex-wrap">
-                {images.map((image, index) => (
-                  <div className="flex flex-col">
-                    {image.type === "image_url" && (
-                      <img
-                        className="rounded m-1 p-1 border-2 border-gray-400 w-32"
-                        src={image.image_url?.url}
-                      />
-                    )}
-                    <span className="flex justify-between">
-                      <button
-                        onClick={() => {
-                          const image_url = prompt("Image URL");
-                          if (!image_url) {
-                            return;
-                          }
-                          images[index].image_url = {
-                            url: image_url,
-                            detail: enableHighResolution ? "high" : "low",
-                          };
-                          setImages([...images]);
-                        }}
-                      >
-                        🖋
-                      </button>
-                      <span
-                        onClick={() => {
-                          if (image.image_url === undefined) return;
-                          image.image_url.detail =
-                            image.image_url?.detail === "low" ? "high" : "low";
-                          setImages([...images]);
-                        }}
-                      >
-                        <label>HiRes</label>
-                        <input
-                          type="checkbox"
-                          checked={image.image_url?.detail === "high"}
-                        />
-                      </span>
-                      <button
-                        onClick={() => {
-                          if (!confirm("Are you sure to delete this image?")) {
-                            return;
-                          }
-                          images.splice(index, 1);
-                          setImages([...images]);
-                        }}
-                      >
-                        ❌
-                      </button>
-                    </span>
-                  </div>
-                ))}
-              </div>
-            </div>
-          </div>
+          <AddImage
+            chatStore={chatStore}
+            setChatStore={setChatStore}
+            setShowAddImage={setShowAddImage}
+            images={images}
+            setImages={setImages}
+          />
         )}
         <textarea
           rows={Math.min(10, (inputMsg.match(/\n/g) || []).length + 2)}
diff --git a/src/settings.tsx b/src/settings.tsx
index 119d595..01d74fa 100644
--- a/src/settings.tsx
+++ b/src/settings.tsx
@@ -87,7 +87,9 @@ const Input = (props: {
     | "whisper_api"
     | "whisper_key"
     | "tts_api"
-    | "tts_key";
+    | "tts_key"
+    | "image_gen_api"
+    | "image_gen_key";
   help: string;
 }) => {
   const [hideInput, setHideInput] = useState(true);
@@ -492,6 +494,16 @@ export default (props: {
             help={"TTS Speed"}
             {...props}
           />
+          <Input
+            field="image_gen_api"
+            help="DALL image gen key, eg. https://api.openai.com/v1/images/generations"
+            {...props}
+          />
+          <Input
+            field="image_gen_key"
+            help="image generation service api key"
+            {...props}
+          />
 
           <div className="flex justify-between">
             <p className="m-2 p-2">