Update src/streamlit_app.py

src/streamlit_app.py

@@ -195,47 +195,48 @@ class Utils:
         Download audio from a YouTube video and save it as a WAV file in a temporary directory.
         Returns the path to the saved WAV file.
         """
-        try:
-            # Get video info to use its title in the filename
-            with yt_dlp.YoutubeDL({'quiet': True}) as ydl:
-                info_dict = ydl.extract_info(youtube_url, download=False)
-                original_title = info_dict.get('title', 'audio')
-                formatted_title = Utils._format_filename(original_title)
-
-            # Create a temporary directory
-            temp_dir = tempfile.mkdtemp()
-            output_path_no_ext = os.path.join(temp_dir, formatted_title)
-
-            ydl_opts = {
-                'format': 'bestaudio/best',
-                'postprocessors': [{
-                    'key': 'FFmpegExtractAudio',
-                    'preferredcodec': 'wav',
-                    'preferredquality': '192',
-                }],
-                'outtmpl': output_path_no_ext,
-                'quiet': True
-            }
-
-            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                ydl.download([youtube_url])
-
-            # Wait for yt_dlp to actually create the WAV file
-            expected_output = output_path_no_ext + ".wav"
-            timeout = 5
-            while not os.path.exists(expected_output) and timeout > 0:
-                time.sleep(1)
-                timeout -= 1
-
-            if not os.path.exists(expected_output):
-                raise FileNotFoundError(f"Audio file was not saved as expected: {expected_output}")
-
-            st.toast(f"Audio downloaded and saved to: {expected_output}")
-            return expected_output
-
-        except Exception as e:
-            st.toast(f"Failed to download {youtube_url}: {e}")
-            return None
+        with st.spinner("Downloading and converting YouTube audio..."):
+            try:
+                # Get video info to use its title in the filename
+                with yt_dlp.YoutubeDL({'quiet': True}) as ydl:
+                    info_dict = ydl.extract_info(youtube_url, download=False)
+                    original_title = info_dict.get('title', 'audio')
+                    formatted_title = Utils._format_filename(original_title)
+    
+                # Create a temporary directory
+                temp_dir = tempfile.mkdtemp()
+                output_path_no_ext = os.path.join(temp_dir, formatted_title)
+    
+                ydl_opts = {
+                    'format': 'bestaudio/best',
+                    'postprocessors': [{
+                        'key': 'FFmpegExtractAudio',
+                        'preferredcodec': 'wav',
+                        'preferredquality': '192',
+                    }],
+                    'outtmpl': output_path_no_ext,
+                    'quiet': True
+                }
+    
+                with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                    ydl.download([youtube_url])
+    
+                # Wait for yt_dlp to actually create the WAV file
+                expected_output = output_path_no_ext + ".wav"
+                timeout = 5
+                while not os.path.exists(expected_output) and timeout > 0:
+                    time.sleep(1)
+                    timeout -= 1
+    
+                if not os.path.exists(expected_output):
+                    raise FileNotFoundError(f"Audio file was not saved as expected: {expected_output}")
+    
+                st.toast(f"Audio downloaded and saved to: {expected_output}")
+                return expected_output
+    
+            except Exception as e:
+                st.toast(f"Failed to download {youtube_url}: {e}")
+                return None
 
 class Generation:
     def __init__(