Update app.py

app.py

@@ -15,9 +15,11 @@ import torchaudio
 import glob
 import gradio as gr
 from gradio_log import Log
+from pathlib import Path
 
 # gradio_log
 log_file = 'amt/log.txt'
+Path(log_file).touch()
 
 # @title Load Checkpoint
 model_name = 'YPTF.MoE+Multi (noPS)' # @param ["YMT3+", "YPTF+Single (noPS)", "YPTF+Multi (PS)", "YPTF.MoE+Multi (noPS)", "YPTF.MoE+Multi (PS)"]
@@ -74,9 +76,6 @@ def prepare_media(source_path_or_url: os.PathLike,
         #     audio_file = mp4_file[:-3] + 'mp3'
         #     subprocess.run(['ffmpeg', '-i', mp4_file, '-ac', '1', audio_file])
         #     os.remove(mp4_file)
-        # except Exception as e:
-        #     try:
-        # try:
         with open(log_file, 'w') as lf:
             # Try alternative
             # print(f"Failed with PyTube, error: {e}. Trying yt-dlp...")
@@ -98,15 +97,13 @@ def prepare_media(source_path_or_url: os.PathLike,
             for line in iter(process.stdout.readline, ''):
                 # Filter out unnecessary messages
                 if "www.google.com/device" in line:
-                    lf.write(line)
-                    lf.flush()  
+                    hl_text = line.replace("https://www.google.com/device", "\033[93mhttps://www.google.com/device\x1b[0m").split()
+                    hl_text[-1] = "\x1b[31;1m" + words[-1] + "\x1b[0m"
+                    lf.write(' '.join(hl_line)); lf.flush()    
             process.stdout.close()
             process.wait()
         
         audio_file += '.mp3'
-        # except Exception as e:
-        #     print(f"Alternative downloader failed, error: {e}. Please try again later!")
-        #     return None
     else:
         raise ValueError(source_type)
 
@@ -144,12 +141,6 @@ def play_video(youtube_url):
         return None
     return create_html_youtube_player(youtube_url)
 
-def init_youtube_account():
-    prepare_media("https://www.youtube.com/watch?v=vMboypSkj3c", 'youtube_url', simulate=True)
-
-# def display_log():
-#     Log(log_file, tail=2, dark=True, xterm_font_size=12, elem_id='mylog') # gradio-log
-
 
 AUDIO_EXAMPLES = glob.glob('examples/*.*', recursive=True)
 YOUTUBE_EXAMPLES = ["https://youtu.be/5vJBhdjvVcE?si=s3NFG_SlVju0Iklg",
@@ -262,6 +253,5 @@ with gr.Blocks(theme=theme, css=css) as demo:
                 # Play
                 play_video_button.click(play_video, inputs=youtube_url, outputs=youtube_player)
             with gr.Column(scale=1):
-                # init_youtube_account() # : dead..
                 Log(log_file, dark=True, xterm_font_size=12, elem_id='mylog')
 demo.launch(debug=True)