Update roop/globals.py

roop/globals.py

@@ -1,17 +1,76 @@
-from typing import List
+from typing import List, Optional
 
-source_path = None
-target_path = None
-output_path = None
+# Global configuration variables
+source_path: Optional[str] = None
+target_path: Optional[str] = None
+output_path: Optional[str] = None
 frame_processors: List[str] = []
-keep_fps = None
-keep_audio = None
-keep_frames = None
-many_faces = None
-video_encoder = None
-video_quality = None
-max_memory = None
+keep_fps: Optional[bool] = None
+keep_audio: Optional[bool] = None
+keep_frames: Optional[bool] = None
+many_faces: Optional[bool] = None
+video_encoder: Optional[str] = None
+video_quality: Optional[int] = None
+max_memory: Optional[int] = None
 execution_providers: List[str] = []
-execution_threads = None
-headless = None
-log_level = 'error'
+execution_threads: Optional[int] = None
+headless: Optional[bool] = None
+log_level: str = 'error'
+
+
+def initialize_globals(
+    source_path: Optional[str] = None,
+    target_path: Optional[str] = None,
+    output_path: Optional[str] = None,
+    frame_processors: List[str] = [],
+    keep_fps: Optional[bool] = None,
+    keep_audio: Optional[bool] = None,
+    keep_frames: Optional[bool] = None,
+    many_faces: Optional[bool] = None,
+    video_encoder: Optional[str] = None,
+    video_quality: Optional[int] = None,
+    max_memory: Optional[int] = None,
+    execution_providers: List[str] = [],
+    execution_threads: Optional[int] = None,
+    headless: Optional[bool] = None,
+    log_level: str = 'error'
+) -> None:
+    """
+    Initialize or reset global variables.
+
+    Args:
+        source_path (Optional[str]): Path to the source file.
+        target_path (Optional[str]): Path to the target file.
+        output_path (Optional[str]): Path to the output file.
+        frame_processors (List[str]): List of frame processors.
+        keep_fps (Optional[bool]): Whether to keep the original FPS.
+        keep_audio (Optional[bool]): Whether to keep the original audio.
+        keep_frames (Optional[bool]): Whether to keep temporary frames.
+        many_faces (Optional[bool]): Whether to process every face.
+        video_encoder (Optional[str]): Video encoder to use.
+        video_quality (Optional[int]): Video quality setting.
+        max_memory (Optional[int]): Maximum amount of memory in GB.
+        execution_providers (List[str]): List of execution providers.
+        execution_threads (Optional[int]): Number of execution threads.
+        headless (Optional[bool]): Whether to run in headless mode.
+        log_level (str): Log level to use.
+    """
+    global source_path, target_path, output_path, frame_processors, keep_fps, keep_audio
+    global keep_frames, many_faces, video_encoder, video_quality, max_memory, execution_providers
+    global execution_threads, headless, log_level
+
+    source_path = source_path
+    target_path = target_path
+    output_path = output_path
+    frame_processors = frame_processors
+    keep_fps = keep_fps
+    keep_audio = keep_audio
+    keep_frames = keep_frames
+    many_faces = many_faces
+    video_encoder = video_encoder
+    video_quality = video_quality
+    max_memory = max_memory
+    execution_providers = execution_providers
+    execution_threads = execution_threads
+    headless = headless
+    log_level = log_level