Update app.py

app.py

@@ -3,38 +3,32 @@ from PIL import Image, ImageSequence
 import imageio
 
 def edit_file(file, loop_count, speed_factor, frame_skip):
-    file_name = file.name.split("/")[-1]  # 元ファイル名を取得
-    base_name = file_name.rsplit(".", 1)[0]  # 拡張子を除いた部分
+    file_name = file.name.split("/")[-1]
+    base_name = file_name.rsplit(".", 1)[0]
 
-    # ファイルがGIFか動画か判定
+    # 動画またはGIFの読み込み
     if file.name.endswith((".mp4", ".avi", ".mov", ".mkv")):
-        # 動画をGIFに変換
         gif_path = f"{base_name}.gif"
         video = imageio.get_reader(file.name)
         frames = []
         for i, frame in enumerate(video):
-            # 指定したフレーム数ごとにスキップ
-            if i % frame_skip != 0:
-                continue
-            # 再生速度調整
-            frames.append(frame)
-        
-        # GIFとして保存
+            if i % frame_skip == 0:  # フレーム間引き
+                frames.append(frame)
         imageio.mimsave(gif_path, frames, duration=1/speed_factor)
         gif = Image.open(gif_path)
     else:
         gif = Image.open(file.name)
 
-    # GIF編集処理
+    # GIF編集
     frames = []
     durations = []
+    original_duration = gif.info.get("duration", 100)  # フレーム間の時間(ms)
     for i, frame in enumerate(ImageSequence.Iterator(gif)):
-        if i % frame_skip != 0:  # フレームスキップ
-            continue
-        frames.append(frame.copy())
-        durations.append(gif.info.get("duration", 100) // speed_factor)
+        if i % frame_skip == 0:  # フレーム間引き
+            frames.append(frame.copy())
+            durations.append(original_duration // speed_factor)  # 再生速度調整
 
-    # 保存先をアップロードされたファイル名ベースに
+    # 保存
     output_path = f"{base_name}_edited.gif"
     frames[0].save(
         output_path,
@@ -45,7 +39,7 @@ def edit_file(file, loop_count, speed_factor, frame_skip):
         disposal=2
     )
 
-    return output_path, output_path  # プレビュー表示用に同じパスを返す
+    return output_path, output_path
 
 # Gradioインターフェース
 interface = gr.Interface(
@@ -54,11 +48,11 @@ interface = gr.Interface(
         gr.File(label="GIFまたは動画をアップロード"),
         gr.Number(label="ループ回数（0は無限ループ）", value=0),
         gr.Number(label="再生速度の倍率（例: 2は2倍速）", value=1),
-        gr.Number(label="フレームスキップ数", value=1),
+        gr.Number(label="フレームスキップ数（1以上）", value=1, min=1),
     ],
     outputs=[
         gr.File(label="ダウンロードリンク"),
-        gr.Image(label="プレビュー", type="filepath")  # 変更箇所
+        gr.Image(label="プレビュー", type="filepath")
     ]
 )