deepfuze

2024-06-13 07:56:13 +05:30
commit 47d3520c19
184 changed files with 10075 additions and 0 deletions
@@ -0,0 +1,41 @@
+from typing import Optional
+import gradio
+
+import deepfuze.globals
+import deepfuze.choices
+from deepfuze import wording
+from deepfuze.typing import TempFrameFormat
+from deepfuze.filesystem import is_video
+from deepfuze.uis.core import get_ui_component
+
+TEMP_FRAME_FORMAT_DROPDOWN : Optional[gradio.Dropdown] = None
+
+
+def render() -> None:
+	global TEMP_FRAME_FORMAT_DROPDOWN
+
+	TEMP_FRAME_FORMAT_DROPDOWN = gradio.Dropdown(
+		label = wording.get('uis.temp_frame_format_dropdown'),
+		choices = deepfuze.choices.temp_frame_formats,
+		value = deepfuze.globals.temp_frame_format,
+		visible = is_video(deepfuze.globals.target_path)
+	)
+
+
+def listen() -> None:
+	TEMP_FRAME_FORMAT_DROPDOWN.change(update_temp_frame_format, inputs = TEMP_FRAME_FORMAT_DROPDOWN)
+	target_video = get_ui_component('target_video')
+	if target_video:
+		for method in [ 'upload', 'change', 'clear' ]:
+			getattr(target_video, method)(remote_update, outputs = TEMP_FRAME_FORMAT_DROPDOWN)
+
+
+def remote_update() -> gradio.Dropdown:
+	if is_video(deepfuze.globals.target_path):
+		return gradio.Dropdown(visible = True)
+	return gradio.Dropdown(visible = False)
+
+
+def update_temp_frame_format(temp_frame_format : TempFrameFormat) -> None:
+	deepfuze.globals.temp_frame_format = temp_frame_format
+