Update app.py

app.py

@@ -12,7 +12,7 @@ import cv2
 import subprocess
 import tempfile
 
-# Папки для файлов (создаются автоматически)
+# Папки (создаются автоматически)
 IMG_DIR = './img'
 VIDEO_DIR = './video'
 OUTPUT_DIR = './output'
@@ -32,11 +32,15 @@ class WanAnimateApp:
 
     def get_img_files(self):
         """Список файлов в папке img"""
-        return [f for f in os.listdir(IMG_DIR) if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+        if os.path.exists(IMG_DIR):
+            return [f for f in os.listdir(IMG_DIR) if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+        return []
 
     def get_video_files(self):
         """Список файлов в папке video"""
-        return [f for f in os.listdir(VIDEO_DIR) if f.lower().endswith(('.mp4', '.avi', '.mov'))]
+        if os.path.exists(VIDEO_DIR):
+            return [f for f in os.listdir(VIDEO_DIR) if f.lower().endswith(('.mp4', '.avi', '.mov'))]
+        return []
 
     def load_img_from_dropdown(self, selected_file):
         """Загружает выбранное фото в ref_img"""
@@ -62,7 +66,7 @@ class WanAnimateApp:
         model,
     ):
         if ref_img_path is None or video_path is None:
-            return None, "Выберите или загрузите оба файла."
+            return None, "Выберите файл из списка или загрузите новый."
 
         try:
             # Local processing
@@ -139,28 +143,70 @@ def start_app():
         
         gr.HTML("""
             <details>
-                <summary>‼️Usage (использование)</summary>
-                Wan-Animate поддерживает два режима:
+                <summary>‼️Usage (использования)</summary>
+                Wan-Animate supports two modes:
                 <ul>
-                    <li>Move Mode: анимирует персонажа в изображении движениями из видео</li>
-                    <li>Mix Mode: заменяет персонажа в видео на персонажа из изображения</li>
+                    <li>Move Mode: animate the character in input image with movements from the input video</li>
+                    <li>Mix Mode: replace the character in input video with the character in input image</li>
+                </ul>
+                Wan-Animate supports two modes:
+                <ul>
+                    <li>Move Mode: Use the movements extracted from the input video to drive the character in the input image</li>
+                    <li>Mix Mode: Use the character in the input image to replace the character in the input video</li>
+                </ul>
+                Currently, the following restrictions apply to inputs:
+                <ul>
+                    <li>Video file size: Less than 200MB</li>
+                    <li>Video resolution: The shorter side must be greater than 200, and the longer side must be less than 2048</li>
+                    <li>Video duration: 2s to 30s</li>
+                    <li>Video aspect ratio: 1:3 to 3:1</li>
+                    <li>Video formats: mp4, avi, mov</li>
+                    <li>Image file size: Less than 5MB</li>
+                    <li>Image resolution: The shorter side must be greater than 200, and the longer side must be less than 4096</li>
+                    <li>Image formats: jpg, png, jpeg, webp, bmp</li>
+                </ul>
+                Current, the inference quality has two variants. You can use our open-source code for more flexible configuration.
+                <ul>
+                    <li>wan-pro: 25fps, 720p</li>
+                    <li>wan-std: 15fps, 720p</li>
                 </ul>
-                Ограничения: Видео <200MB, 2–30s, 200–2048 res; Изображение <5MB, 200–4096 res.
-                Quality: wan-pro (25fps, 720p), wan-std (15fps, 720p).
             </details>
         """)
 
         with gr.Row():
             with gr.Column():    
+                # Dropdown for img
+                img_dropdown = gr.Dropdown(
+                    label="Select Image from img folder (выбери фото)",
+                    choices=app.get_img_files(),
+                    value=None,
+                )
                 ref_img = gr.Image(
-                    label="Reference Image (загрузи фото)",
+                    label="Reference Image (изображение)",
                     type="filepath",
-                    sources=["upload"],
+                    value=None,
+                )
+                img_dropdown.change(
+                    fn=app.load_img_from_dropdown,
+                    inputs=[img_dropdown],
+                    outputs=[ref_img],
+                )
+
+                # Dropdown for video
+                video_dropdown = gr.Dropdown(
+                    label="Select Video from video folder (выбери видео)",
+                    choices=app.get_video_files(),
+                    value=None,
                 )
-                
                 video = gr.Video(
-                    label="Template Video (загрузи видео)",
+                    label="Template Video (шаблонное видео)",
                     sources=["upload"],
+                    value=None,
+                )
+                video_dropdown.change(
+                    fn=app.load_video_from_dropdown,
+                    inputs=[video_dropdown],
+                    outputs=[video],
                 )
                 
                 with gr.Row():