Update app.py

app.py

@@ -5,47 +5,60 @@ import os
 import time
 
 def process_video(url):
-    start = time.time()
+    start_total = time.time()
     
-    with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp:
-        input_file = tmp.name
-    
-    # Download highest quality (no re-encode)
+    # Step 1: Download full episode
+    full_file = "full_episode.mp4"
+    t_download_start = time.time()
     subprocess.run([
         "ffmpeg", "-y", "-i", url,
-        "-c", "copy", "-bsf:a", "aac_adtstoasc", input_file
+        "-c", "copy", "-bsf:a", "aac_adtstoasc", full_file
     ], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    t_download = time.time() - t_download_start
+    size_full = os.path.getsize(full_file) / (1024**2)
     
-    input_size = os.path.getsize(input_file) / (1024**2)
-    
-    # Watermark ONLY (copy video stream, no re-encode)
-    output_file = "watermarked_smallest.mp4"
+    # Step 2: Watermark full episode
+    watermarked_full = "watermarked_full.mp4"
+    t_wm_start = time.time()
     filter_graph = "drawtext=text='nt-animes':x=8:y=8:fontcolor=white:fontsize=12"
-    
-    cmd = [
-        "ffmpeg", "-y", "-i", input_file,
+    subprocess.run([
+        "ffmpeg", "-y", "-i", full_file,
         "-vf", filter_graph,
-        "-c:v", "copy", "-c:a", "copy", output_file
-    ]
-    subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        "-c:v", "libx264", "-preset", "fast", "-crf", "23",
+        "-c:a", "copy", watermarked_full
+    ], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    t_wm = time.time() - t_wm_start
+    size_wm = os.path.getsize(watermarked_full) / (1024**2)
     
-    os.remove(input_file)
+    # Step 3: Trim 1s from 12:00 for quality test
+    test_clip = "test_clip_12min.mp4"
+    subprocess.run([
+        "ffmpeg", "-y", "-i", watermarked_full,
+        "-ss", "00:12:00", "-t", "00:00:01",
+        "-c", "copy", test_clip
+    ], check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
     
-    output_size = os.path.getsize(output_file) / (1024**2)
-    total_time = time.time() - start
+    # Cleanup full files
+    os.remove(full_file)
+    os.remove(watermarked_full)
     
-    return output_file, f"""
-Time: {total_time:.1f}s
-Input: {input_size:.1f} MB
-Output: {output_size:.1f} MB (smallest possible, no quality loss)
+    total_time = time.time() - start_total
+    
+    stats = f"""
+Total time: {total_time:.1f}s
+Download: {t_download:.1f}s → {size_full:.1f} MB
+Watermark full: {t_wm:.1f}s → {size_wm:.1f} MB
+Test clip: 1s from 12:00 (for quality/watermark check)
 """
+    
+    return test_clip, stats
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Smallest Size + Zero Quality Loss")
+    gr.Markdown("# Full Download + Watermark + 12min Test Clip")
     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")
+    video_out = gr.Video(label="1s Test Clip (12:00)")
+    info = gr.Textbox(label="Time & Size Stats")
     
     btn.click(process_video, url_in, [video_out, info])