Update app.py

app.py

@@ -331,6 +331,7 @@ import os
 import yt_dlp
 import gdown
 from scipy.io import wavfile
+from pydub import AudioSegment
 
 def download_audio(url, cookie_file=None):
     ydl_opts = {
@@ -358,25 +359,35 @@ def download_audio(url, cookie_file=None):
         try:
             # Create the 'ytdl' directory if it doesn't exist
             os.makedirs('ytdl', exist_ok=True)
-            
+
             # Extract file ID from the URL
             file_id = url.split('/d/')[1].split('/')[0]
             download_url = f'https://drive.google.com/uc?id={file_id}'
-            output_path = 'ytdl/gdrive_audio.wav'
-            gdown.download(download_url, output_path, quiet=False)
-            if not os.path.exists(output_path):
+            temp_output_path = 'ytdl/gdrive_temp_audio'  # Temporary file for raw download
+            gdown.download(download_url, temp_output_path, quiet=False)
+
+            if not os.path.exists(temp_output_path):
                 return None, "Downloaded file not found", None
+
+            # Convert the downloaded file to WAV using pydub
+            output_path = 'ytdl/gdrive_audio.wav'
+            audio = AudioSegment.from_file(temp_output_path)
+            audio.export(output_path, format="wav")
+
+            # Read the converted WAV file
             sample_rate, data = wavfile.read(output_path)
             audio_data = (sample_rate, data)
+            
+            # Clean up the temporary file
+            os.remove(temp_output_path)
+            
             return output_path, "Download successful", audio_data
         except Exception as e:
             return None, f"Google Drive download failed: {str(e)}", None
 
     # Handle YouTube link
     else:
-        # Create the 'ytdl' directory for YouTube downloads as well
         os.makedirs('ytdl', exist_ok=True)
-        
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             try:
                 info_dict = ydl.extract_info(url, download=True)
@@ -396,7 +407,7 @@ def download_audio(url, cookie_file=None):
                     return None, "Download failed: HTTP Error 403. This format requires a GVS PO Token, or the cookies are invalid. Please upload fresh cookies. See https://github.com/yt-dlp/yt-dlp/wiki/PO-Token-Guide for advanced troubleshooting.", None
                 return None, f"Download failed: {str(e)}", None
             except Exception as e:
-                return None, f"Unexpected error: {str(e)}", None
+                return None, f"Unexpected error: {str(e)}", None 
                 
 @spaces.GPU
 def roformer_separator(audio, model_key, seg_size, override_seg_size, overlap, pitch_shift, model_dir, output_dir, out_format, norm_thresh, amp_thresh, batch_size, exclude_stems="", progress=gr.Progress(track_tqdm=True)):