change embed logic

app.py

@@ -159,15 +159,6 @@ def check_password(password):
 def run_speech_translation(audio, source_lang, target_lang, youtube_url):
     transcription, translated_text, audio_url = transcribe_and_speak(audio, source_lang, target_lang, youtube_url)
     
-    # Generate embedded video HTML if YouTube URL is provided
-    embedded_video = ""
-    if youtube_url:
-        try:
-            video_id = fetch_youtube_id(youtube_url)
-            embedded_video = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
-        except Exception as e:
-            print(f"Error embedding video: {e}")
-    
     return transcription, translated_text, audio_url, embedded_video
 
 with gr.Blocks() as demo:
@@ -213,4 +204,19 @@ with gr.Blocks() as demo:
         outputs=[user_transcription_output, user_translation_output, user_audio_output, user_video_output]
     )
 
+    def update_video_embed(youtube_url):
+        if youtube_url:
+            try:
+                video_id = fetch_youtube_id(youtube_url)
+                return f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
+            except Exception as e:
+                print(f"Error embedding video: {e}")
+        return ""
+
+    user_youtube_url.change(
+        fn=update_video_embed,
+        inputs=[user_youtube_url],
+        outputs=[user_video_output]
+    )
+
 demo.launch(auth=(os.getenv("DEV_USER"), os.getenv("DEV_PWD")))