Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
open-webui
Commits
401799c6
Commit
401799c6
authored
Feb 10, 2024
by
Timothy J. Baek
Browse files
feat: tts optimisation
parent
68ed24b7
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
73 additions
and
30 deletions
+73
-30
src/lib/components/chat/Messages/ResponseMessage.svelte
src/lib/components/chat/Messages/ResponseMessage.svelte
+56
-30
src/lib/utils/index.ts
src/lib/utils/index.ts
+17
-0
No files found.
src/lib/components/chat/Messages/ResponseMessage.svelte
View file @
401799c6
...
...
@@ -15,6 +15,7 @@
import CodeBlock from './CodeBlock.svelte';
import { synthesizeOpenAISpeech } from '$lib/apis/openai';
import { extractSentences } from '$lib/utils';
export let modelfiles = [];
export let message;
...
...
@@ -35,8 +36,10 @@
let tooltipInstance = null;
let
a
udio
Map
= {};
let
sentencesA
udio = {};
let speaking = null;
let speakingIdx = null;
let loadingSpeech = false;
$: tokens = marked.lexer(message.content);
...
...
@@ -116,44 +119,67 @@
}
};
const playAudio = (idx) => {
return new Promise((res) => {
speakingIdx = idx;
const audio = sentencesAudio[idx];
audio.play();
audio.onended = async (e) => {
await new Promise((r) => setTimeout(r, 500));
if (Object.keys(sentencesAudio).length - 1 === idx) {
speaking = null;
}
res(e);
};
});
};
const toggleSpeakMessage = async () => {
if (speaking) {
speechSynthesis.cancel();
speaking = null;
audioMap[message.id].pause();
audioMap[message.id].currentTime = 0;
sentencesAudio[speakingIdx].pause();
sentencesAudio[speakingIdx].currentTime = 0;
speaking = null;
speakingIdx = null;
} else {
speaking = true;
if ($settings?.speech?.engine === 'openai') {
loadingSpeech = true;
const res = await synthesizeOpenAISpeech(
localStorage.token,
$settings?.speech?.speaker,
message.content
).catch((error) => {
toast.error(error);
return null;
});
if (res) {
const blob = await res.blob();
const blobUrl = URL.createObjectURL(blob);
console.log(blobUrl);
loadingSpeech = false;
const audio = new Audio(blobUrl);
audioMap[message.id] = audio;
audio.onended = () => {
speaking = null;
if ($settings.conversationMode) {
document.getElementById('voice-input-button')?.click();
}
};
audio.play().catch((e) => console.error('Error playing audio:', e));
const sentences = extractSentences(message.content);
console.log(sentences);
sentencesAudio = sentences.reduce((a, e, i, arr) => {
a[i] = null;
return a;
}, {});
let lastPlayedAudioPromise = Promise.resolve(); // Initialize a promise that resolves immediately
for (const [idx, sentence] of sentences.entries()) {
const res = await synthesizeOpenAISpeech(
localStorage.token,
$settings?.speech?.speaker,
sentence
).catch((error) => {
toast.error(error);
return null;
});
if (res) {
const blob = await res.blob();
const blobUrl = URL.createObjectURL(blob);
const audio = new Audio(blobUrl);
sentencesAudio[idx] = audio;
loadingSpeech = false;
lastPlayedAudioPromise = lastPlayedAudioPromise.then(() => playAudio(idx));
}
}
} else {
let voices = [];
...
...
src/lib/utils/index.ts
View file @
401799c6
...
...
@@ -324,3 +324,20 @@ export const isValidHttpUrl = (string) => {
return
url
.
protocol
===
'
http:
'
||
url
.
protocol
===
'
https:
'
;
};
export
const
removeEmojis
=
(
str
)
=>
{
// Regular expression to match emojis
const
emojiRegex
=
/
[\u
D800-
\u
DBFF
][\u
DC00-
\u
DFFF
]
|
\u
D83C
[\u
DC00-
\u
DFFF
]
|
\u
D83D
[\u
DC00-
\u
DE4F
]
/g
;
// Replace emojis with an empty string
return
str
.
replace
(
emojiRegex
,
''
);
};
export
const
extractSentences
=
(
text
)
=>
{
// Split the paragraph into sentences based on common punctuation marks
const
sentences
=
text
.
split
(
/
(?<
=
[
.!?
])
/
);
return
sentences
.
map
((
sentence
)
=>
removeEmojis
(
sentence
.
trim
()))
.
filter
((
sentence
)
=>
sentence
!==
''
);
};
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment