Update app.py

app.py

@@ -351,8 +351,8 @@ def download_audio(url, cookie_file=None):
         return download_from_youtube(url, cookie_path)
 
 def download_from_youtube(url, cookie_path):
-    # Configuration for video+audio download (default, as audio-only often fails for Shorts)
-    ydl_opts = {
+    # Configuration for video+audio download (first attempt)
+    ydl_opts_video = {
         'format': 'bestvideo+bestaudio/best',
         'postprocessors': [{
             'key': 'FFmpegExtractAudio',
@@ -370,12 +370,12 @@ def download_from_youtube(url, cookie_path):
         'ignoreerrors': False,
         'no_check_certificate': True,
         'verbose': True,
-        'merge_output_format': 'mp4',  # Ensure video+audio merge
+        'merge_output_format': 'mp4',
     }
     
     try:
         logger.info("Attempting video+audio download to extract audio")
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        with yt_dlp.YoutubeDL(ydl_opts_video) as ydl:
             info_dict = ydl.extract_info(url, download=True)
             file_path = ydl.prepare_filename(info_dict).rsplit('.', 1)[0] + '.wav'
             
@@ -386,16 +386,54 @@ def download_from_youtube(url, cookie_path):
             return file_path, "YouTube video+audio download and audio extraction successful", (sample_rate, data)
     
     except yt_dlp.utils.DownloadError as e:
-        if "Sign in to confirm you’re not a bot" in str(e):
-            return None, "Authentication error. Please upload valid YouTube cookies: https://github.com/yt-dlp/yt-dlp/wiki/Extractors#exporting-youtube-cookies", None
         if "Requested format is not available" in str(e):
-            return None, "Requested format not available. Try checking available formats with yt-dlp --list-formats", None
-        return None, f"YouTube download error: {str(e)}", None
+            logger.warning("Video+audio format unavailable, trying audio-only fallback")
+            # Audio-only configuration (fallback)
+            ydl_opts_audio = {
+                'format': 'bestaudio/best',
+                'postprocessors': [{
+                    'key': 'FFmpegExtractAudio',
+                    'preferredcodec': 'wav',
+                    'preferredquality': '192',
+                }],
+                'outtmpl': 'ytdl/%(title)s.%(ext)s',
+                'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/95.0.4638.54 Safari/537.36',
+                'geo_bypass': True,
+                'force_ipv4': True,
+                'referer': 'https://www.youtube.com/',
+                'noplaylist': True,
+                'cookiefile': cookie_path,
+                'extractor_retries': 5,
+                'ignoreerrors': False,
+                'no_check_certificate': True,
+                'verbose': True,
+            }
+            
+            try:
+                logger.info("Attempting audio-only download")
+                with yt_dlp.YoutubeDL(ydl_opts_audio) as ydl:
+                    info_dict = ydl.extract_info(url, download=True)
+                    file_path = ydl.prepare_filename(info_dict).rsplit('.', 1)[0] + '.wav'
+                    
+                    if not os.path.exists(file_path):
+                        return None, "Downloaded audio file not found after audio-only processing", None
+                    
+                    sample_rate, data = scipy.io.wavfile.read(file_path)
+                    return file_path, "YouTube audio-only download successful", (sample_rate, data)
+            
+            except Exception as fallback_e:
+                return None, f"Audio-only fallback failed: {str(fallback_e)}", None
+                
+        elif "Sign in to confirm you’re not a bot" in str(e):
+            return None, "Authentication error. Please upload valid YouTube cookies: https://github.com/yt-dlp/yt-dlp/wiki/Extractors#exporting-youtube-cookies", None
+        else:
+            return None, f"YouTube download error: {str(e)}", None
+            
     except yt_dlp.utils.GeoRestrictedError:
         return None, "Video is geo-restricted in your region", None
     except Exception as e:
         return None, f"Unexpected error during YouTube download: {str(e)}", None
-
+        
 def download_from_google_drive(url):
     temp_output_path = 'ytdl/gdrive_temp_audio'
     output_path = 'ytdl/gdrive_audio.wav'