4k

app_two_lora.py

@@ -9,6 +9,47 @@ from huggingface_hub import hf_hub_download
 import numpy as np
 import random
 import os
+
+import os
+import tempfile
+import random
+import numpy as np
+import torch
+import gradio as gr
+import subprocess
+import shutil
+import random
+def upscale_to_4k_and_replace(input_video_path):
+    import tempfile, subprocess, shutil
+
+    # Skip if file too small or missing
+    if not os.path.exists(input_video_path) or os.path.getsize(input_video_path) < 1000:
+        raise RuntimeError("❌ Input video is missing or empty")
+
+    with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp_upscaled:
+        upscaled_path = tmp_upscaled.name
+
+    cmd = [
+        "ffmpeg",
+        "-loglevel", "error",  # Only show errors
+        "-i", input_video_path,
+        "-vf", "scale=3840:2160:flags=lanczos",
+        "-c:v", "libx264",
+        "-crf", "18",
+        "-preset", "slow",
+        "-y",
+        upscaled_path,
+    ]
+
+    result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+
+    if result.returncode != 0:
+        print("❌ FFmpeg error:\n", result.stderr.decode())
+        raise RuntimeError("FFmpeg upscale failed")
+
+    shutil.move(upscaled_path, input_video_path)
+
+    
 # LIGHT WEIGHT 1.3b
 # MODEL_ID = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
 # LORA_REPO_ID = "Kijai/WanVideo_comfy"
@@ -288,6 +329,7 @@ def generate_video(prompt, height, width,
     with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
         video_path = tmpfile.name
     export_to_video(output_frames_list, video_path, fps=FIXED_OUTPUT_FPS)
+    upscale_to_4k_and_replace(video_path)
     return video_path, current_seed