Update app.py

app.py

@@ -18,8 +18,6 @@ from gradio_log import Log
 
 # gradio_log
 log_file = 'amt/log.txt'
-with open(log_file, 'w'):
-    pass # delete
 
 # @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)"]
@@ -60,49 +58,53 @@ model = load_model_checkpoint(args=args)
 
 def prepare_media(source_path_or_url: os.PathLike,
                   source_type: Literal['audio_filepath', 'youtube_url'],
-                  delete_video: bool = True) -> Dict:
+                  delete_video: bool = True,
+                  simulate = False) -> Dict:
     """prepare media from source path or youtube, and return audio info"""
     # Get audio_file
     if source_type == 'audio_filepath':
         audio_file = source_path_or_url
     elif source_type == 'youtube_url':
-        # Download from youtube
+        # # Download from youtube
+        # try:
+        #     # Try PyTube first
+        #     yt = YouTube(source_path_or_url)
+        #     audio_stream = min(yt.streams.filter(only_audio=True), key=lambda s: s.bitrate)
+        #     mp4_file = audio_stream.download(output_path='downloaded') # ./downloaded
+        #     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:
-            # Try PyTube first
-            yt = YouTube(source_path_or_url)
-            audio_stream = min(yt.streams.filter(only_audio=True), key=lambda s: s.bitrate)
-            mp4_file = audio_stream.download(output_path='downloaded') # ./downloaded
-            audio_file = mp4_file[:-3] + 'mp3'
-            subprocess.run(['ffmpeg', '-i', mp4_file, '-ac', '1', audio_file])
-            os.remove(mp4_file)
+            with open(log_file, 'w') as lf:
+                # Try alternative
+                print(f"Failed with PyTube, error: {e}. Trying yt-dlp...")
+                audio_file = './downloaded/yt_audio'
+                # subprocess.run(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
+                #     '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
+                #     '--force-overwrites', '--cookies', 'amt/src/extras/c.txt'])
+                # subprocess.run(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
+                #     '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
+                #     '--force-overwrites', '--cookies', 'amt/src/extras/c.txt'])
+                command = ['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
+                    '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
+                    '--force-overwrites', '--username', 'oauth2', '--password', '', '-v']
+                if simulate:
+                    command = command + ['-s']
+                process = subprocess.Popen(command,
+                    stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
+            
+                for line in iter(process.stdout.readline, ''):
+                    lf.write(line)
+                    lf.flush()  
+                process.stdout.close()
+                process.wait()
+            
+            audio_file += '.mp3'
         except Exception as e:
-            try:
-                with open(log_file, 'w') as lf:
-                    # Try alternative
-                    print(f"Failed with PyTube, error: {e}. Trying yt-dlp...")
-                    audio_file = './downloaded/yt_audio'
-                    # subprocess.run(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
-                    #     '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
-                    #     '--force-overwrites', '--cookies', 'amt/src/extras/c.txt'])
-                    # subprocess.run(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
-                    #     '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
-                    #     '--force-overwrites', '--cookies', 'amt/src/extras/c.txt'])
-                    process = subprocess.Popen(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
-                        '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
-                        '--force-overwrites', '--username', 'oauth2', '--password', '',
-                        '-v'],
-                        stdout=subprocess.PIPE, stderr=subprocess.STDOUT, text=True)
-                
-                    for line in iter(process.stdout.readline, ''):
-                        lf.write(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
+            print(f"Alternative downloader failed, error: {e}. Please try again later!")
+            return None
     else:
         raise ValueError(source_type)
 
@@ -140,8 +142,11 @@ def play_video(youtube_url):
         return None
     return create_html_youtube_player(youtube_url)
 
-def display_log():
-    Log(log_file, tail=2, dark=True, xterm_font_size=12, elem_id='mylog') # gradio-log
+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)
@@ -256,9 +261,7 @@ with gr.Blocks(theme=theme, css=css) as demo:
                 transcribe_video_button.click(process_video, inputs=youtube_url, outputs=output_tab2)
                 # Play
                 play_video_button.click(play_video, inputs=youtube_url, outputs=youtube_player)
-                display_log()
-            # with gr.Column(scale=1):
-            #     # Logger
-            #     transcribe_video_button.click(display_log, inputs=None, outputs=None)
-
+            with gr.Colunmn(scale=1):
+                init_youtube_account()
+                Log(log_file, tail=2, dark=True, xterm_font_size=12, elem_id='mylog')
 demo.launch(debug=True)