app.py

import gradio as gr
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]  # 拡張子を除いた部分

    # ファイルが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として保存
        imageio.mimsave(gif_path, frames, duration=1/speed_factor)
        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:  # フレームスキップ
            continue
        frames.append(frame.copy())
        durations.append(gif.info.get("duration", 100) // speed_factor)

    # 保存先をアップロードされたファイル名ベースに
    output_path = f"{base_name}_edited.gif"
    frames[0].save(
        output_path,
        save_all=True,
        append_images=frames[1:],
        loop=loop_count,
        duration=durations,
        disposal=2
    )

    return output_path, output_path  # プレビュー表示用に同じパスを返す

# Gradioインターフェース
interface = gr.Interface(
    fn=edit_file,
    inputs=[
        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="ダウンロードリンク"),
        gr.Image(label="プレビュー", type="file")
    ]
)

interface.launch()