Update app.py

app.py

@@ -1,6 +1,7 @@
 import sys
 import time
-from fastapi import FastAPI, BackgroundTasks, Request, HTTPException
+from fastapi import FastAPI, BackgroundTasks, Request, HTTPException, Security
+from fastapi.security import APIKeyHeader
 from fastapi.responses import FileResponse
 from fastapi.concurrency import run_in_threadpool
 import yt_dlp
@@ -16,12 +17,6 @@ import json
 from dotenv import load_dotenv
 import mimetypes
 import tempfile
-from mutagen.mp4 import MP4, MP4Cover, MP4FreeForm
-from mutagen.mp3 import MP3
-from mutagen.id3 import ID3, USLT, SYLT, Encoding, APIC, TIT2, TPE1, TALB, TPE2, TDRC, TCON, TRCK, COMM
-from mutagen.oggvorbis import OggVorbis
-from mutagen.oggopus import OggOpus
-from mutagen.flac import FLAC, Picture
 from PIL import Image
 from io import BytesIO
 from pathlib import Path
@@ -32,23 +27,9 @@ import logging
 import gc
 
 tmp_dir = tempfile.gettempdir()
-BASE_URL = "https://chrunos-multi.hf.space"
-
-MP4_TAGS_MAP = {
-    "album": "\xa9alb",
-    "album_artist": "aART",
-    "artist": "\xa9ART",
-    "composer": "\xa9wrt",
-    "copyright": "cprt",
-    "lyrics": "\xa9lyr",
-    "comment": "desc",
-    "media_type": "stik",
-    "producer": "\xa9prd",
-    "rating": "rtng",
-    "release_date": "\xa9day",
-    "title": "\xa9nam",
-    "url": "\xa9url",
-}
+BASE_URL = "https://chrunos-zam.hf.space"
+
+
 def env_to_cookies(env_content: str, output_file: str) -> None:
     """Convert environment variable content back to cookie file"""
     try:
@@ -115,12 +96,6 @@ def create_temp_cookie_file():
 load_dotenv()
 app = FastAPI()
 
-ydl_opts = {
-    'format': 'best',
-    'quiet': True,
-    #'outtmpl': f'{VIDEO_DIR}/%(id)s.%(ext)s',
-    'max_filesize': 50 * 1024 * 1024
-}
 
 
 @app.get('/')
@@ -230,60 +205,25 @@ async def download_high_quality_video(request: Request):
 
 
 
-@app.post("/max")
-async def download_high_quality_video(request: Request):
-    data = await request.json()
-    video_url = data.get('url')
-    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")
-
-    timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
-    output_template = str(Path(global_download_dir) / f'%(title)s_{timestamp}.%(ext)s')
 
-    # Convert quality string to height
-    height_map = {
-        '480': 480,
-        '720': 720,
-        '1080': 1080
-    }
-    max_height = height_map.get(quality, 1080)  # Use the quality variable correctly
+api_key_header = APIKeyHeader(name="X-API-Key")
 
-    # Determine format string based on quality
-    format_str = f'bestvideo[height<={max_height}][vcodec^=avc]+bestaudio/best'
-
-    ydl_opts = {
-        'format': format_str,
-        'outtmpl': output_template,
-        'quiet': True,
-        'no_warnings': True,
-        'noprogress': True,
-        'merge_output_format': 'mp4',
-        'cookiefile': cookiefile
-    }
-
-    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}
+# Store this securely in your environment variables
+API_KEY = os.getenv("API_KEY")
 
+async def verify_api_key(api_key: str = Security(api_key_header)):
+    if api_key != API_KEY:
+        raise HTTPException(
+            status_code=403,
+            detail="Invalid API key"
+        )
+    return api_key
 
 @app.post("/audio")
-async def download_audio(request: Request):
+async def download_audio(
+    request: Request,
+    api_key: str = Security(verify_api_key)
+):
     data = await request.json()
     video_url = data.get('url')
     cookiefile = "firefox-cookies.txt"