embed youtube video

app.py

@@ -26,7 +26,9 @@ RAPID_API_KEY = os.getenv("RAPID_API_KEY")
 
 def fetch_youtube_id(youtube_url: str) -> str:
     if 'v=' in youtube_url:
-        return youtube_url.split("v=")[1]
+        return youtube_url.split("v=")[1].split("&")[0]
+    elif 'youtu.be/' in youtube_url:
+        return youtube_url.split("youtu.be/")[1]
     elif 'shorts' in youtube_url:
         return youtube_url.split("/")[-1]
     else:
@@ -156,7 +158,17 @@ 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)
-    return transcription, translated_text, audio_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:
     gr.Markdown("# Speech Translation")
@@ -178,6 +190,8 @@ with gr.Blocks() as demo:
         user_translation_output = gr.Textbox(label="Translation")
         user_audio_output = gr.Audio(label="Translated Speech")
     
+    user_video_output = gr.HTML(label="YouTube Video")
+
     def update_button_state(audio, youtube_url):
         print(audio, youtube_url)
         return gr.Button(interactive=bool(audio) or bool(youtube_url))
@@ -196,7 +210,7 @@ with gr.Blocks() as demo:
     user_button.click(
         fn=run_speech_translation,
         inputs=[user_audio_input, user_source_lang, user_target_lang, user_youtube_url],
-        outputs=[user_transcription_output, user_translation_output, user_audio_output]
+        outputs=[user_transcription_output, user_translation_output, user_audio_output, user_video_output]
     )
 
 demo.launch(auth=(os.getenv("DEV_USER"), os.getenv("DEV_PWD")))