Update app.py

app.py

@@ -1,47 +1,23 @@
 import gradio as gr
-from PIL import Image, ImageSequence
-import imageio
+from moviepy.editor import VideoFileClip
 
 def edit_file(file, loop_count, speed_factor, frame_skip):
     file_name = file.name.split("/")[-1]
     base_name = file_name.rsplit(".", 1)[0]
-
+    
     # 動画またはGIFの読み込み
-    if file.name.endswith((".mp4", ".avi", ".mov", ".mkv")):
-        gif_path = f"{base_name}.gif"
-        video = imageio.get_reader(file.name)
-        frames = []
-        for i, frame in enumerate(video):
-            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編集
-    frames = []
-    durations = []
-    original_duration = gif.info.get("duration", 100)  # フレーム間の時間(ms)
-    for i, frame in enumerate(ImageSequence.Iterator(gif)):
-        if i % frame_skip == 0:  # フレーム間引き
-            frames.append(frame.copy())
-            durations.append(original_duration // speed_factor)  # 再生速度調整
-
-    # Pillowのループ設定の仕様に合わせて調整
-    adjusted_loop_count = 0 if loop_count == 0 else loop_count - 1
-
-    # 保存
+    clip = VideoFileClip(file.name)
+    
+    # フレームスキップと速度調整
+    if frame_skip > 1:
+        clip = clip.subclip(0, clip.duration).set_fps(clip.fps / frame_skip)
+    if speed_factor != 1:
+        clip = clip.fx(VideoFileClip.speedx, factor=speed_factor)
+    
+    # GIFとして書き出し
     output_path = f"{base_name}_edited.gif"
-    frames[0].save(
-        output_path,
-        save_all=True,
-        append_images=frames[1:],
-        loop=adjusted_loop_count,  # 修正されたループ回数
-        duration=durations,
-        disposal=2
-    )
-
+    clip.write_gif(output_path, loop=loop_count)
+    
     return output_path, output_path
 
 # Gradioインターフェース