Update app.py

app.py

@@ -14,6 +14,7 @@ import torch
 import torchaudio
 import glob
 import gradio as gr
+from gradio_log import Log
 
 
 
@@ -73,19 +74,28 @@ def prepare_media(source_path_or_url: os.PathLike,
             os.remove(mp4_file)
         except Exception as e:
             try:
-                # 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'])
-                subprocess.run(['yt-dlp', '-x', source_path_or_url, '-f', 'bestaudio',
-                    '-o', audio_file, '--audio-format', 'mp3', '--restrict-filenames',
-                    '--force-overwrites', '--username', 'oauth2', '--password', '',
-                    '--cache-dir', 'amt/src/extras/cache', '-v'])
+                with open('amt/log.txt', 'w') as log_file:
+                    # 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, ''):
+                    log_file.write(line)
+                    log_file.flush()  # 파일에 내용을 즉시 쓰기
+
+                process.stdout.close()
+                process.wait()
                 audio_file += '.mp3'
             except Exception as e:
                 print(f"Alternative downloader failed, error: {e}. Please try again later!")
@@ -166,6 +176,8 @@ css = """
 
 with gr.Blocks(theme=theme, css=css) as demo:
 
+    Log(log_file, dark=True, xterm_font_size=12)
+
     with gr.Row():
         with gr.Column(scale=10):
             gr.Markdown(