mod

app.py

@@ -3,7 +3,7 @@ import logging
 import boto3
 import requests
 import yt_dlp
-from flask import Flask, request, Response, stream_with_context
+from flask import Flask, request, Response
 
 logging.basicConfig(format='%(asctime)s [%(levelname)s] %(filename)s:%(lineno)d %(message)s',
                     datefmt='%Y-%m-%d %H:%M:%S', level=logging.INFO)
@@ -35,15 +35,12 @@ def p():
             for item in info.get("formats", []):
                 if item.get("format_id") == format_id:
                     url = item.get("url")
-                    response = requests.get(url, stream=True)
+                    response = requests.get(url)
                     logging.info(f"videoId: {video_id} status code: {response.status_code}")
-                    response_iter = response.iter_content(chunk_size=4096)
-                    response_iter = stream_with_context(response_iter)
-                    forwarded_response = Response(
-                        response_iter,
+                    return Response(
+                        response.content,
                         status=response.status_code,
                         headers=dict(response.headers.items()))
-                    return forwarded_response
     return {"status": 0}