Update app.py

app.py

@@ -21,7 +21,15 @@ def fetch_youtube_id(youtube_url: str) -> str:
         return youtube_url.split("/")[-1]
     else:
         raise Exception("Unsupported URL format")
-         
+"""
+def fetch_youtube_id(youtube_url: str) -> str:
+    if 'v=' in youtube_url:
+        return youtube_url.split("v=")[1]
+    elif 'shorts' in youtube_url:
+        return youtube_url.split("/")[-1]
+    else:
+        raise Exception("Unsupported URL format")
+"""            
 def download_youtube_audio(youtube_url: str, output_dir: Optional[str] = None) -> str:
     video_id = fetch_youtube_id(youtube_url)
     
@@ -76,7 +84,7 @@ def run_asr(audio_file, youtube_url):
     temp_file = None
     with_timestamp=False
     model_choice = "local_whisper"
-    print ( "audio_file local_whisper ",audio_file, youtube_url )
+    
     try:
         if last_modified == 2 and youtube_url:
             # It's a YouTube URL
@@ -107,15 +115,34 @@ def run_asr(audio_file, youtube_url):
         if temp_file and os.path.exists(temp_file):
             os.remove(temp_file)
 
+def embed_youtube(youtube_url):
+    globals last_modified
+    if youtube_url:
+        try:
+            video_id = fetch_youtube_id(youtube_url)
+            embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
+            last_modified = 2;
+            return gr.update(value=embed_html, visible=True), "", None
+        except Exception as e:
+            return gr.update(value="", visible=False), f"Invalid YouTube URL: {str(e)}", None
+    return gr.update(value="", visible=False), "", None
+
+def clear_on_audio_input(audio):
+    globals last_modified
+    
+    if audio is not None:
+        last_modified =1
+        return "", gr.update(value="", visible=False), "", gr.update(interactive=True)
+    return gr.update(), gr.update(), gr.update(), gr.update(interactive=False)
 
 # Create the Gradio interface with improved aesthetics
-with gr.Blocks() as demo:
-    gr.Markdown("Audio Transcription Service")
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎙️ Audio Transcription Service")
     gr.Markdown("Upload an audio file, record your voice, or paste a YouTube URL to get an English transcription.")
     
     with gr.Row():
         with gr.Column(scale=2):
-            audio_input = gr.Audio(sources=['microphone', 'upload'], type ='filepath', label="Audio Input")
+            audio_input = gr.Audio(sources=['microphone', 'upload'], type="filepath", label="Audio Input")
             youtube_input = gr.Textbox(label="YouTube URL", placeholder="Or paste a YouTube URL here...")
             video_player = gr.HTML(visible=False)
             #timestamp_toggle = gr.Checkbox(label="Include Timestamps", value=False)
@@ -127,49 +154,29 @@ with gr.Blocks() as demo:
                 lines=10
             )
 
-    run_button = gr.Button("Transcribe Audio", variant="primary", interactive=False)
-    run_button.click(run_asr, inputs=[audio_input, youtube_input], outputs=[result])
-
-    def update_button_state_audio(audio, youtube_url):
-        global last_modified
-        print(" audio: ",audio, " youtube_url: ", youtube_url)
-        last_modified =1
-        # Button is interactive if there's input and progress is 0 or 1 (not in progress)
-        return gr.update(interactive=bool(audio) or bool(youtube_url !=""))
-        #return {run_button: gr.update(interactive=bool(audio) or bool(youtube_url != ""))}
-
-    def update_button_state_youtube(audio, youtube_url):
-        global last_modified
-        print(" youtube: ",audio, " youtube_url: ", youtube_url)
-        last_modified=2
-        # Button is interactive if there's input and progress is 0 or 1 (not in progress)
-        return gr.update(interactive=bool(audio) or bool(youtube_url !=""))
-
-    audio_input.change(
-        fn=update_button_state_audio,
-        inputs=[audio_input, youtube_input],
-        outputs=run_button
+    run_button = gr.Button("🚀 Transcribe Audio", variant="primary", interactive=False)
+    #run_button.click(run_asr, inputs=[audio_input, youtube_input], outputs=[result])
+    run_button.click(
+    fn=lambda: '',  # This lambda function sets the output to empty
+    inputs=None,
+    outputs=result
+    ).then(
+    run_asr,
+    inputs=[audio_input, youtube_input],
+    outputs=result
     )
+    # Update video player, clear transcription and audio input, and enable run button when YouTube URL is entered
     youtube_input.change(
-        fn=update_button_state_youtube,
-        inputs=[audio_input, youtube_input],
-        outputs=run_button
+        fn=lambda url: (*embed_youtube(url), gr.update(interactive=bool(url))),
+        inputs=[youtube_input],
+        outputs=[video_player, result, audio_input, run_button]
     )
 
-    def update_video_embed(youtube_url):
-        if youtube_url:
-            try:
-                video_id =  fetch_youtube_id(youtube_url)
-                embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
-                return embed_html
-            except Exception as e:
-                print(f"Error embedding video: {e}")
-        return ""
-
-    youtube_input.change(
-        fn=update_video_embed,
-        inputs=[youtube_input],
-        outputs=[video_player]
+    # Clear transcription, YouTube input, video player, and update run button when audio is input
+    audio_input.change(
+        fn=clear_on_audio_input,
+        inputs=[audio_input],
+        outputs=[result, video_player, youtube_input, run_button]
     )
 
     gr.Markdown("### How to use:")