minor updates

app.py

@@ -1,5 +1,5 @@
 from gradio_client import Client
-import yt_dlp # pip install yt-dlp
+import yt_dlp  # pip install yt-dlp
 
 import subprocess
 
@@ -15,20 +15,44 @@ VIDEO_DIRECTORY = "videos"
 
 client = Client("abidlabs/music-separation")
 
+
 def acapellify(audio_path):
     result = client.predict(audio_path, api_name="/predict")
     return result[0]
 
+
 # based on https://github.com/gradio-app/gradio/blob/bebfb72b353a4280155cf7070441fc476ac10172/guides/06_client-libraries/fastapi-app-with-the-gradio-client.md
 def process_video(video_path):
     old_audio = os.path.basename(video_path).split(".")[0] + ".m4a"
-    subprocess.run(['ffmpeg', '-y', '-i', video_path, '-vn', '-acodec', 'copy', old_audio])
-    
+    subprocess.run(
+        ["ffmpeg", "-y", "-i", video_path, "-vn", "-acodec", "copy", old_audio]
+    )
+
     new_audio = acapellify(old_audio)
-    
-    new_video = f"acap_{Path(video_path).name}"
+
+    new_video = f"{Path(video_path).name}"
     new_video_path = f"{VIDEO_DIRECTORY}/{new_video}"
-    subprocess.call(['ffmpeg', '-y', '-i', video_path, '-i', new_audio, '-map', '0:v', '-map', '1:a', '-c:v', 'copy', '-c:a', 'aac', '-strict', 'experimental', new_video_path])
+    subprocess.call(
+        [
+            "ffmpeg",
+            "-y",
+            "-i",
+            video_path,
+            "-i",
+            new_audio,
+            "-map",
+            "0:v",
+            "-map",
+            "1:a",
+            "-c:v",
+            "copy",
+            "-c:a",
+            "aac",
+            "-strict",
+            "experimental",
+            new_video_path,
+        ]
+    )
 
     # remove old audio and video
     os.remove(old_audio)
@@ -39,49 +63,48 @@ def process_video(video_path):
 
     return new_video_path, new_audio_path
 
-# filename default value will return name of video on youtube
-def download_yt_url(url: str, filename: str = "%(title)s", format = 'mp4'):
 
-    output_path = f"{VIDEO_DIRECTORY}/{filename}.{format}"
+# filename default value will return name of video on youtube
+def download_yt_url(url: str, filename: str = "%(title)s", format="mp4"):
+    output_path = f"{filename}.{format}"
 
     # restrict video length so one user doesn't take up all our bandwidth
     def video_filter(info):
-        MAX_DURATION = 10*60
-        duration = info.get('duration')
+        MAX_DURATION = 10 * 60
+        duration = info.get("duration")
         if duration and duration > MAX_DURATION:
-            raise Exception(f'The video is too long at {duration}s, choose a video less than {MAX_DURATION}s')
+            raise gr.Error(
+                f"The video is too long at {duration}s, choose a video less than {MAX_DURATION}s"
+            )
 
     ydl_opts = {
-        'match_filter': video_filter,
-        'format': f'{format}/bestaudio',
-        'outtmpl': output_path,
+        "match_filter": video_filter,
+        "format": f"{format}/bestaudio",
+        "outtmpl": output_path,
     }
 
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         error_code = ydl.download([url])
-        #info = ydl.extract_info(url, extra_info={"output_path": output_path})
+        # info = ydl.extract_info(url, extra_info={"output_path": output_path})
 
     if error_code:
-        raise Exception(f"Failed to download video, error code: {error_code}")
-    
+        raise gr.Error(f"Failed to download video, error code: {error_code}")
+
     return output_path
 
 
 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>
+        <center><iframe width="100%" 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.split("=", 1)[-1].split("=", 1)[0]
 
@@ -91,18 +114,19 @@ def acapellify_url(yt_url, gr_request: gr.Request):
     return new_video_path, new_audio_path, wrap_html(filename)
 
 
-
-
-
 def load_mystery_video(gr_request: gr.Request):
-
     video_paths = list(Path(VIDEO_DIRECTORY).glob("*.mp4"))
+    n_videos = len(video_paths)
 
-    selected_video = video_paths[random.randrange(len(video_paths))]
+    if not n_videos:
+        raise gr.Error(
+            "No videos archived yet. Enter a Youtube URL to add the first video!"
+        )
 
-    filename = selected_video.name.replace("acap_", "").split(".", 1)[0]
+    selected_video = video_paths[random.randrange(n_videos)]
+
+    filename = selected_video.name.split(".", 1)[0]
     selected_audio = Path(VIDEO_DIRECTORY) / f"{filename}.m4a"
-    
 
     selected_video = str(selected_video)
     selected_audio = str(selected_audio)
@@ -113,15 +137,13 @@ def load_mystery_video(gr_request: gr.Request):
 with open("header.md", "r") as markdown_file:
     markdown_text = markdown_file.read()
 
-with gr.Blocks(theme=theme) as demo:
-
+with gr.Blocks(theme=theme, css="footer {visibility: hidden}") as demo:
     with gr.Row():
         header = gr.Markdown(markdown_text)
     with gr.Row().style(equal_height=True):
-        
         with gr.Column(scale=0.4, variant="panel"):
             input_url = gr.Textbox(label="Youtube URL")
-            
+
             process_video_btn = gr.Button("Acapellify", variant="primary")
             mystery_btn = gr.Button("Mysterious Video")
 
@@ -131,8 +153,16 @@ with gr.Blocks(theme=theme) as demo:
             output_video = gr.Video(label="Acapellified Video")
             output_audio = gr.Audio(label="Acapellified Audio")
 
-    process_video_btn.click(fn=acapellify_url, inputs=[input_url], outputs=[output_video, output_audio, static_path_display])
-    mystery_btn.click(fn=load_mystery_video, inputs=[], outputs=[output_video, output_audio, static_path_display])
+    process_video_btn.click(
+        fn=acapellify_url,
+        inputs=[input_url],
+        outputs=[output_video, output_audio, static_path_display],
+    )
+    mystery_btn.click(
+        fn=load_mystery_video,
+        inputs=[],
+        outputs=[output_video, output_audio, static_path_display],
+    )
 
 if __name__ == "__main__":
     os.makedirs(VIDEO_DIRECTORY, exist_ok=True)

header.md

@@ -1,3 +1,2 @@
 # <p style="text-align: center;">Acapellify: Extract vocals from Youtube Video URL</p>
-### <p style="text-align: center;">Based on https://gradio.app/fastapi-app-with-the-gradio-client/</p>
-### <p style="text-align: center;">Hosted on HF Spaces, backend uses gradio client linked to abidlabs/music-separation</p>
+### <p style="text-align: center;">Based on https://gradio.app/fastapi-app-with-the-gradio-client/, Hosted on HF Spaces</p>

theme.py

@@ -1,8 +1,5 @@
-
 import gradio as gr
-from gradio.themes.utils import colors, fonts, sizes
-
-from gradio.themes import Soft, Base
+from gradio.themes.utils import colors
 
 
 yfu_red = colors.Color(
@@ -23,11 +20,11 @@ yfu_red = colors.Color(
 white = colors.Color(
     name="white",
     c50="#ff2c65",
-    c100="#ff2c65", # text color
+    c100="#ff2c65",  # text color
     c200="#ff2c65",
     c300="#ff2c65",
     c400="#ff2c65",
-    c500="#ff2c65", # secondary text color
+    c500="#ff2c65",  # secondary text color
     c600="#ffffff",
     c700="#ffffff",
     c800="#ffffff",
@@ -42,32 +39,25 @@ theme = gr.themes.Soft(
 ).set(
     body_background_fill="white",
     panel_background_fill="white",
-
     background_fill_primary="white",
     background_fill_primary_dark="white",
-
     panel_background_fill_dark="white",
     panel_border_color="*primary_50",
     panel_border_color_dark="*primary_50",
-
     body_text_color="*primary_50",
     body_text_color_dark="*primary_50",
     body_text_color_subdued="*primary_50",
     body_text_color_subdued_dark="*primary_50",
-
     table_even_background_fill="white",
     table_even_background_fill_dark="white",
     table_odd_background_fill="*primary_100",
     table_odd_background_fill_dark="*primary_100",
     table_row_focus="*primary_200",
     table_row_focus_dark="*primary_200",
-
     input_background_fill_focus="*primary_200",
     input_background_fill_focus_dark="*primary_200",
-
     checkbox_background_color_focus="*primary_200",
     checkbox_background_color_focus_dark="*primary_200",
-
     border_color_primary_dark="*primary_50",
     checkbox_border_color_dark="*primary_50",
     table_border_color_dark="*primary_50",
@@ -79,13 +69,8 @@ theme = gr.themes.Soft(
     checkbox_label_text_color_selected="*primary_50",
     button_cancel_text_color="white",
     block_border_color="*primary_50",
-
-    #button_secondary_background_fill=None,
-    #button_secondary_background_fill_dark=None,
     button_secondary_background_fill_hover="*primary_200",
     button_secondary_background_fill_hover_dark="*primary_200",
-
     stat_background_fill="*primary_200",
     stat_background_fill_dark="*primary_200",
-
 )