Update app.py

app.py

@@ -412,6 +412,43 @@ def get_models():
     return models
 
 
+# 事前定義された解像度リスト（グローバル変数として追加）
+PREDEFINED_RESOLUTIONS = [
+    (416, 960), (448, 864), (480, 832), (512, 768), (544, 704), 
+    (576, 672), (608, 640), (640, 608), (672, 576), (704, 544), 
+    (768, 512), (832, 480), (864, 448), (960, 416)
+]
+
+# 最も近いアスペクト比を見つける関数
+def find_closest_aspect_ratio(width, height, target_resolutions):
+    """
+    事前定義された解像度リストから、元の画像のアスペクト比に最も近い解像度を見つける
+    
+    引数:
+        width: 元の画像の幅
+        height: 元の画像の高さ
+        target_resolutions: 目標解像度のリスト（幅, 高さ）のタプル
+        
+    戻り値:
+        tuple: 最も近いアスペクト比の (target_width, target_height)
+    """
+    original_aspect = width / height
+    
+    # 各目標解像度に対してアスペクト比の差を計算
+    min_diff = float('inf')
+    closest_resolution = None
+    
+    for target_width, target_height in target_resolutions:
+        target_aspect = target_width / target_height
+        diff = abs(original_aspect - target_aspect)
+        
+        if diff < min_diff:
+            min_diff = diff
+            closest_resolution = (target_width, target_height)
+    
+    return closest_resolution
+
+
 stream = AsyncStream()
 
 
@@ -515,14 +552,24 @@ def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_wind
 
         try:
             H, W, C = input_image.shape
-            height, width = find_nearest_bucket(H, W, resolution=640)
+            
+            # 事前定義された解像度から最も近いアスペクト比を見つける
+            target_width, target_height = find_closest_aspect_ratio(W, H, PREDEFINED_RESOLUTIONS)
+            
+            # リサイズメッセージを表示
+            resize_message = f'画像を最適なアスペクト比にリサイズしています: {target_width}x{target_height}'
+            stream.output_queue.push(('progress', (None, '', make_progress_bar_html(0, resize_message))))
             
             # CPUモードの場合、処理サイズを小さくする
             if cpu_fallback_mode:
-                height = min(height, 320)
-                width = min(width, 320)
+                scale_factor = min(320 / target_height, 320 / target_width)
+                target_height = int(target_height * scale_factor)
+                target_width = int(target_width * scale_factor)
                 
-            input_image_np = resize_and_center_crop(input_image, target_width=width, target_height=height)
+            print(f'元の画像サイズ: {W}x{H}, リサイズ先: {target_width}x{target_height}')
+            
+            # 選択された解像度にリサイズ
+            input_image_np = resize_and_center_crop(input_image, target_width=target_width, target_height=target_height)
 
             Image.fromarray(input_image_np).save(os.path.join(outputs_folder, f'{job_id}.png'))
 
@@ -1489,10 +1536,10 @@ with block:
                 # タッチ操作を最適化するためにslider-containerクラスを追加
                 with gr.Group(elem_classes="slider-container"):
                     total_second_length = gr.Slider(
-                        label="動画の長さ（最大5秒） / Video Length (max 5 seconds)", 
-                        minimum=1, 
-                        maximum=5, 
-                        value=3, 
+                        label="動画の長さ（最大1秒） / Video Length (max 1 seconds)", 
+                        minimum=0.5, 
+                        maximum=1, 
+                        value=1, 
                         step=0.1
                     )
                     
@@ -1654,4 +1701,5 @@ def create_error_html(error_msg, is_timeout=False):
             }}
         }})();
     </script>
-    """
+    """
+