update

app.py

@@ -10,13 +10,13 @@ from download_video import download_mp3_selenium
 # Function to download the audio, title, and thumbnail from YouTube
 def download_video_info(url):
     try:
-        # Call the function to download video and get title and thumbnail
-        title, thumbnail_url = download_mp3_selenium(url)
+        # Call the function to download video and get title, thumbnail, and logs
+        title, thumbnail_url, logs_output = download_mp3_selenium(url)
         audio_file = "downloaded_video.mp4"  # Path to the downloaded audio (MP4)
 
-        return audio_file, title, thumbnail_url
+        return audio_file, title, thumbnail_url, logs_output
     except Exception as e:
-        return None, None, str(e)
+        return None, None, None, str(e)
 
 # Function to transcribe the downloaded audio using Whisper
 def transcribe_audio(audio_path, model_size="base"):
@@ -24,14 +24,14 @@ def transcribe_audio(audio_path, model_size="base"):
     result = model.transcribe(audio_path)
     return result['text']
 
-# Split logic: First fetch title and thumbnail, then transcribe
+# Split logic: First fetch title, thumbnail, and logs, then transcribe
 def get_video_info_and_transcribe(youtube_url, model_size="base"):
-    # Fetch title and thumbnail first
-    audio_path, title, thumbnail_url = download_video_info(youtube_url)
+    # Fetch title, thumbnail, and logs first
+    audio_path, title, thumbnail_url, logs_output = download_video_info(youtube_url)
     
     # If fetching video info fails
     if not audio_path or not os.path.exists(audio_path):
-        return gr.update(value=f"Error fetching video: {thumbnail_url}"), None, None, None
+        return gr.update(value=f"Error fetching video: {thumbnail_url}"), None, None, gr.update(value=logs_output)
 
     # Show title and thumbnail to the user while the transcription is happening
     title_output = gr.update(value=title)
@@ -45,7 +45,7 @@ def get_video_info_and_transcribe(youtube_url, model_size="base"):
     # Start transcription
     transcription = transcribe_audio(audio_path, model_size)
 
-    return title_output, thumbnail_output, gr.update(value=transcription)
+    return title_output, thumbnail_output, gr.update(value=transcription), gr.update(value=logs_output)
 
 # Gradio interface setup using gradio.components
 with gr.Blocks() as interface:
@@ -59,7 +59,7 @@ with gr.Blocks() as interface:
         thumbnail_output = gr.Image(label="Thumbnail", interactive=False, scale=1)
         transcription_output = gr.Textbox(label="Transcription", interactive=False, scale=1)
     
-    logs_output = gr.Textbox(label="ChromeDriver Logs")
+    logs_output = gr.Textbox(label="ChromeDriver Logs", interactive=False)
 
     transcribe_button = gr.Button("Transcribe")
 

download_video.py

@@ -9,7 +9,7 @@ import requests
 def download_mp3_selenium(youtube_url):
     # Set up the Selenium WebDriver
     options = webdriver.ChromeOptions()
-    # options.add_argument("--headless")
+    options.add_argument("--headless")
     options.add_argument("--no-sandbox")
     options.add_argument('--disable-dev-shm-usage')
     options.add_argument('--disable-gpu')  # Disable GPU to ensure it runs in cloud environments
@@ -18,6 +18,7 @@ def download_mp3_selenium(youtube_url):
     options.add_argument("--window-size=1920x1080")
     options.add_argument("--start-maximized")
 
+    log_contents = ""  # Initialize log_contents
     driver = webdriver.Chrome(options=options)
 
     try: