Upload 3 files

packages.txt

@@ -1,4 +1,5 @@
-libgl1
-libglib2.0-0
-libgles2
-libegl1
+libgl1
+libglib2.0-0
+libgles2
+libegl1
+ffmpeg

requirements.txt

@@ -1,91 +1,93 @@
-absl-py==2.4.0
-accelerate==1.14.0
-annotated-doc==0.0.4
-annotated-types==0.7.0
-anyio==4.14.0
-brotli==1.2.0
-certifi==2026.6.17
-cffi==2.0.0
-charset-normalizer==3.4.7
-click==8.4.1
-colorama==0.4.6
-contourpy==1.3.3
-cycler==0.12.1
-decord==0.6.0
-einops==0.8.2
-fastapi==0.138.0
-filelock==3.29.0
-flatbuffers==25.12.19
-fonttools==4.63.0
-fsspec==2026.4.0
-gradio==5.50.0
-gradio-client==1.14.0
-groovy==0.1.2
-h11==0.16.0
-hf-xet==1.5.1
-httpcore==1.0.9
-httpx==0.28.1
-huggingface-hub==0.36.2
-idna==3.18
-Jinja2==3.1.6
-kiwisolver==1.5.0
-markdown-it-py==4.2.0
-MarkupSafe==3.0.3
-matplotlib==3.11.0
-mdurl==0.1.2
-mediapipe==0.10.35
-mpmath==1.3.0
-networkx==3.6.1
-numpy==2.4.4
-nvidia-ml-py==13.610.43
-opencv-contrib-python==4.13.0.92
-opencv-python==4.13.0.92
-orjson==3.11.9
-packaging==26.0
-pandas<3.0
-pillow<12.0
-pip==26.1.2
-polars==1.41.2
-polars-runtime-32==1.41.2
-protobuf==7.35.1
-psutil==7.2.2
-pycparser==3.0
-pydantic<=2.12.3
-pydub==0.25.1
-Pygments==2.20.0
-pyparsing==3.3.2
-python-dateutil==2.9.0.post0
-python-dotenv==1.2.2
-python-multipart==0.0.32
-pytz==2026.2
-PyYAML==6.0.3
-regex==2026.5.9
-requests==2.34.2
-rich==15.0.0
-safehttpx==0.1.7
-safetensors==0.8.0
-scipy==1.17.1
-semantic-version==2.10.0
-sentencepiece==0.2.1
-setuptools==70.2.0
-shellingham==1.5.4
-six==1.17.0
-sounddevice==0.5.5
-starlette>=0.40.0,<1.0
-sympy==1.14.0
-tiktoken==0.13.0
-tokenizers==0.19.1
-tomlkit>=0.12.0,<0.14.0
-torch==2.12.1
-torchvision==0.27.1
-tqdm==4.68.3
-transformers==4.44.2
-typer==0.25.1
-typing_extensions==4.15.0
-typing-inspection==0.4.2
-tzdata==2026.2
-ultralytics==8.4.75
-ultralytics-thop==2.0.20
-urllib3==2.7.0
-uvicorn==0.49.0
-wheel==0.47.0
+absl-py==2.4.0
+accelerate==1.14.0
+annotated-doc==0.0.4
+annotated-types==0.7.0
+anyio==4.14.0
+brotli==1.2.0
+certifi==2026.6.17
+cffi==2.0.0
+charset-normalizer==3.4.7
+click==8.4.1
+colorama==0.4.6
+contourpy==1.3.3
+cycler==0.12.1
+decord==0.6.0
+einops==0.8.2
+fastapi==0.138.0
+filelock==3.29.0
+flatbuffers==25.12.19
+fonttools==4.63.0
+fsspec==2026.4.0
+gradio==5.50.0
+gradio-client==1.14.0
+groovy==0.1.2
+h11==0.16.0
+hf-xet==1.5.1
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.36.2
+idna==3.18
+imageio==2.37.3
+imageio-ffmpeg==0.6.0
+Jinja2==3.1.6
+kiwisolver==1.5.0
+markdown-it-py==4.2.0
+MarkupSafe==3.0.3
+matplotlib==3.11.0
+mdurl==0.1.2
+mediapipe==0.10.35
+mpmath==1.3.0
+networkx==3.6.1
+numpy==2.4.4
+nvidia-ml-py==13.610.43
+opencv-contrib-python==4.13.0.92
+opencv-python==4.13.0.92
+orjson==3.11.9
+packaging==26.0
+pandas<3.0
+pillow<12.0
+pip==26.1.2
+polars==1.41.2
+polars-runtime-32==1.41.2
+protobuf==7.35.1
+psutil==7.2.2
+pycparser==3.0
+pydantic<=2.12.3
+pydub==0.25.1
+Pygments==2.20.0
+pyparsing==3.3.2
+python-dateutil==2.9.0.post0
+python-dotenv==1.2.2
+python-multipart==0.0.32
+pytz==2026.2
+PyYAML==6.0.3
+regex==2026.5.9
+requests==2.34.2
+rich==15.0.0
+safehttpx==0.1.7
+safetensors==0.8.0
+scipy==1.17.1
+semantic-version==2.10.0
+sentencepiece==0.2.1
+setuptools==70.2.0
+shellingham==1.5.4
+six==1.17.0
+sounddevice==0.5.5
+starlette>=0.40.0,<1.0
+sympy==1.14.0
+tiktoken==0.13.0
+tokenizers==0.19.1
+tomlkit>=0.12.0,<0.14.0
+torch==2.12.1+cpu
+torchvision==0.27.1+cpu
+tqdm==4.68.3
+transformers==4.44.2
+typer==0.25.1
+typing_extensions==4.15.0
+typing-inspection==0.4.2
+tzdata==2026.2
+ultralytics==8.4.75
+ultralytics-thop==2.0.20
+urllib3==2.7.0
+uvicorn==0.49.0
+wheel==0.47.0

visualize_pose.py

@@ -9,12 +9,17 @@ Encodes the output as a high-definition H.264 MP4 video for browser playback.
 """
 
 import os
+import shutil
 import tempfile
 import time
 import subprocess
 import cv2
 import numpy as np
-import imageio_ffmpeg
+
+try:
+    import imageio_ffmpeg
+except ImportError:
+    imageio_ffmpeg = None
 
 # ==============================================================================
 # 1. LANDMARK SKELETON CONNECTIONS & FACIAL DETAIL INDICES
@@ -328,35 +333,71 @@ def render_keypoints_video(
         ts = int(time.time() * 1000)
         out_path = os.path.join(out_dir, f"keypoints_1080p_{ts}.mp4")
 
-        ffmpeg_bin = imageio_ffmpeg.get_ffmpeg_exe()
-        cmd = [
-            ffmpeg_bin, "-y",
-            "-f", "rawvideo",
-            "-vcodec", "rawvideo",
-            "-s", f"{out_w}x{out_h}",
-            "-pix_fmt", "rgb24",
-            "-r", str(fps),
-            "-i", "-",
-            "-c:v", "libx264",
-            "-preset", "fast",
-            "-crf", "18",         # Near lossless visual quality
-            "-pix_fmt", "yuv420p",
-            "-movflags", "+faststart",
-            out_path
-        ]
-
-        proc = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-        for frame, kps in zip(video_frames, keypoints_list):
-            drawn = draw_keypoints_frame(
-                frame=frame,
-                kps=kps,
-                target_width=out_w,
-                target_height=out_h,
-                **render_kwargs
-            )
-            proc.stdin.write(drawn.tobytes())
-        proc.stdin.close()
-        proc.wait()
+        # Resolve ffmpeg binary (from imageio_ffmpeg, system PATH, or fallback)
+        ffmpeg_bin = None
+        if imageio_ffmpeg is not None:
+            try:
+                ffmpeg_bin = imageio_ffmpeg.get_ffmpeg_exe()
+            except Exception:
+                ffmpeg_bin = None
+        if not ffmpeg_bin:
+            ffmpeg_bin = shutil.which("ffmpeg")
+
+        use_ffmpeg = False
+        if ffmpeg_bin:
+            try:
+                test_proc = subprocess.run([ffmpeg_bin, "-version"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                if test_proc.returncode == 0:
+                    use_ffmpeg = True
+            except Exception:
+                use_ffmpeg = False
+
+        if use_ffmpeg:
+            cmd = [
+                ffmpeg_bin, "-y",
+                "-f", "rawvideo",
+                "-vcodec", "rawvideo",
+                "-s", f"{out_w}x{out_h}",
+                "-pix_fmt", "rgb24",
+                "-r", str(fps),
+                "-i", "-",
+                "-c:v", "libx264",
+                "-preset", "fast",
+                "-crf", "18",         # Near lossless visual quality
+                "-pix_fmt", "yuv420p",
+                "-movflags", "+faststart",
+                out_path
+            ]
+
+            proc = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+            for frame, kps in zip(video_frames, keypoints_list):
+                drawn = draw_keypoints_frame(
+                    frame=frame,
+                    kps=kps,
+                    target_width=out_w,
+                    target_height=out_h,
+                    **render_kwargs
+                )
+                proc.stdin.write(drawn.tobytes())
+            proc.stdin.close()
+            proc.wait()
+        else:
+            # Fallback to OpenCV VideoWriter if ffmpeg is completely absent
+            fourcc = cv2.VideoWriter_fourcc(*'avc1')
+            writer = cv2.VideoWriter(out_path, fourcc, float(fps), (out_w, out_h))
+            if not writer.isOpened():
+                fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+                writer = cv2.VideoWriter(out_path, fourcc, float(fps), (out_w, out_h))
+            for frame, kps in zip(video_frames, keypoints_list):
+                drawn = draw_keypoints_frame(
+                    frame=frame,
+                    kps=kps,
+                    target_width=out_w,
+                    target_height=out_h,
+                    **render_kwargs
+                )
+                writer.write(cv2.cvtColor(drawn, cv2.COLOR_RGB2BGR))
+            writer.release()
 
         if os.path.exists(out_path) and os.path.getsize(out_path) > 0:
             print(f"Rendered crisp 1080p keypoints video: {out_w}x{out_h} at {out_path}")