Spaces:
Runtime error
Runtime error
File size: 4,452 Bytes
53e65f3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
import os
import sys
import random
import datetime
import pandas as pd
import gradio as gr
import soundfile as sf
global samples_to_record
with open("Record Prompts - Extra.csv", "r") as f:
header = f.readline()
samples_to_record = [(index, line.strip()) for index, line in enumerate(f)]
# shuffle the samples
random.shuffle(samples_to_record)
samples_to_record = random.choices(samples_to_record, k=50)
def load_data():
with open("Record Prompts - Extra.csv", "r") as f:
global samples_to_record
header = f.readline()
samples_to_record = [(index, line.strip()) for index, line in enumerate(f)]
# shuffle the samples
random.shuffle(samples_to_record)
global index
def save_audio():
return gr.Info("Recording done, play to doublecheck or click on 'Save Audio' to save the audio", duration=5), gr.Button(visible=True)
def save_transcription(audio, file_name, text,recorder_path):
if os.path.exists(file_name):
file_name = file_name.replace('.wav', f'{random.randint(0, 1000)}.wav')
sf.write(file_name, audio[1], 16000)
text_file = file_name.replace('.wav', '.txt')
os.system(f"touch {text_file}")
os.system(f"echo '{text}' >> {text_file}")
os.system(f"echo '{file_name}\t{text}' >> {recorder_path}/recordings.txt")
return gr.Button("Save Audio",interactive=False)
def activate_button():
return gr.Button("Save Audio",interactive=True), gr.Audio(sources=["microphone"], label="Begin Presenting", type="numpy")
def clear():
return None,gr.Button(visible=False)
def clear_markdown():
return gr.Markdown(value="")
def show_button():
return gr.Button(visible=True)
def record_new(recorder_dir, transcription_index, audio_name):
with open(f"{recorder_dir}/recordings.txt", "r") as f:
done = len(f.readlines())
if done >= len(samples_to_record):
# return gr.Column(visible=False), gr.Column(visible=False), gr.Markdown(value="All recordings done!"), "", -1, ""
done = random.randint(0, len(samples_to_record)-1)
# gr.Info("Minimum recordings done! Please exit the application.", duration=15)
ind, text = samples_to_record[done]
st,err,end,instruction=text.split(',')
transcription_index=ind
audio_name = f"{recorder_dir}/{ind:05}.wav"
return gr.Markdown(value=highlight_template.format(st,err,end)),gr.Markdown(value=instruction_template.format(instruction)), \
audio_name, transcription_index, audio_name,done
def begin_session(name, gender, age, dialect):
os.makedirs(f"recordings/{name}-{gender}-{dialect}", exist_ok=True)
current_time = datetime.datetime.now()
os.system(f"echo -e 'Login on: {current_time} \nName: {name}\nGender: {gender}\nAge: {age}\nDialect: {dialect}' >> recordings/{name}-{gender}-{dialect}/info.txt")
if os.path.exists(f"recordings/{name}-{gender}-{dialect}/recordings.txt"):
return gr.Info("Welcome back! Please continue from where you stopped", duration=3), f"recordings/{name}-{gender}-{dialect}"
else:
os.system(f"touch recordings/{name}-{gender}-{dialect}/recordings.txt")
return gr.Info("Welcome! Please begin your session", duration=3), f"recordings/{name}-{gender}-{dialect}"
highlight_template = '<p style="font-family:"Traditional Arabic",font-size:150px;">{0}<span style="color:red;">{1}</span>{2}</p>'
instruction_template = '<p style="font-family:"Traditional Arabic",font-size:150px;"><mark>{0}</mark></p>'
def begin_record(recorder_dir, audio_name, transcription_index):
load_data()
with open(f"{recorder_dir}/recordings.txt", "r") as f:
done = len(f.readlines())
if done >= len(samples_to_record):
# return gr.Column(visible=False), gr.Column(visible=False), gr.Markdown(value="All recordings done!"), "", -1, ""
done = random.randint(0, len(samples_to_record)-1)
ind, text = samples_to_record[done]
st,err,end,instruction=text.split(',')
i = audio_name
audio_name = f"{recorder_dir}/{ind:05}.wav"
transcription_index=ind
return gr.Column(visible=True), gr.Column(visible=False), \
gr.Markdown(value=highlight_template.format(st,err,end)),gr.Markdown(value=instruction_template.format(instruction)), \
audio_name, transcription_index, audio_name, done
def increase(num):
if num == 15:
return 1
return num + 1 |