Update app.py

app.py

@@ -1,22 +1,32 @@
 import gradio as gr
 from PIL import Image, ImageSequence
+from moviepy.editor import VideoFileClip
 
-def edit_gif(gif_file, loop_count, speed_factor, frame_skip):
-    # GIFを読み込む
-    gif = Image.open(gif_file)
+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に変換
+        clip = VideoFileClip(file.name)
+        gif_path = f"{base_name}.gif"
+        clip.write_gif(gif_path, fps=10)
+        gif = Image.open(gif_path)
+    else:
+        gif = Image.open(file.name)
+
+    # GIF編集処理
     frames = []
     durations = []
-
-    # フレームを順に処理
     for i, frame in enumerate(ImageSequence.Iterator(gif)):
-        # 指定したフレーム数ごとにスキップ
-        if i % frame_skip != 0:
+        if i % frame_skip != 0:  # フレームスキップ
             continue
         frames.append(frame.copy())
-        durations.append(gif.info.get("duration", 100) // speed_factor)  # 再生速度調整
+        durations.append(gif.info.get("duration", 100) // speed_factor)
 
-    # 新しいGIFを保存
-    output_path = "edited.gif"
+    # 保存先をアップロードされたファイル名ベースに
+    output_path = f"{base_name}_edited.gif"
     frames[0].save(
         output_path,
         save_all=True,
@@ -25,18 +35,22 @@ def edit_gif(gif_file, loop_count, speed_factor, frame_skip):
         duration=durations,
         disposal=2
     )
-    return output_path
+
+    return output_path, output_path  # プレビュー表示用に同じパスを返す
 
 # Gradioインターフェース
 interface = gr.Interface(
-    fn=edit_gif,
+    fn=edit_file,
     inputs=[
-        gr.File(label="アップロードするGIF"),
+        gr.File(label="GIFまたは動画をアップロード"),
         gr.Number(label="ループ回数（0は無限ループ）", value=0),
         gr.Number(label="再生速度の倍率（例: 2は2倍速）", value=1),
         gr.Number(label="フレームスキップ数", value=1),
     ],
-    outputs=gr.File(label="編集されたGIF")
+    outputs=[
+        gr.File(label="ダウンロードリンク"),
+        gr.Image(label="プレビュー", type="file")
+    ]
 )
 
 interface.launch()