Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

add auto-transcribe option

Browse Source
This commit is contained in:
EliasVincent 2023-03-13 21:41:19 +01:00
parent 48aa52849b
commit 02e1113d95
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
extensions/whisper_stt/script.py
View File

@ -32,13 +32,23 @@ def update_hijack(val):
return val return val
def auto_transcribe(audio, audio_auto, text_state=""):
if audio is None:
return "", ""
if audio_auto:
return do_stt(audio, text_state)
return "", ""
def ui(): def ui():
tr_state = gr.State(value="") tr_state = gr.State(value="")
output_transcription = gr.Textbox(label="STT-Input", output_transcription = gr.Textbox(label="STT-Input",
placeholder="Speech Preview. Click \"Generate\" to send", placeholder="Speech Preview. Click \"Generate\" to send",
interactive=True) interactive=True)
output_transcription.change(fn=update_hijack, inputs=[output_transcription], outputs=[tr_state]) output_transcription.change(fn=update_hijack, inputs=[output_transcription], outputs=[tr_state])
audio_auto = gr.Checkbox(label="Auto-Transcribe", value=True)
with gr.Row(): with gr.Row():
audio = gr.Audio(source="microphone") audio = gr.Audio(source="microphone")
audio.change(fn=auto_transcribe, inputs=[audio, audio_auto, tr_state], outputs=[output_transcription, tr_state])
transcribe_button = gr.Button(value="Transcribe") transcribe_button = gr.Button(value="Transcribe")
transcribe_button.click(do_stt, inputs=[audio, tr_state], outputs=[output_transcription, tr_state]) transcribe_button.click(do_stt, inputs=[audio, tr_state], outputs=[output_transcription, tr_state])