Create replayproof_media.py

replayproof_media.py

@@ -0,0 +1,59 @@
+from __future__ import annotations
+
+import os
+import hashlib
+from typing import List, Optional
+
+import imageio.v2 as imageio
+import numpy as np
+from PIL import Image, ImageDraw
+
+
+def sha256_file(path: str) -> str:
+    h = hashlib.sha256()
+    with open(path, "rb") as f:
+        for chunk in iter(lambda: f.read(1024 * 1024), b""):
+            h.update(chunk)
+    return h.hexdigest()
+
+
+def watermark_image(img: Image.Image, text: str) -> Image.Image:
+    """
+    Simple bottom-right watermark.
+    Keeps dependencies minimal (no fonts required).
+    """
+    out = img.copy()
+    d = ImageDraw.Draw(out)
+
+    pad = 8
+    w, h = out.size
+    # textlength exists on modern Pillow; if missing, approximate
+    try:
+        tw = d.textlength(text)
+    except Exception:
+        tw = 8 * len(text)
+
+    x = max(pad, int(w - tw - pad))
+    y = h - 20
+
+    d.rectangle([x - 6, y - 2, w - pad + 2, y + 16], fill=(0, 0, 0))
+    d.text((x, y), text, fill=(235, 235, 235))
+    return out
+
+
+def export_gif(
+    frames: List[Image.Image],
+    out_path: str,
+    fps: int = 10,
+    watermark: Optional[str] = None,
+) -> str:
+    os.makedirs(os.path.dirname(out_path), exist_ok=True)
+
+    imgs = []
+    for im in frames:
+        if watermark:
+            im = watermark_image(im, watermark)
+        imgs.append(np.array(im.convert("RGB")))
+
+    imageio.mimsave(out_path, imgs, duration=1.0 / max(1, int(fps)))
+    return out_path