Update app.py

app.py

@@ -2,16 +2,15 @@ import cv2
 import mediapipe as mp
 import tempfile
 import gradio as gr
-import shutil
 import os
+import subprocess
 
 mp_face_mesh = mp.solutions.face_mesh
 
 def process_video_with_landmarks(video_path):
-    # OpenCVで読み込み確認
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened():
-        return "❌ エラー: 動画ファイルを開けませんでした。形式が正しいか確認してください。"
+        return "❌ エラー: 動画ファイルを開けませんでした。形式を確認してください。"
 
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
@@ -19,9 +18,9 @@ def process_video_with_landmarks(video_path):
     if fps == 0:
         fps = 25  # fallback
 
-    # 一時ファイル（出力動画）
-    temp_output = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-    out = cv2.VideoWriter(temp_output.name, cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height))
+    # 一時AVIファイルで書き出す
+    temp_avi = tempfile.NamedTemporaryFile(delete=False, suffix=".avi")
+    out = cv2.VideoWriter(temp_avi.name, cv2.VideoWriter_fourcc(*'XVID'), fps, (width, height))
 
     face_mesh = mp_face_mesh.FaceMesh(
         static_image_mode=False,
@@ -39,7 +38,6 @@ def process_video_with_landmarks(video_path):
         rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
         results = face_mesh.process(rgb)
 
-        # 青いランドマーク点を描画
         if results.multi_face_landmarks:
             for face_landmarks in results.multi_face_landmarks:
                 for lm in face_landmarks.landmark:
@@ -53,10 +51,26 @@ def process_video_with_landmarks(video_path):
     out.release()
     face_mesh.close()
 
-    return temp_output.name
+    # mp4へ変換
+    temp_mp4 = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+    ffmpeg_cmd = [
+        "ffmpeg",
+        "-y",
+        "-i", temp_avi.name,
+        "-vcodec", "libx264",
+        "-crf", "23",
+        "-preset", "medium",
+        temp_mp4.name
+    ]
+
+    try:
+        subprocess.run(ffmpeg_cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    except subprocess.CalledProcessError:
+        return "❌ ffmpegによるmp4変換に失敗しました。"
+
+    return temp_mp4.name
 
 def gradio_interface(video_file):
-    # video_file: パス (str) または File オブジェクト (gradioの仕様により)
     if isinstance(video_file, str) and os.path.isfile(video_file):
         return process_video_with_landmarks(video_file)
     else:
@@ -65,9 +79,9 @@ def gradio_interface(video_file):
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=gr.Video(label="動画ファイルをアップロード"),
-    outputs=gr.File(label="ランドマーク描画済み動画 (.mp4)"),
-    title="Face Mesh ランドマーク付き動画出力",
-    description="動画から顔のランドマークを検出し、各点を青く描画した動画を出力します。対応形式: .mp4（推奨）"
+    outputs=gr.File(label="再生可能なランドマーク付きmp4動画"),
+    title="Face Mesh ランドマーク付き動画出力（再生保証）",
+    description="動画に顔ランドマーク（青点）を描画し、再生互換性のあるmp4形式で出力します。"
 )
 
 iface.launch()