From 1eed88b0f5abbc8ef7bd16caf1195c77a79e729f Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Tue, 28 Jan 2025 13:30:04 +0100
Subject: [PATCH] Override for Gradio's nasty convert_video_to_playable_mp4()

---
 facefusion/uis/core.py      |  4 +++-
 facefusion/uis/overrides.py | 29 +++++++++++++++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 facefusion/uis/overrides.py

diff --git a/facefusion/uis/core.py b/facefusion/uis/core.py
index a7a5469..b8153f4 100644
--- a/facefusion/uis/core.py
+++ b/facefusion/uis/core.py
@@ -7,6 +7,7 @@ from typing import Any, Dict, List, Optional
 import gradio
 from gradio.themes import Size
 
+import facefusion.uis.overrides as uis_overrides
 from facefusion import logger, metadata, state_manager, wording
 from facefusion.exit_helper import hard_exit
 from facefusion.filesystem import resolve_relative_path
@@ -74,7 +75,8 @@ def init() -> None:
 	os.environ['GRADIO_TEMP_DIR'] = os.path.join(state_manager.get_item('temp_path'), 'gradio')
 
 	warnings.filterwarnings('ignore', category = UserWarning, module = 'gradio')
-	gradio.processing_utils._check_allowed = lambda path, check_in_upload_folder: None
+	gradio.processing_utils._check_allowed = uis_overrides.check_allowed #type:ignore
+	gradio.processing_utils.convert_video_to_playable_mp4 = uis_overrides.convert_video_to_playable_mp4
 
 
 def launch() -> None:
diff --git a/facefusion/uis/overrides.py b/facefusion/uis/overrides.py
new file mode 100644
index 0000000..e92a423
--- /dev/null
+++ b/facefusion/uis/overrides.py
@@ -0,0 +1,29 @@
+from facefusion import ffmpeg_builder
+from facefusion.ffmpeg import run_ffmpeg
+from facefusion.filesystem import get_file_size
+from facefusion.temp_helper import create_temp_directory, get_temp_file_path
+
+
+def convert_video_to_playable_mp4(video_path : str) -> str:
+	video_file_size = get_file_size(video_path)
+	max_file_size = 512 * 1024 * 1024
+
+	if video_file_size > max_file_size:
+		create_temp_directory(video_path)
+		temp_video_path = get_temp_file_path(video_path)
+		commands = ffmpeg_builder.chain(
+			ffmpeg_builder.set_input(video_path),
+			ffmpeg_builder.set_video_duration(10),
+			ffmpeg_builder.force_output(temp_video_path)
+		)
+
+		process = run_ffmpeg(commands)
+		process.communicate()
+
+		if process.returncode == 0:
+			return temp_video_path
+	return video_path
+
+
+def check_allowed(path : str, check_in_upload_folder : bool) -> None:
+	return None