Update app.py

app.py

@@ -187,8 +187,14 @@ async def download_high_quality_video(request: Request):
 async def download_high_quality_video(request: Request):
     data = await request.json()
     video_url = data.get('url')
-    # Get quality from request, default to 1080p if not specified
-    quality = data.get('quality', '1080')  # Can be '480', '720', '1080', '1440', '2160' (4K)
+    quality = data.get('quality', '1080')  # Default to 1080p if not specified
+
+    # Check if the requested quality is above 1080p
+    if int(quality) > 1080:
+        error_message = "Quality above 1080p is for premium users. Please check the URL for more information."
+        help_url = "https://chrunos.com/premium-shortcuts/"  # Replace with your actual URL
+        return {"error": error_message, "url": help_url}
+
     cookiefile = "firefox-cookies.txt"
     env_to_cookies_from_env("firefox-cookies.txt")
 
@@ -199,15 +205,12 @@ async def download_high_quality_video(request: Request):
     height_map = {
         '480': 480,
         '720': 720,
-        '1080': 1080,
-        '1440': 1440,
-        '2160': 2160
+        '1080': 1080
     }
-    max_height = height_map.get(quality, 1080)
-    if max_height > 1080:
-        format_str = f'bestvideo[height<={max_height}]+bestaudio/best' 
-    else: 
-        format_str = f'bestvideo[height<={max_height}][vcodec^=avc]+bestaudio/best'
+    max_height = height_map.get(quality, 1080)  # Use the quality variable correctly
+
+    # Determine format string based on quality
+    format_str = f'bestvideo[height<={max_height}][vcodec^=avc]+bestaudio/best'
 
     ydl_opts = {
         'format': format_str,
@@ -230,6 +233,7 @@ async def download_high_quality_video(request: Request):
     download_url = f"{BASE_URL}/file/{encoded_filename}"
     return {"url": download_url}
 
+
 @app.post("/audio")
 async def download_audio(request: Request):
     data = await request.json()