From 75a431360beec4c5158836afe96fe924ad2dc49f Mon Sep 17 00:00:00 2001
From: ecwu <bitlabwzh@gmail.com>
Date: Sun, 5 Jan 2025 00:11:31 +0800
Subject: [PATCH] refactor: reorganize SetAPIsTemplate component and update
 imports for improved structure

---
 src/components/ListAPI.tsx               | 21 +----
 src/components/MessageBubble.tsx         | 95 ++++++++++++++++++++++-
 src/components/Settings.tsx              |  2 +-
 src/{ => components}/setAPIsTemplate.tsx |  6 +-
 src/tts.tsx                              | 97 ------------------------
 5 files changed, 100 insertions(+), 121 deletions(-)
 rename src/{ => components}/setAPIsTemplate.tsx (95%)
 delete mode 100644 src/tts.tsx

diff --git a/src/components/ListAPI.tsx b/src/components/ListAPI.tsx
index e7e2b9e..4f06e0c 100644
--- a/src/components/ListAPI.tsx
+++ b/src/components/ListAPI.tsx
@@ -19,25 +19,18 @@ import {
 
 interface APITemplateItemProps {
   label: string;
-  shortLabel: string;
   apiField: string;
   keyField: string;
 }
-function ListAPIs({
-  label,
-  shortLabel,
-  apiField,
-  keyField,
-}: APITemplateItemProps) {
+function ListAPIs({ label, apiField, keyField }: APITemplateItemProps) {
   const ctx = useContext(AppContext);
   if (ctx === null) return <></>;
 
   return (
     <NavigationMenuItem>
       <NavigationMenuTrigger>
-        <span className="lg:hidden">{shortLabel}</span>
+        {label}{" "}
         <span className="hidden lg:inline">
-          {label}{" "}
           {ctx.templateAPIs.find(
             (t) =>
               ctx.chatStore[apiField as keyof ChatStore] === t.endpoint &&
@@ -215,7 +208,6 @@ const ListAPI: React.FC = () => {
           {ctx.templateAPIs.length > 0 && (
             <ListAPIs
               label="Chat API"
-              shortLabel="API"
               apiField="apiEndpoint"
               keyField="apiKey"
             />
@@ -223,23 +215,16 @@ const ListAPI: React.FC = () => {
           {ctx.templateAPIsWhisper.length > 0 && (
             <ListAPIs
               label="Whisper API"
-              shortLabel="Whisper"
               apiField="whisper_api"
               keyField="whisper_key"
             />
           )}
           {ctx.templateAPIsTTS.length > 0 && (
-            <ListAPIs
-              label="TTS API"
-              shortLabel="TTS"
-              apiField="tts_api"
-              keyField="tts_key"
-            />
+            <ListAPIs label="TTS API" apiField="tts_api" keyField="tts_key" />
           )}
           {ctx.templateAPIsImageGen.length > 0 && (
             <ListAPIs
               label="Image Gen API"
-              shortLabel="ImgGen"
               apiField="image_gen_api"
               keyField="image_gen_key"
             />
diff --git a/src/components/MessageBubble.tsx b/src/components/MessageBubble.tsx
index d99bd38..a6c5984 100644
--- a/src/components/MessageBubble.tsx
+++ b/src/components/MessageBubble.tsx
@@ -1,11 +1,11 @@
 import { XMarkIcon } from "@heroicons/react/24/outline";
 import Markdown from "react-markdown";
-import { useContext, useState } from "react";
+import { useContext, useState, useMemo } from "react";
 import { ChatStoreMessage } from "@/types/chatstore";
+import { addTotalCost } from "@/utils/totalCost";
 
 import { Tr } from "@/translate";
 import { getMessageText } from "@/chatgpt";
-import TTSButton, { TTSPlay } from "@/tts";
 import { EditMessage } from "@/editMessage";
 import logprobToColor from "@/utils/logprob";
 import {
@@ -15,12 +15,15 @@ import {
   ChatBubbleActionWrapper,
 } from "@/components/ui/chat/chat-bubble";
 import { Badge } from "@/components/ui/badge";
+import { Button } from "@/components/ui/button";
 import { useToast } from "@/hooks/use-toast";
 import {
   ClipboardIcon,
   PencilIcon,
   MessageSquareOffIcon,
   MessageSquarePlusIcon,
+  AudioLinesIcon,
+  LoaderCircleIcon,
 } from "lucide-react";
 import { AppContext } from "@/pages/App";
 
@@ -143,6 +146,94 @@ function MessageToolResp({ chat, copyToClipboard }: ToolRespondMessageProps) {
   );
 }
 
+interface TTSProps {
+  chat: ChatStoreMessage;
+}
+interface TTSPlayProps {
+  chat: ChatStoreMessage;
+}
+export function TTSPlay(props: TTSPlayProps) {
+  const src = useMemo(() => {
+    if (props.chat.audio instanceof Blob) {
+      return URL.createObjectURL(props.chat.audio);
+    }
+    return "";
+  }, [props.chat.audio]);
+
+  if (props.chat.hide) {
+    return <></>;
+  }
+  if (props.chat.audio instanceof Blob) {
+    return <audio className="w-64" src={src} controls />;
+  }
+  return <></>;
+}
+function TTSButton(props: TTSProps) {
+  const [generating, setGenerating] = useState(false);
+  const ctx = useContext(AppContext);
+  if (!ctx) return <div>error</div>;
+
+  return (
+    <Button
+      variant="ghost"
+      size="icon"
+      onClick={() => {
+        const api = ctx.chatStore.tts_api;
+        const api_key = ctx.chatStore.tts_key;
+        const model = "tts-1";
+        const input = getMessageText(props.chat);
+        const voice = ctx.chatStore.tts_voice;
+
+        const body: Record<string, any> = {
+          model,
+          input,
+          voice,
+          response_format: ctx.chatStore.tts_format || "mp3",
+        };
+        if (ctx.chatStore.tts_speed_enabled) {
+          body["speed"] = ctx.chatStore.tts_speed;
+        }
+
+        setGenerating(true);
+
+        fetch(api, {
+          method: "POST",
+          headers: {
+            Authorization: `Bearer ${api_key}`,
+            "Content-Type": "application/json",
+          },
+          body: JSON.stringify(body),
+        })
+          .then((response) => response.blob())
+          .then((blob) => {
+            // update price
+            const cost = (input.length * 0.015) / 1000;
+            ctx.chatStore.cost += cost;
+            addTotalCost(cost);
+            ctx.setChatStore({ ...ctx.chatStore });
+
+            // save blob
+            props.chat.audio = blob;
+            ctx.setChatStore({ ...ctx.chatStore });
+
+            const url = URL.createObjectURL(blob);
+            const audio = new Audio(url);
+            audio.play();
+          })
+          .finally(() => {
+            setGenerating(false);
+          });
+      }}
+    >
+      {generating ? (
+        <LoaderCircleIcon className="h-4 w-4 animate-spin" />
+      ) : (
+        <AudioLinesIcon className="h-4 w-4" />
+      )}
+    </Button>
+  );
+}
+
 export default function Message(props: { messageIndex: number }) {
   const ctx = useContext(AppContext);
   if (ctx === null) return <></>;
diff --git a/src/components/Settings.tsx b/src/components/Settings.tsx
index ab84823..70a904a 100644
--- a/src/components/Settings.tsx
+++ b/src/components/Settings.tsx
@@ -12,7 +12,7 @@ import {
 import { models } from "@/types/models";
 import { tr, Tr, langCodeContext, LANG_OPTIONS } from "@/translate";
 import { isVailedJSON } from "@/utils/isVailedJSON";
-import { SetAPIsTemplate } from "@/setAPIsTemplate";
+import { SetAPIsTemplate } from "@/components/setAPIsTemplate";
 import { autoHeight } from "@/utils/textAreaHelp";
 import { getDefaultParams } from "@/utils/getDefaultParam";
 
diff --git a/src/setAPIsTemplate.tsx b/src/components/setAPIsTemplate.tsx
similarity index 95%
rename from src/setAPIsTemplate.tsx
rename to src/components/setAPIsTemplate.tsx
index e120eb2..4f1b8e4 100644
--- a/src/setAPIsTemplate.tsx
+++ b/src/components/setAPIsTemplate.tsx
@@ -11,7 +11,7 @@ import {
   DialogTitle,
   DialogTrigger,
 } from "@/components/ui/dialog";
-import { Button } from "./components/ui/button";
+import { Button } from "./ui/button";
 import { Input } from "@/components/ui/input";
 import { Label } from "@/components/ui/label";
 import { SaveIcon } from "lucide-react";
@@ -71,12 +71,12 @@ export function SetAPIsTemplate({
                   }
                   return;
                 }
-                const tmp: TemplateAPI = {
+                const temp: TemplateAPI = {
                   name: name.value,
                   endpoint,
                   key: APIkey,
                 };
-                temps.push(tmp);
+                temps.push(temp);
                 setTemps([...temps]);
               }}
             >
diff --git a/src/tts.tsx b/src/tts.tsx
deleted file mode 100644
index 26f3b90..0000000
--- a/src/tts.tsx
+++ /dev/null
@@ -1,97 +0,0 @@
-import { SpeakerWaveIcon } from "@heroicons/react/24/outline";
-import { useContext, useMemo, useState } from "react";
-
-import { addTotalCost } from "@/utils/totalCost";
-import { ChatStore, ChatStoreMessage } from "@/types/chatstore";
-import { Message, getMessageText } from "@/chatgpt";
-import { AudioLinesIcon, LoaderCircleIcon } from "lucide-react";
-import { Button } from "./components/ui/button";
-import { AppContext } from "./pages/App";
-
-interface TTSProps {
-  chat: ChatStoreMessage;
-}
-interface TTSPlayProps {
-  chat: ChatStoreMessage;
-}
-export function TTSPlay(props: TTSPlayProps) {
-  const src = useMemo(() => {
-    if (props.chat.audio instanceof Blob) {
-      return URL.createObjectURL(props.chat.audio);
-    }
-    return "";
-  }, [props.chat.audio]);
-
-  if (props.chat.hide) {
-    return <></>;
-  }
-  if (props.chat.audio instanceof Blob) {
-    return <audio className="w-64" src={src} controls />;
-  }
-  return <></>;
-}
-export default function TTSButton(props: TTSProps) {
-  const [generating, setGenerating] = useState(false);
-  const ctx = useContext(AppContext);
-  if (!ctx) return <div>error</div>;
-
-  return (
-    <Button
-      variant="ghost"
-      size="icon"
-      onClick={() => {
-        const api = ctx.chatStore.tts_api;
-        const api_key = ctx.chatStore.tts_key;
-        const model = "tts-1";
-        const input = getMessageText(props.chat);
-        const voice = ctx.chatStore.tts_voice;
-
-        const body: Record<string, any> = {
-          model,
-          input,
-          voice,
-          response_format: ctx.chatStore.tts_format || "mp3",
-        };
-        if (ctx.chatStore.tts_speed_enabled) {
-          body["speed"] = ctx.chatStore.tts_speed;
-        }
-
-        setGenerating(true);
-
-        fetch(api, {
-          method: "POST",
-          headers: {
-            Authorization: `Bearer ${api_key}`,
-            "Content-Type": "application/json",
-          },
-          body: JSON.stringify(body),
-        })
-          .then((response) => response.blob())
-          .then((blob) => {
-            // update price
-            const cost = (input.length * 0.015) / 1000;
-            ctx.chatStore.cost += cost;
-            addTotalCost(cost);
-            ctx.setChatStore({ ...ctx.chatStore });
-
-            // save blob
-            props.chat.audio = blob;
-            ctx.setChatStore({ ...ctx.chatStore });
-
-            const url = URL.createObjectURL(blob);
-            const audio = new Audio(url);
-            audio.play();
-          })
-          .finally(() => {
-            setGenerating(false);
-          });
-      }}
-    >
-      {generating ? (
-        <LoaderCircleIcon className="h-4 w-4 animate-spin" />
-      ) : (
-        <AudioLinesIcon className="h-4 w-4" />
-      )}
-    </Button>
-  );
-}