Update app.py

app.py

@@ -697,7 +697,60 @@ async def download_high_quality_video(request: Request):
     return {"url": download_url, "requests_remaining": rate_limiter.max_requests - rate_limiter.get_current_count(user_ip)}
 
 
+@app.post("/high")
+async def download_high_video(request: Request): 
+    data = await request.json()  
+    video_url = data.get('url')  
+    is_youtube_url = re.search(r'(youtube\.com|youtu\.be)', video_url) is not None  
+    quality = data.get('quality', '1080')  
+    logger.info(f'input: {video_url} {quality}')  
+    cookiefile = "firefox-cookies.txt"  
+    env_to_cookies_from_env("firefox-cookies.txt")  
+
+    timestamp = datetime.now().strftime('%Y%m%d%H%M%S')  
+    output_template = str(Path(global_download_dir) / f'%(title)s_{timestamp}.%(ext)s')  
+
+    # 转换质量字符串为数值  
+    height_map = {  
+        '480': 480,  
+        '720': 720,   
+        '1080': 1080,  
+        '1440': 1440,  
+        '2160': 2160  
+    }  
+    max_height = height_map.get(quality, 1080)  
+    
+    # 为高分辨率视频(>1080p)不限制编码器，对较低分辨率优先选择H.264  
+    if max_height > 1080:  
+        # 支持横屏和竖屏视频 - 对于更高分辨率不限制编码器  
+        format_str = f'bestvideo[height<={max_height}]/bestvideo[width<={max_height}]+bestaudio/best'  
+    else:  
+        # 优先选择H.264编码，同时支持横屏和竖屏(长边不超过指定分辨率的1.8倍)  
+        long_edge = int(max_height * 1.8)  # 假设长边与短边比例约为16:9  
+        format_str = f'bestvideo[vcodec^=avc][height<={long_edge}][width<={long_edge}]+bestaudio/best'  
+
+    ydl_opts = {  
+        'format': format_str,  
+        'outtmpl': output_template,  
+        'quiet': True,  
+        'no_warnings': True,  
+        'noprogress': True,  
+        'merge_output_format': 'mp4'  
+    }
 
+    if is_youtube_url:
+        ydl_opts["cookiefile"] = "firefox-cookies.txt"
+    await run_in_threadpool(lambda: yt_dlp.YoutubeDL(ydl_opts).download([video_url]))
+    
+    downloaded_files = list(Path(global_download_dir).glob(f"*_{timestamp}.mp4"))
+    if not downloaded_files:
+        return {"error": "Download failed"}
+
+    downloaded_file = downloaded_files[0]
+    encoded_filename = urllib.parse.quote(downloaded_file.name)
+    download_url = f"{BASE_URL}/file/{encoded_filename}"
+    gc.collect()
+    return {"url": download_url}