-
-
- {selectedModel != loadedModel && progress == 0 && (
-
-
+
);
};
diff --git a/playground/src/components/gearIcon.tsx b/playground/src/components/gearIcon.tsx
new file mode 100644
index 0000000..db17e06
--- /dev/null
+++ b/playground/src/components/gearIcon.tsx
@@ -0,0 +1,141 @@
+const GearIcon = () => {
+ return (
+
+ );
+};
+
+export default GearIcon;
diff --git a/playground/src/components/languageDropdown.tsx b/playground/src/components/languageDropdown.tsx
new file mode 100644
index 0000000..ee3467f
--- /dev/null
+++ b/playground/src/components/languageDropdown.tsx
@@ -0,0 +1,189 @@
+import React, { useState } from "react";
+import { ConfigOptions } from "./configModal";
+
+const AvailableLanguages = {
+ en: "English",
+ zh: "Chinese",
+ de: "German",
+ es: "Spanish",
+ ru: "Russian",
+ ko: "Korean",
+ fr: "French",
+ ja: "Japanese",
+ pt: "Portuguese",
+ tr: "Turkish",
+ pl: "Polish",
+ ca: "Catalan",
+ nl: "Dutch",
+ ar: "Arabic",
+ sv: "Swedish",
+ it: "Italian",
+ id: "Indonesian",
+ hi: "Hindi",
+ fi: "Finnish",
+ vi: "Vietnamese",
+ he: "Hebrew",
+ uk: "Ukrainian",
+ el: "Greek",
+ ms: "Malay",
+ cs: "Czech",
+ ro: "Romanian",
+ da: "Danish",
+ hu: "Hungarian",
+ ta: "Tamil",
+ no: "Norwegian",
+ th: "Thai",
+ ur: "Urdu",
+ hr: "Croatian",
+ bg: "Bulgarian",
+ lt: "Lithuanian",
+ la: "Latin",
+ mi: "Maori",
+ ml: "Malayalam",
+ cy: "Welsh",
+ sk: "Slovak",
+ te: "Telugu",
+ fa: "Persian",
+ lv: "Latvian",
+ bn: "Bengali",
+ sr: "Serbian",
+ az: "Azerbaijani",
+ sl: "Slovenian",
+ kn: "Kannada",
+ et: "Estonian",
+ mk: "Macedonian",
+ br: "Breton",
+ eu: "Basque",
+ is: "Icelandic",
+ hy: "Armenian",
+ ne: "Nepali",
+ mn: "Mongolian",
+ bs: "Bosnian",
+ kk: "Kazakh",
+ sq: "Albanian",
+ sw: "Swahili",
+ gl: "Galician",
+ mr: "Marathi",
+ pa: "Punjabi",
+ si: "Sinhala",
+ km: "Khmer",
+ sn: "Shona",
+ yo: "Yoruba",
+ so: "Somali",
+ af: "Afrikaans",
+ oc: "Occitan",
+ ka: "Georgian",
+ be: "Belarusian",
+ tg: "Tajik",
+ sd: "Sindhi",
+ gu: "Gujarati",
+ am: "Amharic",
+ yi: "Yiddish",
+ lo: "Lao",
+ uz: "Uzbek",
+ fo: "Faroese",
+ ht: "Haitian creole",
+ ps: "Pashto",
+ tk: "Turkmen",
+ nn: "Nynorsk",
+ mt: "Maltese",
+ sa: "Sanskrit",
+ lb: "Luxembourgish",
+ my: "Myanmar",
+ bo: "Tibetan",
+ tl: "Tagalog",
+ mg: "Malagasy",
+ as: "Assamese",
+ tt: "Tatar",
+ haw: "Hawaiian",
+ ln: "Lingala",
+ ha: "Hausa",
+ ba: "Bashkir",
+ jw: "Javanese",
+ su: "Sundanese",
+ yue: "Cantonese",
+};
+
+interface LanguageDropdownProps {
+ configOptions: ConfigOptions;
+ setConfigOptions: React.Dispatch
>;
+}
+
+const LanguageDropdown = (props: LanguageDropdownProps) => {
+ const [open, setOpen] = useState(false);
+ const [selectedLanguage, setSelectedLanguage] = useState(
+ props.configOptions.language
+ );
+
+ const toggleOpen = () => setOpen((prev) => !prev);
+
+ const selectLanguage = (lang: string) => {
+ props.setConfigOptions((prev: ConfigOptions) => ({
+ ...prev,
+ language: lang,
+ }));
+ setSelectedLanguage(lang);
+ setOpen(false);
+ };
+
+ return (
+
+
+
+
+
+
+ {open && (
+
+ )}
+
{transcript &&
transcript.segments.map(
- (segment: TSSegment) => {
+ (segment: Segment) => {
return (
>;
+ raw_audio: boolean,
+ options: any
+ ): Promise
>;
public async transcribe(
audio: Uint8Array,
raw_audio: boolean,
+ options: any,
callback: (decoded: Segment) => void
): Promise>;
- public async transcribe(
+ async transcribe(
audio: Uint8Array,
raw_audio: boolean,
+ options: any,
callback?: (decoded: Segment) => void
- ): Promise> {
+ ): Promise> {
if (this.session == null) {
return Result.err(new Error("Session not initialized"));
}
if (callback) {
if (this.session instanceof Session) {
- return await this.session.stream(audio, raw_audio, callback);
+ return await this.session.stream(
+ audio,
+ raw_audio,
+ options,
+ callback
+ );
} else {
return await this.session!.stream(
audio,
raw_audio,
+ options,
Comlink.proxy(callback)
);
}
} else {
- return await this.session!.run(audio);
+ return await this.session!.run(audio, options);
}
}
diff --git a/src/session.worker.ts b/src/session.worker.ts
index 9425bd5..0a37946 100644
--- a/src/session.worker.ts
+++ b/src/session.worker.ts
@@ -62,8 +62,9 @@ export class Session {
}
public async run(
- audio: Uint8Array
- ): Promise> {
+ audio: Uint8Array,
+ options: any
+ ): Promise> {
if (!this.whisperSession) {
return Result.err(
new Error(
@@ -72,12 +73,13 @@ export class Session {
);
}
- return Result.ok(await this.whisperSession.run(audio));
+ return Result.ok(await this.whisperSession.run(audio, options));
}
public async stream(
audio: Uint8Array,
raw_audio: boolean,
+ options: any,
callback: (decoded: whisper.Segment) => void
): Promise> {
if (!this.whisperSession) {
@@ -89,7 +91,12 @@ export class Session {
}
return Result.ok(
- await this.whisperSession.stream(audio, raw_audio, callback)
+ await this.whisperSession.stream(
+ audio,
+ raw_audio,
+ options,
+ callback
+ )
);
}
}
-
+
- window.open(
- "https://github.com/FL33TW00D/whisper-turbo",
- "_blank"
- )
- }
- />
-