映像処理機能を改善し、オーディオをマージする際の一時ファイルを作成して削除するロジックを追加しました。音声が無い場合のエラーハンドリングも強化しました。

ben_base.py

@@ -1,4 +1,3 @@
-
 import math
 import torch
 import torch.nn as nn
@@ -1102,18 +1101,26 @@ class BEN_Base(nn.Module):
             alpha_webm_path = os.path.join(output_path, "foreground.webm")
             pil_images_to_webm_alpha(foregrounds, alpha_webm_path, fps=original_fps)
         else:
-            fg_output = os.path.join(output_path, 'foreground.mp4')
-            pil_images_to_mp4(foregrounds, fg_output, fps=original_fps, rgb_value=rgb_value)
-            cv2.destroyAllWindows()
-            
+            # 1) 一時ファイルに映像のみを書き出す
+            fg_no_audio = os.path.join(output_path, 'foreground_noaudio.mp4')
+            pil_images_to_mp4(foregrounds, fg_no_audio, fps=original_fps, rgb_value=rgb_value)
+
+            # 2) オーディオをマージして最終ファイルを生成
+            final_fg_output = os.path.join(output_path, 'foreground.mp4')
             try:
-                fg_audio_output = os.path.join(output_path, 'foreground.mp4')
-                add_audio_to_video(fg_output, video_path, fg_audio_output)
+                add_audio_to_video(fg_no_audio, video_path, final_fg_output)
+                # 映像のみの一時ファイルを削除
+                if os.path.exists(fg_no_audio):
+                    os.remove(fg_no_audio)
             except Exception as e:
-                print("No audio found in the original video")
+                # 音声が無い場合は映像のみのファイルをそのままリネーム/保持
+                print("No audio track found or failed to merge audio. Keeping video without audio.")
                 print(e)
+                if not os.path.exists(final_fg_output):
+                    os.rename(fg_no_audio, final_fg_output)
+
+            cv2.destroyAllWindows()
 
-