Schmieder, Jacob
154 lines
5.2 KiB
Python
154 lines
5.2 KiB
Python
"""
|
|
This file contains ervery function that will be called when the user interacts with the
|
|
UI like pressing a button or uploading a file.
|
|
"""
|
|
|
|
import gradio as gr
|
|
import scraibe.app.global_var as gv
|
|
from scraibe import Transcript
|
|
from .multi import start_model_worker
|
|
|
|
def select_task(choice):
|
|
# tell the app that it is still in use
|
|
if choice == 'Auto Transcribe':
|
|
|
|
return (gr.update(visible = True),
|
|
gr.update(visible = True),
|
|
gr.update(visible = True))
|
|
|
|
|
|
elif choice == 'Transcribe':
|
|
|
|
return (gr.update(visible = False),
|
|
gr.update(visible = True),
|
|
gr.update(visible = True))
|
|
|
|
|
|
elif choice == 'Diarisation':
|
|
|
|
return (gr.update(visible = True),
|
|
gr.update(visible = False),
|
|
gr.update(visible = False))
|
|
|
|
def select_origin(choice):
|
|
|
|
# tell the app that it is still in use
|
|
if choice == "Upload Audio":
|
|
|
|
return (gr.update(visible = True),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None))
|
|
|
|
elif choice == "Record Audio":
|
|
|
|
return (gr.update(visible = False, value = None),
|
|
gr.update(visible = True),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None))
|
|
|
|
elif choice == "Upload Video":
|
|
|
|
return (gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = True),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None))
|
|
|
|
elif choice == "Record Video":
|
|
|
|
return (gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = True),
|
|
gr.update(visible = False, value = None))
|
|
|
|
elif choice == "File or Files":
|
|
|
|
return (gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = False, value = None),
|
|
gr.update(visible = True))
|
|
|
|
def run_scraibe(task,
|
|
num_speakers,
|
|
translate,
|
|
language,
|
|
audio1,
|
|
audio2,
|
|
video1,
|
|
video2,
|
|
file_in,
|
|
progress = gr.Progress(track_tqdm=False)):
|
|
|
|
# get *args which are not None
|
|
if gv.MODEL_PROCESS is None or not gv.MODEL_PROCESS.is_alive():
|
|
#progress(0.0, desc='Loading model...')
|
|
gv.MODEL_PROCESS = start_model_worker(gv.MODEL_PARAMS,
|
|
gv.REQUEST_QUEUE,
|
|
gv.LAST_ACTIVE_TIME,
|
|
gv.RESPONSE_QUEUE,
|
|
gv.LOADED_EVENT,
|
|
gv.RUNNING_EVENT)
|
|
|
|
# progress(0.1, desc='Starting task...')
|
|
source = audio1 or audio2 or video1 or video2 or file_in
|
|
|
|
if isinstance(source, list):
|
|
source = [s.name for s in source]
|
|
if len(source) == 1:
|
|
source = source[0]
|
|
|
|
config = dict(source = source,
|
|
task = task,
|
|
num_speakers = num_speakers,
|
|
translate = translate,
|
|
language = language)
|
|
|
|
gv.REQUEST_QUEUE.put(config)
|
|
|
|
if task == 'Auto Transcribe':
|
|
|
|
out_str , out_json = gv.RESPONSE_QUEUE.get()
|
|
|
|
if isinstance(source, str):
|
|
return (gr.update(value = out_str, visible = True),
|
|
gr.update(value = out_json, visible = True),
|
|
gr.update(visible = True),
|
|
gr.update(visible = True))
|
|
else:
|
|
return (gr.update(value = out_str, visible = True),
|
|
gr.update(value = out_json, visible = True),
|
|
gr.update(visible = False),
|
|
gr.update(visible = False))
|
|
|
|
elif task == 'Transcribe':
|
|
|
|
out = gv.RESPONSE_QUEUE.get()
|
|
|
|
return (gr.update(value = out, visible = True),
|
|
gr.update(value = None, visible = False),
|
|
gr.update(visible = False),
|
|
gr.update(visible = False))
|
|
|
|
elif task == 'Diarisation':
|
|
|
|
out = gv.RESPONSE_QUEUE.get()
|
|
|
|
return (gr.update(value = None, visible = False),
|
|
gr.update(value = out, visible = True),
|
|
gr.update(visible = False),
|
|
gr.update(visible = False))
|
|
|
|
def annotate_output(annoation : str, out_json : dict):
|
|
# get *args which are not None
|
|
|
|
trans = Transcript.from_json(out_json)
|
|
trans = trans.annotate(*annoation.split(","))
|
|
|
|
return gr.update(value = str(trans)),gr.update(value = trans.get_json())
|
|
|