Error Handling + Joe's Suggestion

Error Handling For Model Loading
Error Handling for audio duration and textgrid length (warning if audio>textgrid - only annotated will be transcribed, error is textgrid>audio)
Changed Dropdown Menu Option Name (Joe Emailed About This)

app.py

@@ -47,31 +47,32 @@ def load_model_and_predict(
     audio_in: str,
     model_state: dict,
 ):
-    if audio_in is None:
+    try:
+        if audio_in is None:
+            return (
+                "",
+                model_state,
+                gr.Textbox(label=TEXTGRID_NAME_INPUT_LABEL, interactive=False),
+            )
+
+        if model_state["model_name"] != model_name:
+            model_state = {
+                "loaded_model": pipeline(task="automatic-speech-recognition", model=model_name),
+                "model_name": model_name,
+            }
+
+        prediction = model_state["loaded_model"](audio_in)["text"]
         return (
-            "",
+            prediction,
             model_state,
-            gr.Textbox(label=TEXTGRID_NAME_INPUT_LABEL, interactive=False),
-        )
-
-    if model_state["model_name"] != model_name:
-        model_state = {
-            "loaded_model": pipeline(
-                task="automatic-speech-recognition", model=model_name
+            gr.Textbox(
+                label=TEXTGRID_NAME_INPUT_LABEL,
+                interactive=True,
+                value=Path(audio_in).with_suffix(".TextGrid").name,
             ),
-            "model_name": model_name,
-        }
-
-    prediction = model_state["loaded_model"](audio_in)["text"]
-    return (
-        prediction,
-        model_state,
-        gr.Textbox(
-            label=TEXTGRID_NAME_INPUT_LABEL,
-            interactive=True,
-            value=Path(audio_in).with_suffix(".TextGrid").name,
-        ),
-    )
+        )
+    except Exception as e:
+        raise gr.Error(f"Failed to load model: {str(e)}")
 
 
 def get_textgrid_contents(audio_in, textgrid_tier_name, transcription_prediction):
@@ -144,6 +145,34 @@ def extract_tier_names(textgrid_file):
         return gr.update(choices=tier_names, value=tier_names[0] if tier_names else None)
     except Exception as e:
         return gr.update(choices=[], value=None)
+    
+
+def validate_textgrid_for_intervals(audio_path, textgrid_file):
+    try:
+        if not audio_path or not textgrid_file:
+            return gr.update(interactive=False)
+
+        audio_duration = librosa.get_duration(path=audio_path)
+        tg = tgt.io.read_textgrid(textgrid_file.name)
+        tg_end_time = max(tier.end_time for tier in tg.tiers)
+
+        if tg_end_time > audio_duration:
+            raise gr.Error(
+                f"TextGrid ends at {tg_end_time:.2f}s but audio is only {audio_duration:.2f}s. "
+                "Please upload matching files."
+            )
+
+        epsilon = 0.01
+        if abs(tg_end_time - audio_duration) > epsilon:
+            gr.Warning(
+                f"TextGrid ends at {tg_end_time:.2f}s but audio is {audio_duration:.2f}s. "
+                "Only the annotated portion will be transcribed."
+            )
+
+        return gr.update(interactive=True)
+
+    except Exception as e:
+        raise gr.Error(f"Invalid TextGrid or audio file:\n{str(e)}")
 
 
 def launch_demo():
@@ -154,10 +183,6 @@ def launch_demo():
         "model_name": DEFAULT_MODEL,
     }
 
-    # Helper function - enables the interval transcribe button
-    def enable_interval_transcribe_btn(audio, textgrid):
-        return gr.update(interactive=(audio is not None and textgrid is not None))
-
     with gr.Blocks() as demo:
         gr.Markdown("""# Automatic International Phonetic Alphabet Transcription
         This demo allows you to experiment with producing phonetic transcriptions of uploaded or recorded audio using a selected automatic speech recognition (ASR) model.""")
@@ -172,7 +197,7 @@ def launch_demo():
 
         # Dropdown for transcription type selection
         transcription_type = gr.Dropdown(
-            choices=["Full Audio", "Interval"],
+            choices=["Full Audio", "TextGrid Interval"],
             label="Transcription Type",
             value=None,
             interactive=True,
@@ -209,7 +234,7 @@ def launch_demo():
         transcription_type.change(
             fn=lambda t: (
                 gr.update(visible=t == "Full Audio"),
-                gr.update(visible=t == "Interval"),
+                gr.update(visible=t == "TextGrid Interval"),
             ),
             inputs=transcription_type,
             outputs=[full_audio_section, interval_section],
@@ -248,13 +273,13 @@ def launch_demo():
 
         # Enable interval transcribe button only when both files are uploaded
         interval_audio.change(
-            fn=enable_interval_transcribe_btn,
+            fn=validate_textgrid_for_intervals,
             inputs=[interval_audio, interval_textgrid_file],
             outputs=[interval_transcribe_btn],
         )
 
         interval_textgrid_file.change(
-            fn=enable_interval_transcribe_btn,
+            fn=validate_textgrid_for_intervals,
             inputs=[interval_audio, interval_textgrid_file],
             outputs=[interval_transcribe_btn],
         )
@@ -286,4 +311,4 @@ def launch_demo():
     demo.launch(max_file_size="100mb")
 
 if __name__ == "__main__":
-    launch_demo()
+    launch_demo()