Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py
+# app.py - YouTube Downloader API (No UI)
 from flask import Flask, request, send_file, jsonify
 import yt_dlp
 import os
@@ -8,82 +8,90 @@ import traceback
 
 app = Flask(__name__)
 
-DOWNLOAD_FOLDER = '/tmp/downloads'
+# Use /tmp (only writable directory on Hugging Face)
+DOWNLOAD_FOLDER = "/tmp/downloads"
 os.makedirs(DOWNLOAD_FOLDER, exist_ok=True)
 
-@app.route('/')
-def index():
+
+@app.route("/", methods=["GET"])
+def info():
     return jsonify({
-        "message": "YouTube Downloader API",
-        "usage": {
-            "endpoint": "/download",
-            "method": "POST",
-            "json_body": {
-                "url": "YouTube_URL",
-                "format": "mp3 or mp4"
-            }
+        "api": "/download",
+        "method": "POST",
+        "body": {"url": "YouTube_URL", "format": "mp3 or mp4"},
+        "example": {
+            "url": "https://www.youtube.com/watch?v=dQw4w9WgXcQ",
+            "format": "mp3"
         }
     })
 
-@app.route('/download', methods=['POST'])
-def download_video():
+
+@app.route("/download", methods=["POST"])
+def download():
     data = request.get_json()
 
-    url = data.get('url')
-    fmt = data.get('format', 'mp3').lower()
+    url = data.get("url")
+    fmt = (data.get("format") or "").lower().strip()
 
     if not url:
-        return jsonify({"error": "URL is required"}), 400
-
-    if fmt not in ['mp3', 'mp4']:
+        return jsonify({"error": "Missing 'url' in request"}), 400
+    if fmt not in ["mp3", "mp4"]:
         return jsonify({"error": "Format must be 'mp3' or 'mp4'"}), 400
 
     job_id = str(uuid.uuid4())[:8]
 
-    if fmt == 'mp3':
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'outtmpl': os.path.join(DOWNLOAD_FOLDER, f'{job_id}_%(title)s.%(ext)s'),
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
-            'quiet': True,
-            'noplaylist': True,
-        }
-    else:
-        ydl_opts = {
-            'format': 'bestvideo+bestaudio/best',
-            'outtmpl': os.path.join(DOWNLOAD_FOLDER, f'{job_id}_%(title)s.%(ext)s'),
-            'merge_output_format': 'mp4',
-            'quiet': True,
-            'noplaylist': True,
-        }
+    # Robust yt-dlp options with retries and delays
+    ydl_opts = {
+        'format': 'bestaudio/best' if fmt == 'mp3' else 'bestvideo+bestaudio/best',
+        'outtmpl': os.path.join(DOWNLOAD_FOLDER, f'{job_id}_%(title)s.%(ext)s'),
+        'merge_output_format': 'mp4' if fmt == 'mp4' else None,
+        'noplaylist': True,
+        'quiet': True,
+        'retries': 6,
+        'fragment_retries': 10,
+        'socket_timeout': 15,
+        'http_headers': {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 '
+                          '(KHTML, like Gecko) Chrome/120.0 Safari/537.36'
+        },
+        'sleep_interval': 5,
+        'max_sleep_interval': 15,
+        'source_address': None,
+        'extractor_retries': 3,
+    }
 
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             info = ydl.extract_info(url, download=True)
-            title = info.get('title', 'downloaded_media')
-            ext = 'mp3' if fmt == 'mp3' else 'mp4'
+            title = info.get("title", "media").replace("/", "_").replace("\\", "_")
+            ext = "mp3" if fmt == "mp3" else "mp4"
+
+        pattern = os.path.join(DOWNLOAD_FOLDER, f"{job_id}_*.{ext}")
+        matches = glob.glob(pattern)
 
-        search_pattern = os.path.join(DOWNLOAD_FOLDER, f"{job_id}_*.{ext}")
-        matching_files = glob.glob(search_pattern)
-        if not matching_files:
-            return jsonify({"error": "File not found after download."}), 500
+        if not matches:
+            return jsonify({"error": "Download completed but file not found."}), 500
 
-        file_path = matching_files[0]
         return send_file(
-            file_path,
+            matches[0],
             as_attachment=True,
-            download_name=f"{title}.{ext}",
-            mimetype='audio/mpeg' if fmt == 'mp3' else 'video/mp4'
+            download_name=f"{title}.{ext}"
         )
 
     except Exception as e:
         tb = ''.join(traceback.format_exception(None, e, e.__traceback__))
-        return jsonify({"error": str(e), "traceback": tb}), 500
+        return jsonify({
+            "error": str(e),
+            "details": tb.splitlines()[-1]  # Show last line of error
+        }), 500
 
 
-if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+# Optional: Health check route
+@app.route("/ping", methods=["GET"])
+def ping():
+    try:
+        import socket
+        socket.gethostbyname("www.youtube.com")
+        return jsonify({"status": "ok", "dns": "resolved"})
+    except Exception as e:
+        return jsonify({"status": "fail", "dns_error": str(e)})