Update app.py

app.py

@@ -10,6 +10,7 @@ import numpy as np
 import cv2
 import subprocess
 import tempfile
+import shutil
 
 class WanAnimateApp:
     def __init__(self):
@@ -20,22 +21,31 @@ class WanAnimateApp:
             variant="fp16",
             device_map="cpu"
         )
+        self.pipe.reset_device_map()  # Фикс: Сброс карты устройств перед offload
         self.pipe.enable_model_cpu_offload()
         self.pipe.enable_vae_slicing()
 
-    def predict(self, ref_img, video, model_id, model):
-        if ref_img is None:
-            return None, "Upload image and video."
+    def predict(
+        self, 
+        ref_img,
+        video,
+        model_id,
+        model,
+    ):
+        if ref_img is None or video is None:
+            return None, "Upload both image and video."
+
         try:
-            # Local processing
-            ref_image = Image.open(ref_img).convert("RGB").resize((576, 320))
-            motion_hint = ""
-            if video:
-                cap = cv2.VideoCapture(video)
-                frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-                cap.release()
-                motion_hint = f", dynamic motion from {frame_count} frames"
-            prompt = f"High-quality animation of the character in the image{motion_hint}, smooth movements, detailed."
+            # Local processing (no OSS/API)
+            ref_image = Image.open(ref_img).convert("RGB").resize((576, 320))  # SVD res
+            cap = cv2.VideoCapture(video)
+            frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+            cap.release()
+            motion_hint = f", dynamic motion from {frame_count} frames"
+            if model_id == "wan2.2-animate-move":
+                prompt = f"Animate the character in the reference image{motion_hint}, high quality, smooth movements."
+            else:
+                prompt = f"Replace the character in the video with the reference image{motion_hint}, seamless, detailed."
             num_frames = 25 if model == "wan-pro" else 14
             num_steps = 25 if model == "wan-pro" else 15
             generator = torch.Generator(device="cpu").manual_seed(42)
@@ -46,7 +56,7 @@ class WanAnimateApp:
                 generator=generator,
                 decode_chunk_size=2
             ).frames[0]
-            # Save with ffmpeg (no moviepy)
+            # Save with ffmpeg (stable, no moviepy)
             temp_dir = tempfile.mkdtemp()
             for i, frame in enumerate(output):
                 frame.save(f"{temp_dir}/frame_{i:04d}.png")
@@ -63,43 +73,43 @@ class WanAnimateApp:
 def start_app():
     app = WanAnimateApp()
 
-    with gr.Blocks(title="Wan2.2-Animate (No API)") as demo:
+    with gr.Blocks(title="Wan2.2-Animate (Local CPU)") as demo:
         gr.HTML("""
-            <div style="padding: 2rem; text-align: center;">
-                <h1>Wan2.2-Animate: Local Version (No API)</h1>
-                <p>Animate character from image + video. CPU: 5–15 min.</p>
+            <div style="padding: 2rem; text-align: center; max-width: 1200px; margin: 0 auto; font-family: Arial, sans-serif;">
+                <h1 style="font-size: 2.5rem; font-weight: bold; margin-bottom: 0.5rem; color: #333;">
+                    Wan2.2-Animate: Unified Character Animation and Replacement with Holistic Replication
+                </h1>
+                <h3 style="font-size: 1.5rem; font-weight: bold; margin-bottom: 0.5rem; color: #333;">
+                    Local version without API (SVD Proxy)
+                </h3>
+                <div style="font-size: 1.25rem; margin-bottom: 1.5rem; color: #555;">
+                    Tongyi Lab, Alibaba
+                </div>
+                <div style="display: flex; flex-wrap: wrap; justify-content: center; gap: 1rem; margin-bottom: 1rem;">
+                    <a href="https://arxiv.org/abs/2509.14055" target="_blank" style="display: inline-flex; align-items: center; padding: 0.5rem 1rem; background-color: #f0f0f0; color: #333; text-decoration: none; border-radius: 9999px; font-weight: 500;">
+                        <span style="margin-right: 0.5rem;">📄</span>Paper
+                    </a>
+                    <a href="https://github.com/Wan-Video/Wan2.2" target="_blank" style="display: inline-flex; align-items: center; padding: 0.5rem 1rem; background-color: #f0f0f0; color: #333; text-decoration: none; border-radius: 9999px; font-weight: 500;">
+                        <span style="margin-right: 0.5rem;">💻</span>GitHub
+                    </a>
+                    <a href="https://huggingface.co/Wan-AI/Wan2.2-Animate-14B" target="_blank" style="display: inline-flex; align-items: center; padding: 0.5rem 1rem; background-color: #f0f0f0; color: #333; text-decoration: none; border-radius: 9999px; font-weight: 500;">
+                        <span style="margin-right: 0.5rem;">🤗</span>HF Model
+                    </a>
+                    <a href="https://www.modelscope.cn/models/Wan-AI/Wan2.2-Animate-14B" target="_blank" style="display: inline-flex; align-items: center; padding: 0.5rem 1rem; background-color: #f0f0f0; color: #333; text-decoration: none; border-radius: 9999px; font-weight: 500;">
+                        <span style="margin-right: 0.5rem;">🤖</span>MS Model
+                    </a>
+                </div>
+                <div style="display: flex; flex-wrap: wrap; justify-content: center; gap: 1rem;">
+                    <a href="https://huggingface.co/spaces/Wan-AI/Wan2.2-Animate" target="_blank" style="display: inline-flex; align-items: center; padding: 0.5rem 1rem; background-color: #f0f0f0; color: #333; text-decoration: none; border-radius: 9999px; font-weight: 500;">
+                        <span style="margin-right: 0.5rem;">🤗</span>HF Space
+                    </a>
+                    <a href="https://www.modelscope.cn/studios/Wan-AI/Wan2.2-Animate" target="_blank" style="display: inline-flex; align-items: center; padding: 0.5rem 1rem; background-color: #f0f0f0; color: #333; text-decoration: none; border-radius: 9999px; font-weight: 500;">
+                        <span style="margin-right: 0.5rem;">🤖</span>MS Studio
+                    </a>
+                </div>
             </div>
         """)
+        
         gr.HTML("""
-            <details><summary>Usage</summary>
-            Move Mode: Animate image with video movements. Mix Mode: Replace video character with image.
-            Use small files (512x512 image, 2–5s video).
-            </details>
-        """)
-        with gr.Row():
-            with gr.Column():
-                ref_img = gr.Image(label="Reference Image", type="filepath", sources=["upload"])
-                video = gr.Video(label="Template Video", sources=["upload"])
-                with gr.Row():
-                    model_id = gr.Dropdown(label="Mode", choices=["wan2.2-animate-move", "wan2.2-animate-mix"], value="wan2.2-animate-move")
-                    model = gr.Dropdown(label="Quality", choices=["wan-pro", "wan-std"], value="wan-std")
-                run_button = gr.Button("Generate")
-            with gr.Column():
-                output_video = gr.Video(label="Output Video")
-                output_status = gr.Textbox(label="Status")
-        run_button.click(
-            fn=app.predict,
-            inputs=[ref_img, video, model_id, model],
-            outputs=[output_video, output_status],
-        )
-        gr.Examples(
-            examples=[["examples/1.jpg", "examples/1.mp4", "wan2.2-animate-move", "wan-std"]],
-            inputs=[ref_img, video, model_id, model],
-            outputs=[output_video, output_status],
-            fn=app.predict,
-            cache_examples=False,
-        )
-    demo.queue(concurrency_limit=1).launch(server_name="0.0.0.0", server_port=7860)
-
-if __name__ == "__main__":
-    start_app()
+            <details>
+                <summary>‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪��‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪‪