mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

interactive preview window

Browse source
This commit is contained in:
EliasVincent 2023-03-09 21:03:49 +01:00
parent 7a03d0bda3
commit 00359ba054
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
extensions/whisper_stt/script.py
View file

@ -1,17 +1,12 @@
import gradio as gr import gradio as gr
import speech_recognition as sr import speech_recognition as sr
input_hijack = { input_hijack = {
'state': False, 'state': False,
'value': ["", ""] 'value': ["", ""]
} }
def input_modifier(string):
return string
def do_stt(): def do_stt():
transcription = "" transcription = ""
r = sr.Recognizer() r = sr.Recognizer()
@ -30,7 +25,13 @@ def do_stt():
return transcription return transcription
def update_hijack(val):
input_hijack.update({"state": True, "value": [val, val]})
return val
def ui(): def ui():
speech_button = gr.Button(value="🎙️") speech_button = gr.Button(value="🎙️")
output_transcription = gr.Textbox(label="STT-Preview", placeholder="Speech Preview. Click \"Generate\" to send") output_transcription = gr.Textbox(label="STT-Input", placeholder="Speech Preview. Click \"Generate\" to send", interactive=True)
output_transcription.change(fn=update_hijack, inputs=[output_transcription])
speech_button.click(do_stt, outputs=[output_transcription]) speech_button.click(do_stt, outputs=[output_transcription])