Spaces:
Build error
Build error
Enhance language selection and error handling in transcribe_audio function
Browse files
app.py
CHANGED
|
@@ -12,6 +12,7 @@ os.makedirs(MODEL_DIR, exist_ok=True)
|
|
| 12 |
language_options = [(f"{code}: {name[0]}", code)
|
| 13 |
for code, name in LANGUAGE_CODES.items()]
|
| 14 |
language_options.sort(key=lambda x: x[0])
|
|
|
|
| 15 |
|
| 16 |
MODELS = {
|
| 17 |
"base (140M)": "base",
|
|
@@ -69,16 +70,22 @@ def ensure_model_downloaded(model_key):
|
|
| 69 |
|
| 70 |
|
| 71 |
def update_regions(language):
|
| 72 |
-
if
|
|
|
|
|
|
|
|
|
|
| 73 |
regions = language_to_regions[language]
|
| 74 |
regions.sort(key=lambda x: x[0])
|
| 75 |
-
|
|
|
|
| 76 |
return [], None, False
|
| 77 |
|
| 78 |
|
| 79 |
@spaces.GPU
|
| 80 |
def transcribe_audio(audio_file, model_name, language, region, predict_timestamps, padding_speech):
|
| 81 |
try:
|
|
|
|
|
|
|
| 82 |
ensure_assets_downloaded()
|
| 83 |
|
| 84 |
model_key = MODELS[model_name]
|
|
@@ -108,6 +115,7 @@ def transcribe_audio(audio_file, model_name, language, region, predict_timestamp
|
|
| 108 |
f", region: {result.region}" if result.region else "")
|
| 109 |
return output_text, detected_info
|
| 110 |
except Exception as e:
|
|
|
|
| 111 |
return f"Error: {str(e)}", "Transcription failed"
|
| 112 |
|
| 113 |
|
|
@@ -128,21 +136,21 @@ with gr.Blocks(title="Dolphin Speech Recognition") as demo:
|
|
| 128 |
with gr.Row():
|
| 129 |
model_dropdown = gr.Dropdown(
|
| 130 |
choices=list(MODELS.keys()),
|
| 131 |
-
value=list(MODELS.keys())[
|
| 132 |
label="Model Size"
|
| 133 |
)
|
| 134 |
|
| 135 |
with gr.Row():
|
| 136 |
language_dropdown = gr.Dropdown(
|
| 137 |
choices=language_options,
|
| 138 |
-
value=
|
| 139 |
-
label="Language
|
| 140 |
-
info="
|
| 141 |
)
|
| 142 |
region_dropdown = gr.Dropdown(
|
| 143 |
choices=[],
|
| 144 |
value=None,
|
| 145 |
-
label="Region
|
| 146 |
visible=False
|
| 147 |
)
|
| 148 |
|
|
@@ -162,10 +170,17 @@ with gr.Blocks(title="Dolphin Speech Recognition") as demo:
|
|
| 162 |
output_text = gr.Textbox(label="Transcription", lines=10)
|
| 163 |
language_info = gr.Textbox(label="Detected Language", lines=1)
|
| 164 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 165 |
language_dropdown.change(
|
| 166 |
-
fn=
|
| 167 |
inputs=[language_dropdown],
|
| 168 |
-
outputs=[region_dropdown
|
| 169 |
)
|
| 170 |
|
| 171 |
transcribe_button.click(
|
|
|
|
| 12 |
language_options = [(f"{code}: {name[0]}", code)
|
| 13 |
for code, name in LANGUAGE_CODES.items()]
|
| 14 |
language_options.sort(key=lambda x: x[0])
|
| 15 |
+
language_options = [("Auto-detect", None)] + language_options
|
| 16 |
|
| 17 |
MODELS = {
|
| 18 |
"base (140M)": "base",
|
|
|
|
| 70 |
|
| 71 |
|
| 72 |
def update_regions(language):
|
| 73 |
+
if not language:
|
| 74 |
+
return [], None, False
|
| 75 |
+
|
| 76 |
+
if language in language_to_regions:
|
| 77 |
regions = language_to_regions[language]
|
| 78 |
regions.sort(key=lambda x: x[0])
|
| 79 |
+
default_value = regions[0][1] if regions else None
|
| 80 |
+
return regions, default_value, True
|
| 81 |
return [], None, False
|
| 82 |
|
| 83 |
|
| 84 |
@spaces.GPU
|
| 85 |
def transcribe_audio(audio_file, model_name, language, region, predict_timestamps, padding_speech):
|
| 86 |
try:
|
| 87 |
+
if not audio_file:
|
| 88 |
+
return "Please upload or record audio first", ""
|
| 89 |
ensure_assets_downloaded()
|
| 90 |
|
| 91 |
model_key = MODELS[model_name]
|
|
|
|
| 115 |
f", region: {result.region}" if result.region else "")
|
| 116 |
return output_text, detected_info
|
| 117 |
except Exception as e:
|
| 118 |
+
print(f"Error in transcribe_audio: {str(e)}")
|
| 119 |
return f"Error: {str(e)}", "Transcription failed"
|
| 120 |
|
| 121 |
|
|
|
|
| 136 |
with gr.Row():
|
| 137 |
model_dropdown = gr.Dropdown(
|
| 138 |
choices=list(MODELS.keys()),
|
| 139 |
+
value=list(MODELS.keys())[0],
|
| 140 |
label="Model Size"
|
| 141 |
)
|
| 142 |
|
| 143 |
with gr.Row():
|
| 144 |
language_dropdown = gr.Dropdown(
|
| 145 |
choices=language_options,
|
| 146 |
+
value=language_options[0][1],
|
| 147 |
+
label="Language",
|
| 148 |
+
info="Default is auto-detect"
|
| 149 |
)
|
| 150 |
region_dropdown = gr.Dropdown(
|
| 151 |
choices=[],
|
| 152 |
value=None,
|
| 153 |
+
label="Region",
|
| 154 |
visible=False
|
| 155 |
)
|
| 156 |
|
|
|
|
| 170 |
output_text = gr.Textbox(label="Transcription", lines=10)
|
| 171 |
language_info = gr.Textbox(label="Detected Language", lines=1)
|
| 172 |
|
| 173 |
+
def on_language_change(language):
|
| 174 |
+
regions, default_value, is_visible = update_regions(language)
|
| 175 |
+
return {
|
| 176 |
+
region_dropdown: gr.update(
|
| 177 |
+
choices=regions, value=default_value, visible=is_visible)
|
| 178 |
+
}
|
| 179 |
+
|
| 180 |
language_dropdown.change(
|
| 181 |
+
fn=on_language_change,
|
| 182 |
inputs=[language_dropdown],
|
| 183 |
+
outputs=[region_dropdown]
|
| 184 |
)
|
| 185 |
|
| 186 |
transcribe_button.click(
|