2023-09-24 21:36:57 +08:00
|
|
|
import os
|
2024-09-10 07:07:58 +08:00
|
|
|
from typing import List, Dict, Any
|
2023-09-24 21:36:57 +08:00
|
|
|
|
|
|
|
ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
|
2024-10-14 22:16:48 +08:00
|
|
|
WORKFLOW_DIR = os.path.join(ROOT_DIR, "workflow")
|
2023-09-24 21:36:57 +08:00
|
|
|
|
|
|
|
file_types = [
|
2024-10-14 22:16:48 +08:00
|
|
|
("Image", ("*.png", "*.jpg", "*.jpeg", "*.gif", "*.bmp")),
|
|
|
|
("Video", ("*.mp4", "*.mkv")),
|
2023-09-24 21:36:57 +08:00
|
|
|
]
|
|
|
|
|
2024-09-10 07:07:58 +08:00
|
|
|
souce_target_map = []
|
2024-09-10 16:58:33 +08:00
|
|
|
simple_map = {}
|
2024-09-10 07:07:58 +08:00
|
|
|
|
2023-09-24 21:36:57 +08:00
|
|
|
source_path = None
|
|
|
|
target_path = None
|
|
|
|
output_path = None
|
|
|
|
frame_processors: List[str] = []
|
2024-10-14 23:02:17 +08:00
|
|
|
keep_fps = True # Initialize with default value
|
|
|
|
keep_audio = True # Initialize with default value
|
|
|
|
keep_frames = False # Initialize with default value
|
|
|
|
many_faces = False # Initialize with default value
|
|
|
|
map_faces = False # Initialize with default value
|
|
|
|
color_correction = False # Initialize with default value
|
|
|
|
nsfw_filter = False # Initialize with default value
|
2023-09-24 21:36:57 +08:00
|
|
|
video_encoder = None
|
|
|
|
video_quality = None
|
2024-10-14 23:02:17 +08:00
|
|
|
live_mirror = False # Initialize with default value
|
|
|
|
live_resizable = False # Initialize with default value
|
2023-09-24 21:36:57 +08:00
|
|
|
max_memory = None
|
|
|
|
execution_providers: List[str] = []
|
|
|
|
execution_threads = None
|
|
|
|
headless = None
|
2024-10-14 22:16:48 +08:00
|
|
|
log_level = "error"
|
2024-10-14 23:02:17 +08:00
|
|
|
fp_ui: Dict[str, bool] = {"face_enhancer": False} # Initialize with default value
|
2023-12-14 14:59:22 +08:00
|
|
|
camera_input_combobox = None
|
2024-10-04 14:41:22 +08:00
|
|
|
webcam_preview_running = False
|
2024-10-14 23:02:17 +08:00
|
|
|
show_fps = False # Initialize with default value
|
2024-10-28 13:26:01 +08:00
|
|
|
mouth_mask = False
|
|
|
|
show_mouth_mask_box = False
|
|
|
|
mask_down_size = 0.5
|
|
|
|
mask_size = 1.0
|
|
|
|
mask_feather_ratio = 8
|
2024-10-28 14:15:50 +08:00
|
|
|
opacity_switch = False
|
|
|
|
face_opacity = 100
|
2024-10-28 16:20:44 +08:00
|
|
|
selected_camera = None
|