Initial commit

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,3 @@
+*.gif
+test.*
+*__pycache__*

README.md

@@ -0,0 +1,12 @@
+---
+title: Video to Gif
+emoji: 🎥
+colorFrom: yellow
+colorTo: gray
+sdk: gradio
+sdk_version: 5.22.0
+app_file: app.py
+pinned: false
+short_description: Convert videos to GIFs
+license: apache-2.0
+---

app.py

@@ -0,0 +1,100 @@
+import os
+import math
+import gradio as gr
+from PIL import Image, ImageSequence
+from moviepy.editor import VideoFileClip
+from utils import clean_dir, TMP_DIR, EN_US
+
+ZH2EN = {
+    "倍速": "Speed",
+    "状态栏": "Status",
+    "文件名": "Filename",
+    "下载动图": "Download GIF",
+    "视频过长": "Video too long",
+    "视频转 GIF 动图": "Video to GIF",
+    "上传视频 (请确保完整上传再点击提交，若时长超过五秒可先在线裁剪)": "Upload video (please ensure the video is completely uploaded before clicking submit, you can crop it online to < 5s)",
+}
+
+
+def _L(zh_txt: str):
+    return ZH2EN[zh_txt] if EN_US else zh_txt
+
+
+def get_frame_dur(gif: Image):
+    # 获取 GIF 图像中第一帧的 duration
+    dur = gif.info.get("duration", 100)
+    # 返回每一帧的 duration
+    return [frame.info.get("duration", dur) for frame in ImageSequence.Iterator(gif)]
+
+
+def resize_gif(target_width: int, target_height: int, input_gif, output_gif):
+    gif = Image.open(input_gif)
+    modified_frames = []
+    for frame in ImageSequence.Iterator(gif):
+        resized_frame = frame.resize((target_width, target_height), Image.LANCZOS)
+        modified_frames.append(resized_frame)
+
+    frame_durations = get_frame_dur(gif)
+    # 将修改后的帧作为新的 GIF 保存
+    modified_frames[0].save(
+        output_gif,
+        format="GIF",
+        append_images=modified_frames[1:],
+        save_all=True,
+        duration=frame_durations,
+        loop=0,
+    )
+
+    return output_gif
+
+
+# outer func
+def infer(video_path: str, speed: float, target_w=640, cache=f"{TMP_DIR}/gif"):
+    status = "Success"
+    gif_name = gif_out = None
+    try:
+        clean_dir(cache)
+        with VideoFileClip(video_path, audio_fps=16000) as clip:
+            if clip.duration > 5:
+                raise ValueError(_L("视频过长"))
+
+            clip.speedx(speed).to_gif(f"{cache}/input.gif", fps=12)
+            w, h = clip.size
+
+        gif_in = f"{cache}/input.gif"
+        target_h = math.ceil(target_w * h / w)
+        gif_name = os.path.basename(video_path)
+        gif_out = resize_gif(target_w, target_h, gif_in, f"{cache}/output.gif")
+
+    except Exception as e:
+        status = f"{e}"
+
+    return status, gif_name, gif_out
+
+
+if __name__ == "__main__":
+    example = (
+        "https://www.modelscope.cn/studio/Genius-Society/video2gif/resolve/master"
+        if EN_US
+        else "."
+    )
+    gr.Interface(
+        fn=infer,
+        inputs=[
+            gr.Video(
+                label=_L(
+                    "上传视频 (请确保完整上传再点击提交，若时长超过五秒可先在线裁剪)"
+                )
+            ),
+            gr.Slider(label=_L("倍速"), minimum=0.5, maximum=2.0, step=0.25, value=1.0),
+        ],
+        outputs=[
+            gr.Textbox(label=_L("状态栏"), show_copy_button=True),
+            gr.Textbox(label=_L("文件名"), show_copy_button=True),
+            gr.Image(label=_L("下载动图"), type="filepath", show_share_button=False),
+        ],
+        flagging_mode="never",
+        examples=[[f"{example}/examples/herta.mp4", 2]],
+        cache_examples=False,
+        title=_L("视频转 GIF 动图"),
+    ).launch()

requirements.txt

@@ -0,0 +1 @@
+moviepy==1.0.3

utils.py

@@ -0,0 +1,12 @@
+import os
+import shutil
+
+EN_US = os.getenv("LANG") != "zh_CN.UTF-8"
+TMP_DIR = "./__pycache__"
+
+
+def clean_dir(dir_path: str):
+    if os.path.exists(dir_path):
+        shutil.rmtree(dir_path)
+
+    os.makedirs(dir_path)