fixes

Signed-off-by: Piotr Żelasko <[email protected]>

app.py

@@ -74,11 +74,16 @@ def postprocess(transcript, prompt):
     ans = ans.split("<|im_start|>assistant")[-1]  # get rid of the prompt
     if "<think>" in ans:
         ans = ans.split("<think>")[-1]
-        thoughts, ans = ans.split("</think>")[-1]  # get rid of the thinking
+        thoughts, ans = ans.split("</think>")  # get rid of the thinking
     else:
         thoughts = ""
     return ans.strip(), thoughts
 
+def disable_buttons():
+    return gr.update(interactive=False), gr.update(interactive=False)
+
+def enable_buttons():
+    return gr.update(interactive=True), gr.update(interactive=True)
 
 with gr.Blocks(
     title="NeMo Canary-Qwen-2.5B Model",
@@ -162,15 +167,29 @@ with gr.Blocks(
         )
 
     asr_button.click(
+        disable_buttons,
+        outputs=[asr_button, llm_button],
+        trigger_mode="once",
+    ).then(
         fn=transcribe,
         inputs=[audio_file],
         outputs=[transcript_box, raw_transcript]
+    ).then(
+        enable_buttons,
+        outputs=[asr_button, llm_button],
     )
 
     llm_button.click(
+        disable_buttons,
+        outputs=[asr_button, llm_button],
+        trigger_mode="once",
+    ).then(
         fn=postprocess,
         inputs=[raw_transcript, prompt_box],
         outputs=[magic_box, think_box]
+    ).then(
+        enable_buttons,
+        outputs=[asr_button, llm_button],
     )