Update app.py

app.py

@@ -355,8 +355,11 @@ def download_audio(url, cookie_file=None):
     # Check if it's a Google Drive link
     if 'drive.google.com' in url or 'https://drive.google.com' in url:
         try:
+            # 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(url, output_path, quiet=False)
+            gdown.download(download_url, output_path, quiet=False)
             if not os.path.exists(output_path):
                 return None, "Downloaded file not found", None
             sample_rate, data = wavfile.read(output_path)
@@ -364,7 +367,7 @@ def download_audio(url, cookie_file=None):
             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:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
@@ -386,7 +389,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)):