Update app.py

app.py

@@ -6,107 +6,92 @@ import insightface
 import urllib.request
 from insightface.app import FaceAnalysis
 
-# --- 0. モデルの自動ダウンロード設定 ---
+# --- 0. モデルの自動ダウンロード ---
 def download_model_if_missing():
     model_filename = 'inswapper_128.onnx'
-    # 公開されている信頼性の高いURLから取得
     url = "https://huggingface.co/ezioruan/inswapper_128.onnx/resolve/main/inswapper_128.onnx"
-    
     if not os.path.exists(model_filename):
-        print(f"【情報】{model_filename} が見つかりません。ダウンロードを開始します...")
-        try:
-            # プログレスバーなしのシンプルなダウンロード
-            urllib.request.urlretrieve(url, model_filename)
-            print(f"【完了】{model_filename} のダウンロードに成功しました。")
-        except Exception as e:
-            print(f"【エラー】ダウンロードに失敗しました: {e}")
+        print("Downloading model...")
+        urllib.request.urlretrieve(url, model_filename)
 
-# 実行
 download_model_if_missing()
 
 # --- 1. モデルの初期化 ---
-try:
-    # 顔検出モデル (buffalo_l)
-    app = FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])
-    app.prepare(ctx_id=0, det_size=(640, 640))
-    
-    # 顔入れ替えモデル (inswapper_128.onnx)
-    model_path = 'inswapper_128.onnx'
-    if os.path.exists(model_path):
-        swapper = insightface.model_zoo.get_model(model_path, download=False, providers=['CPUExecutionProvider'])
-    else:
-        swapper = None
-except Exception as e:
-    print(f"Initialization Error: {e}")
-    swapper = None
+app = FaceAnalysis(name='buffalo_l', providers=['CPUExecutionProvider'])
+app.prepare(ctx_id=0, det_size=(640, 640))
+swapper = insightface.model_zoo.get_model('inswapper_128.onnx', download=False, providers=['CPUExecutionProvider'])
 
 # --- 2. 動画処理ロジック ---
-def process_video(source_img, target_video):
-    if swapper is None:
-        return None, "モデルファイルが読み込めていません。リポジトリを確認してください。"
+def process_video(source_img, target_video, progress=gr.Progress()):
     if source_img is None or target_video is None:
-        return None, "画像と動画の両方をアップロードしてください。"
+        return None, "画像と動画を選択してください。"
 
-    # ソース顔（入れ替え後の顔）の抽出
+    # ソース顔の抽出
     source_faces = app.get(source_img)
     if not source_faces:
-        return None, "画像から顔を検出できませんでした。"
+        return None, "画像から顔が見つかりません。"
     source_face = source_faces[0]
 
-    # 動画読み込み
+    # 動画の読み込み
     cap = cv2.VideoCapture(target_video)
+    if not cap.isOpened():
+        return None, "動画ファイルを開けませんでした。"
+
     fps = cap.get(cv2.CAP_PROP_FPS)
     width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-    
-    # 出力設定
-    output_path = "output_swapped.mp4"
-    # OpenCVのデフォルトmp4コーデック
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+
+    # 出力ファイル設定 (より汎用的な mp4v を使用)
+    output_path = "output_result.mp4"
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
 
-    frame_count = 0
-    while cap.isOpened():
+    # フレーム処理ループ
+    frame_idx = 0
+    while True:
         ret, frame = cap.read()
         if not ret:
             break
         
-        # フレーム内の全ターゲット顔を検出
-        target_faces = app.get(frame)
-        
-        # 入れ替え実行（トラッキング的に全顔対応）
+        # 顔の検出と入れ替え
+        faces = app.get(frame)
         res_frame = frame.copy()
-        for face in target_faces:
+        for face in faces:
             res_frame = swapper.get(res_frame, face, source_face, paste_back=True)
         
         out.write(res_frame)
-        frame_count += 1
-    
+        frame_idx += 1
+        
+        # 進捗をUIに反映
+        if total_frames > 0:
+            progress(frame_idx / total_frames, desc=f"処理中... {frame_idx}/{total_frames}")
+
     cap.release()
     out.release()
     
-    return output_path, f"成功！ {frame_count} フレームを処理しました。"
+    # 最終的なファイルの存在確認
+    if not os.path.exists(output_path):
+        return None, "動画の生成に失敗しました。"
 
-# --- 3. UI (Gradio) ---
+    return output_path, "完了しました！"
+
+# --- 3. UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("# 🎬 AI Face Swap Video Processor (Stable Ver.)")
-    gr.Markdown("アップロードされた動画の全フレームをスキャンし、顔を自動で追跡して入れ替えます。")
-    
+    gr.Markdown("### 🎬 Stable Face Swapper")
     with gr.Row():
         with gr.Column():
-            src_img = gr.Image(label="1. 入れ替え後の顔画像", type="numpy")
-            tgt_vid = gr.Video(label="2. 対象の動画ファイル")
-            btn = gr.Button("動画を変換する")
-        
+            img_input = gr.Image(label="入れ替える顔画像", type="numpy")
+            vid_input = gr.Video(label="ターゲット動画")
+            submit_btn = gr.Button("変換開始")
         with gr.Column():
-            out_vid = gr.Video(label="3. 生成結果（ダウンロード可）")
-            status = gr.Textbox(label="ステータス", interactive=False)
+            vid_output = gr.Video(label="出力結果")
+            status_text = gr.Textbox(label="ステータス")
 
-    btn.click(
+    submit_btn.click(
         fn=process_video,
-        inputs=[src_img, tgt_vid],
-        outputs=[out_vid, status]
+        inputs=[img_input, vid_input],
+        outputs=[vid_output, status_text]
     )
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()