feat: add Translate Model to setting page

hqwuzhaoyi · Nov 12, 2023 · ef956f9 · ef956f9
1 parent 5856b03
commit ef956f9
Show file tree

Hide file tree

Showing 10 changed files with 91 additions and 29 deletions.
diff --git a/.env.template b/.env.template
@@ -9,7 +9,7 @@ STATIC_PATH=/static            # Static file path
 LANGUAGE=zh-CN                 # Output SRT file and then translate the language
 TRANSLATE_DELAY=1500           # Delay between calling translation interface
 TRANSLATE_GROUP=4              # Translate sentences for grouping translation, how many sentences can be translated at most at a time
-TranslateModel=google          # google or gpt3
+
 
 REDIS_PORT=6379             # Redis port
 REDIS_HOST=subtitle_redis   # Redis address
@@ -30,4 +30,5 @@ AUTH_SECRET = YOUR_KEY_HERE # JWT secret you can run `openssl rand -base64 32` t
 
 # System Setting
 # You can edit in Setting
-OUTPUT_SRT_THEN_TRANSLATE=true # Whether to output the SRT file first and then translate it
+OUTPUT_SRT_THEN_TRANSLATE=true # Whether to output the SRT file first and then translate it
+TranslateModel=google          # google or gpt3
diff --git a/apps/server/src/auth/auth.service.ts b/apps/server/src/auth/auth.service.ts
@@ -106,9 +106,18 @@ export class AuthService {
     };
   }
 
-  async updateProfile(user, { username, password, OUTPUT_SRT_THEN_TRANSLATE }) {
+  async updateProfile(
+    user,
+    { username, password, OUTPUT_SRT_THEN_TRANSLATE, TranslateModel }
+  ) {
     if (typeof OUTPUT_SRT_THEN_TRANSLATE === "boolean") {
-      this.configService.set("OUTPUT_SRT_THEN_TRANSLATE", OUTPUT_SRT_THEN_TRANSLATE ? '1' : '0');
+      this.configService.set(
+        "OUTPUT_SRT_THEN_TRANSLATE",
+        OUTPUT_SRT_THEN_TRANSLATE ? "1" : "0"
+      );
+    }
+    if (TranslateModel) {
+      this.configService.set("TranslateModel", TranslateModel);
     }
 
     return this.usersService.updateProfile(user.id, {

diff --git a/apps/server/src/translate/translate.service.ts b/apps/server/src/translate/translate.service.ts
@@ -7,12 +7,14 @@ import { TranslateModel, TranslateType } from "translator";
 import { staticPath } from "utils";
 import { TranslateResult } from "shared-types";
 import { FilesService } from "@/files/files.service";
+import { CustomConfigService } from "@/config/custom-config.service";
 
 @Injectable()
 export class TranslateService {
   constructor(
     @Inject("STATIC_DIR") private staticDir: string,
-    private readonly filesService: FilesService
+    private readonly filesService: FilesService,
+    private customConfigService: CustomConfigService
   ) {}
 
   async create(createTranslateDto: CreateTranslateDto) {}
@@ -47,6 +49,15 @@ export class TranslateService {
     return translateName;
   }
 
+  private async getTranslateModel() {
+    const translateModel =
+      ((await this.customConfigService.get(
+        "TranslateModel"
+      )) as TranslateType) ?? TranslateType.GPT3;
+
+    return translateModel;
+  }
+
   translateFile(
     filename,
     dir = ""
@@ -55,7 +66,7 @@ export class TranslateService {
     filename: string;
     path: string;
   }> {
-    return new Promise((resolve, reject) => {
+    return new Promise(async (resolve, reject) => {
       const translateName = this.translateFileName(filename);
 
       // const existUrl = this.existFile(translateName, dir);
@@ -68,14 +79,13 @@ export class TranslateService {
       //   return;
       // }
 
-      const model = new TranslateModel(
-        (process.env.TranslateModel as TranslateType) ?? TranslateType.GPT3,
-        {
-          baseUrl: process.env.BASE_URL,
-          gpt3Key: process.env.OPENAI_API_KEY,
-          googleKey: process.env.GOOGLE_TRANSLATE_API_KEY,
-        }
-      )
+      const translateModel = await this.getTranslateModel();
+
+      const model = new TranslateModel(translateModel, {
+        baseUrl: process.env.BASE_URL,
+        gpt3Key: process.env.OPENAI_API_KEY,
+        googleKey: process.env.GOOGLE_TRANSLATE_API_KEY,
+      })
         .translateSrtStreamGroup(
           path.join(this.staticDir, dir, filename),
           path.join(this.staticDir, dir, translateName),
@@ -133,14 +143,13 @@ export class TranslateService {
         return;
       }
 
-      const model = new TranslateModel(
-        (process.env.TranslateModel as TranslateType) ?? TranslateType.GPT3,
-        {
-          baseUrl: process.env.BASE_URL,
-          gpt3Key: process.env.OPENAI_API_KEY,
-          googleKey: process.env.GOOGLE_TRANSLATE_API_KEY,
-        }
-      )
+      const translateModel = await this.getTranslateModel();
+
+      const model = new TranslateModel(translateModel, {
+        baseUrl: process.env.BASE_URL,
+        gpt3Key: process.env.OPENAI_API_KEY,
+        googleKey: process.env.GOOGLE_TRANSLATE_API_KEY,
+      })
         .translateSrtStreamGroup(
           subtitle.filePath,
           translatePath,

diff --git a/apps/web/app/settings/page.tsx b/apps/web/app/settings/page.tsx
@@ -19,6 +19,7 @@ export default async function SettingsProfilePage() {
           username: profile.username,
           OUTPUT_SRT_THEN_TRANSLATE:
             config.OUTPUT_SRT_THEN_TRANSLATE === "1" ? true : false,
+          TranslateModel: config.TranslateModel,
         }}
       />
     </div>

diff --git a/apps/web/app/settings/profile-form.tsx b/apps/web/app/settings/profile-form.tsx
@@ -24,11 +24,11 @@ import {
   SelectTrigger,
   SelectValue,
 } from "@/components/ui/select";
-import { Textarea } from "@/components/ui/textarea";
 import { toast } from "@/components/ui/use-toast";
 import { Switch } from "@/components/ui/switch";
 import { useSession } from "next-auth/react";
-import { redirect, useRouter } from "next/navigation";
+import { useRouter } from "next/navigation";
+import { TranslateType } from "shared-types";
 
 const profileFormSchema = z.object({
   username: z
@@ -46,6 +46,7 @@ const profileFormSchema = z.object({
     })
     .optional(),
   OUTPUT_SRT_THEN_TRANSLATE: z.boolean().optional(),
+  TranslateModel: z.nativeEnum(TranslateType).optional(),
 });
 
 type ProfileFormValues = z.infer<typeof profileFormSchema>;
@@ -148,6 +149,37 @@ export function ProfileForm({
                 </FormItem>
               )}
             />
+            <FormField
+              control={form.control}
+              name="TranslateModel"
+              render={({ field }) => (
+                <FormItem className="flex flex-col justify-between rounded-lg border p-4">
+                  <div className="space-y-0.5">
+                    <FormLabel className="text-base">Translate Model</FormLabel>
+                    <FormDescription>
+                      Choose the translation model you need
+                    </FormDescription>
+                  </div>
+                  <Select
+                    onValueChange={field.onChange}
+                    defaultValue={field.value}
+                  >
+                    <FormControl>
+                      <SelectTrigger>
+                        <SelectValue placeholder="Select a model" />
+                      </SelectTrigger>
+                    </FormControl>
+                    <SelectContent>
+                      {Object.values(TranslateType).map((item) => (
+                        <SelectItem key={item} value={item}>
+                          {item}
+                        </SelectItem>
+                      ))}
+                    </SelectContent>
+                  </Select>
+                </FormItem>
+              )}
+            />
           </div>
         </div>
         <Button type="submit">Update profile</Button>

diff --git a/apps/web/package.json b/apps/web/package.json
@@ -60,6 +60,7 @@
     "tailwind-merge": "^1.14.0",
     "tailwindcss": "3.3.3",
     "tailwindcss-animate": "^1.0.7",
+    "translator": "workspace:^",
     "typescript": "^5.2.2",
     "utils": "workspace:*",
     "zod": "^3.22.4"

diff --git a/packages/shared-types/src/index.ts b/packages/shared-types/src/index.ts
@@ -56,3 +56,8 @@ export enum LanguageEnum {
   Chinese = "cn",
   Auto = "auto",
 }
+
+export enum TranslateType {
+  GOOGLE = "google",
+  GPT3 = "gpt3",
+}
diff --git a/packages/translater/package.json b/packages/translater/package.json
@@ -32,9 +32,10 @@
     "typescript": "^5.2.2"
   },
   "dependencies": {
-    "p-queue": "npm:@esm2cjs/p-queue@^7.3.0",
     "@google-cloud/translate": "^8.0.2",
     "openai": "^4.13.0",
+    "p-queue": "npm:@esm2cjs/p-queue@^7.3.0",
+    "shared-types": "workspace:^",
     "subtitle": "^4.2.1"
   }
 }
diff --git a/packages/translater/src/index.ts b/packages/translater/src/index.ts
@@ -6,10 +6,7 @@ import { GPTTranslator } from "./gpt3";
 import { GoogleTranslator } from "./google";
 import { Translator } from "./types";
 
-export enum TranslateType {
-  GOOGLE = "google",
-  GPT3 = "gpt3",
-}
+export { TranslateType } from "shared-types";
 
 export type TranslateOptions = {
   gpt3Key?: string;

diff --git a/pnpm-lock.yaml b/pnpm-lock.yaml