Update app.py

app.py

@@ -1,29 +1,52 @@
 import gradio as gr
-import requests
+import subprocess
 import tempfile
 import os
+import time
 
-LITTERBOX_API = "https://litterbox.catbox.moe/resources/internals/api.php"
+def process_video(url):
+    start = time.time()
+    
+    with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp:
+        input_file = tmp.name
+    
+    # Download highest quality (no re-encode)
+    subprocess.run([
+        "ffmpeg", "-y", "-i", url,
+        "-c", "copy", "-bsf:a", "aac_adtstoasc", input_file
+    ], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    
+    input_size = os.path.getsize(input_file) / (1024**2)
+    
+    # Watermark ONLY (copy video stream, no re-encode)
+    output_file = "watermarked_smallest.mp4"
+    filter_graph = "drawtext=text='nt-animes':x=8:y=8:fontcolor=white:fontsize=12"
+    
+    cmd = [
+        "ffmpeg", "-y", "-i", input_file,
+        "-vf", filter_graph,
+        "-c:v", "copy", "-c:a", "copy", output_file
+    ]
+    subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    
+    os.remove(input_file)
+    
+    output_size = os.path.getsize(output_file) / (1024**2)
+    total_time = time.time() - start
+    
+    return output_file, f"""
+Time: {total_time:.1f}s
+Input: {input_size:.1f} MB
+Output: {output_size:.1f} MB (smallest possible, no quality loss)
+"""
 
-def upload_litterbox(file):
-    if not file:
-        return "No file uploaded"
-    filename = os.path.basename(file.name)
-    try:
-        with open(file.name, "rb") as f:
-            files = {"fileToUpload": (filename, f)}
-            data = {"reqtype": "fileupload", "time": "72h"}
-            r = requests.post(LITTERBOX_API, data=data, files=files, timeout=300)
-        r.raise_for_status()
-        url = r.text.strip()
-        return f"Success!\nRaw MP4 link: {url}"
-    except Exception as e:
-        return f"Failed: {str(e)}"
+with gr.Blocks() as demo:
+    gr.Markdown("# Smallest Size + Zero Quality Loss")
+    url_in = gr.Textbox(value="https://hlsx3cdn.echovideo.to/naruto/1/master.m3u8", label="m3u8 URL")
+    btn = gr.Button("Process")
+    video_out = gr.Video(label="Watermarked Video")
+    info = gr.Textbox(label="Stats")
+    
+    btn.click(process_video, url_in, [video_out, info])
 
-gr.Interface(
-    fn=upload_litterbox,
-    inputs=gr.File(label="Upload Video"),
-    outputs=gr.Textbox(label="Result"),
-    title="Litterbox Upload Test",
-    description="Upload MP4 → get raw link (up to 1GB, 72h expiry)"
-).launch()
+demo.launch()