add original video display

app.py

@@ -66,19 +66,32 @@ def download_yt_url(url: str, filename: str = "%(title)s", format = 'mp4'):
     
     return output_path
 
-def wrap_html(yt_url):
-    return f"""<h1><center>Original Video: <a href='{yt_url}'>{yt_url}</a></center><h1>"""
+
+def wrap_html(filename):
+
+    yt_url = f"https://www.youtube.com/watch?v={filename}"
+
+    return f"""<h1><center>Original Video</center></h1> 
+        <center><a href='{yt_url}'>{yt_url}</a></center>
+        <br>
+        <center><iframe width="560" height="315" src="https://www.youtube.com/embed/{filename}" title="YouTube video player" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture; web-share" allowfullscreen></iframe></center>
+    """
+
+    
 
 # ideally yt_url should be validated
 def acapellify_url(yt_url, gr_request: gr.Request):
     
     # example filename: https://www.youtube.com/watch?v=TasKo5HHWb4 -> TasKo5HHWb4
-    filename = yt_url.rsplit("=", 1)[-1]
+    filename = yt_url.split("=", 1)[-1].split("=", 1)[0]
 
     video_path = download_yt_url(yt_url, filename)
 
     new_video_path, new_audio_path = process_video(video_path)
-    return new_video_path, new_audio_path, wrap_html(yt_url)
+    return new_video_path, new_audio_path, wrap_html(filename)
+
+
+
 
 
 def load_mystery_video(gr_request: gr.Request):
@@ -87,14 +100,14 @@ def load_mystery_video(gr_request: gr.Request):
 
     selected_video = video_paths[random.randrange(len(video_paths))]
 
-    file_name = selected_video.name.replace("acap_", "").split(".", 1)[0]
-    selected_audio = Path(VIDEO_DIRECTORY) / f"{file_name}.m4a"
-    yt_url = f"https://www.youtube.com/watch?v={file_name}"
+    filename = selected_video.name.replace("acap_", "").split(".", 1)[0]
+    selected_audio = Path(VIDEO_DIRECTORY) / f"{filename}.m4a"
+    
 
     selected_video = str(selected_video)
     selected_audio = str(selected_audio)
 
-    return selected_video, selected_audio, wrap_html(yt_url)
+    return selected_video, selected_audio, wrap_html(filename)
 
 
 with open("header.md", "r") as markdown_file:
@@ -104,7 +117,7 @@ with gr.Blocks(theme=theme) as demo:
 
     with gr.Row():
         header = gr.Markdown(markdown_text)
-    with gr.Row().style():
+    with gr.Row().style(equal_height=True):
         
         with gr.Column(scale=0.4, variant="panel"):
             input_url = gr.Textbox(label="Youtube URL")