Update app.py

app.py

@@ -1,32 +1,46 @@
 import gradio as gr
-from moviepy.editor import VideoFileClip
+import imageio
+import os
 
-def edit_file(file, loop_count, speed_factor, frame_skip):
+def edit_file(file, apply_loop, loop_count, apply_speed, speed_factor, apply_skip, frame_skip):
     file_name = file.name.split("/")[-1]
     base_name = file_name.rsplit(".", 1)[0]
     
-    # 動画またはGIFの読み込み
-    clip = VideoFileClip(file.name)
-    
+    # GIFの読み込み
+    reader = imageio.get_reader(file.name)
+    fps = reader.get_meta_data().get("fps", 10)  # デフォルトFPSは10
+
     # フレームスキップと速度調整
-    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)
+    frames = []
+    for i, frame in enumerate(reader):
+        if apply_skip and i % frame_skip != 0:  # フレームスキップを適用
+            continue
+        frames.append(frame)
+
+    # 再生速度調整
+    if apply_speed and speed_factor != 1:
+        fps = int(fps * speed_factor)
     
-    # GIFとして書き出し
+    # GIFとして保存
     output_path = f"{base_name}_edited.gif"
-    clip.write_gif(output_path, loop=loop_count)
-    
+    writer = imageio.get_writer(output_path, fps=fps, loop=0 if not apply_loop else loop_count)
+
+    for frame in frames:
+        writer.append_data(frame)
+    writer.close()
+
     return output_path, output_path
 
 # Gradioインターフェース
 interface = gr.Interface(
     fn=edit_file,
     inputs=[
-        gr.File(label="GIFまたは動画をアップロード"),
+        gr.File(label="GIFをアップロード"),
+        gr.Checkbox(label="ループ回数を変更する", value=True),
         gr.Number(label="ループ回数（0は無限ループ）", value=0),
+        gr.Checkbox(label="再生速度を変更する", value=False),
         gr.Number(label="再生速度の倍率（例: 2は2倍速）", value=1),
+        gr.Checkbox(label="フレームスキップを変更する", value=False),
         gr.Number(label="フレームスキップ数（1以上）", value=1, minimum=1),
     ],
     outputs=[