Update app.py

app.py

@@ -77,15 +77,10 @@ def prepare_media(source_path_or_url: os.PathLike,
         #     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']
@@ -98,8 +93,8 @@ def prepare_media(source_path_or_url: os.PathLike,
                 # Filter out unnecessary messages
                 if "www.google.com/device" in line:
                     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()    
+                    hl_text[-1] = "\x1b[31;1m" + hl_text[-1] + "\x1b[0m"
+                    lf.write(' '.join(hl_text)); lf.flush()    
             process.stdout.close()
             process.wait()
         
@@ -141,6 +136,9 @@ def play_video(youtube_url):
         return None
     return create_html_youtube_player(youtube_url)
 
+def oauth_google():
+    return create_html_oauth()
+
 
 AUDIO_EXAMPLES = glob.glob('examples/*.*', recursive=True)
 YOUTUBE_EXAMPLES = ["https://youtu.be/5vJBhdjvVcE?si=s3NFG_SlVju0Iklg",
@@ -243,8 +241,10 @@ with gr.Blocks(theme=theme, css=css) as demo:
                 youtube_player = gr.HTML(render=True)
 
             with gr.Column(scale=4):
-                # Submit button
-                transcribe_video_button = gr.Button("Transcribe", variant="primary")
+                with gr.Row():
+                    # Submit button
+                    transcribe_video_button = gr.Button("Transcribe", variant="primary")
+                    oauth_button = gr.Button("google.com/device", variant="primary")
             with gr.Column(scale=1):
                 # Transcribe
                 output_tab2 = gr.HTML(render=True)
@@ -252,6 +252,8 @@ 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)
+                # Authetification
+                oauth_button.click(oauth_google, outputs=youtube_player)
             with gr.Column(scale=1):
                 Log(log_file, dark=True, xterm_font_size=12, elem_id='mylog')
 demo.launch(debug=True)