Upload 3 files

Dockerfile

@@ -0,0 +1,22 @@
+# Use an official Python runtime as a parent image
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /code
+
+# Copy the dependencies file to the working directory
+COPY ./requirements.txt /code/requirements.txt
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Copy the main application file
+COPY ./app.py /code/app.py
+
+# Expose the port that Gunicorn will run on
+# Hugging Face Spaces expects the app to run on port 7860
+EXPOSE 7860
+
+# Define the command to run the application using Gunicorn
+# This will be a robust server for your API
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]

app.py

@@ -0,0 +1,140 @@
+import os
+import re
+import threading
+import uuid
+import requests
+from flask import Flask, request, jsonify
+from urllib.parse import quote
+
+# Initialize Flask App
+app = Flask(__name__)
+
+# --- Secret Management ---
+# Get the base domain of your downloader service from Hugging Face secrets.
+# You must set a secret named 'YTDLP_DOMAIN' in your Space settings.
+# The value should be, for example, 'https://ytdlp.online'
+YTDLP_DOMAIN = os.environ.get("YTDLP_DOMAIN")
+
+# In-memory "database" to store task status
+tasks = {}
+tasks_lock = threading.Lock()
+
+def get_format_code(quality: str) -> str:
+    """Maps user-friendly quality names to yt-dlp format codes."""
+    quality_map = {
+        "480": "135+140",
+        "720": "136+140",
+        "1080": "137+140",
+        "best": "best",
+    }
+    return quality_map.get(quality, "best")
+
+def process_video_task(task_id: str, video_url: str, quality: str):
+    """
+    This function runs in a background thread. It calls the external API,
+    processes the streaming response, extracts the download URL, and updates the task status.
+    """
+    if not YTDLP_DOMAIN:
+        with tasks_lock:
+            tasks[task_id]['status'] = 'error'
+            tasks[task_id]['message'] = "YTDLP_DOMAIN secret is not set in the Space settings."
+        return
+
+    found_url = None
+    try:
+        format_code = get_format_code(quality)
+        command = f"{video_url} -f {format_code}"
+        encoded_command = quote(command)
+        
+        # The endpoint that provides the streaming response
+        stream_api_url = f"{YTDLP_DOMAIN}/stream?command={encoded_command}"
+
+        # Use stream=True to handle the server-sent event stream
+        with requests.get(stream_api_url, stream=True, timeout=300) as response:
+            response.raise_for_status()
+            
+            # Process each line of the streaming response
+            for line_bytes in response.iter_lines():
+                if line_bytes:
+                    line = line_bytes.decode('utf-8')
+                    
+                    # Use regex to find the relative download link
+                    match = re.search(r'href="(/download/download/[^"]+)"', line)
+                    if match:
+                        relative_url = match.group(1)
+                        # Construct the final, absolute URL
+                        final_stream_url = f"{YTDLP_DOMAIN}{relative_url}"
+                        found_url = final_stream_url
+                        # Once found, no need to process the rest of the stream
+                        break 
+        
+        # After checking the whole stream, update the status
+        with tasks_lock:
+            if found_url:
+                tasks[task_id]['status'] = 'completed'
+                tasks[task_id]['result'] = found_url
+            else:
+                # If the loop finished but no URL was found
+                tasks[task_id]['status'] = 'error'
+                tasks[task_id]['message'] = "Command executed, but no download link was found in the response."
+
+    except requests.exceptions.RequestException as e:
+        with tasks_lock:
+            tasks[task_id]['status'] = 'error'
+            tasks[task_id]['message'] = str(e)
+
+
+@app.route('/process', methods=['POST'])
+def start_processing():
+    """
+    Endpoint to start a new video processing task.
+    Accepts a JSON payload with 'video_url' and 'quality'.
+    Returns a task_id immediately.
+    """
+    data = request.get_json()
+    if not data or 'video_url' not in data or 'quality' not in data:
+        return jsonify({"error": "Missing 'video_url' or 'quality' in request body"}), 400
+
+    video_url = data['video_url']
+    quality = data.get('quality', 'best')
+
+    task_id = str(uuid.uuid4())
+    with tasks_lock:
+        tasks[task_id] = {'status': 'processing'}
+
+    thread = threading.Thread(target=process_video_task, args=(task_id, video_url, quality))
+    thread.daemon = True
+    thread.start()
+
+    return jsonify({'task_id': task_id}), 202
+
+@app.route('/api/<string:task_id>', methods=['GET'])
+def get_task_status(task_id: str):
+    """
+    Endpoint to check the status and result of a task.
+    Returns the task's current state. If completed, includes the final URL.
+    """
+    with tasks_lock:
+        task = tasks.get(task_id)
+
+    if task is None:
+        return jsonify({'error': 'Task not found'}), 404
+
+    # If the task completed but has no 'result' key, it's an error
+    if task['status'] == 'completed' and 'result' not in task:
+        return jsonify({
+            'status': 'error',
+            'message': 'Task completed but no result URL was generated.'
+        }), 500
+
+    return jsonify(task)
+
+@app.route('/')
+def index():
+    """A simple index route to show the API is running."""
+    if not YTDLP_DOMAIN:
+        return "<h1>Video Processing API Error</h1><p>Server is missing the YTDLP_DOMAIN secret.</p>", 500
+    return "<h1>Video Processing API is running!</h1><p>Use /process and /api/&lt;task_id&gt; endpoints.</p>"
+
+if __name__ == '__main__':
+    app.run(debug=True, host='0.0.0.0', port=int(os.environ.get("PORT", 7860)))

requirements.txt

@@ -0,0 +1,3 @@
+Flask==3.0.3
+requests==2.32.3
+gunicorn==22.0.0