Update app_4k.py

app_4k.py

@@ -21,37 +21,22 @@ import subprocess
 import os
 import uuid
 
-def generate_4k_ultrawan_video(prompt: str, seed: int = 42, out_dir: str = "/tmp/output"):
-    output_name = f"ultrawan_{uuid.uuid4().hex[:8]}"
-    output_path = os.path.join(out_dir, output_name)
-    os.makedirs(out_dir, exist_ok=True)
+import subprocess
 
+def upscale_to_4k(input_video_path, output_video_path):
+    # Use Lanczos for better quality upscale
     cmd = [
-        "python", "infer.py",
-        "--model_dir", "ultrawan_weights/Wan2.1-T2V-1.3B",
-        "--model_path", "ultrawan_weights/UltraWan/ultrawan-4k.ckpt",
-        "--mode", "lora",
-        "--lora_alpha", "0.5",
-        "--usp", "0",
-        "--height", "2160",
-        "--width", "3840",
-        "--num_frames", "33",
-        "--prompt", prompt,
-        "--seed", str(seed),
-        "--out_dir", output_path
+        "ffmpeg",
+        "-i", input_video_path,
+        "-vf", "scale=3840:2160:flags=lanczos",  # upscale to 4K (3840x2160)
+        "-c:v", "libx264",  # or libx265 for smaller size
+        "-crf", "18",  # quality: lower is better (range 0-51)
+        "-preset", "slow",  # better compression
+        "-y",  # overwrite output file
+        output_video_path,
     ]
+    subprocess.run(cmd, check=True)
 
-    env = os.environ.copy()
-    env["CUDA_VISIBLE_DEVICES"] = "0"
-
-    subprocess.run(cmd, env=env, check=True)
-
-    # Assume the output is a .mp4 inside output_path
-    generated_files = [f for f in os.listdir(output_path) if f.endswith(".mp4")]
-    if not generated_files:
-        raise FileNotFoundError("No output video found.")
-
-    return os.path.join(output_path, generated_files[0])
 
 
 
@@ -255,6 +240,8 @@ 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(video_path, os.path.join(temp_dir, "upscaled_4k.mp4"))
+
 
     return video_path, current_seed