Update roop/globals.py

roop/globals.py

@@ -1,76 +1,74 @@
-from typing import List, Optional
+# globals.py
 
-# Global configuration variables
-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'
+# Module-level variables
+source_path = None
+target_path = None
+output_path = None
+frame_processors = []
+keep_fps = True
+keep_audio = True
+video_encoder = 'libx264'  # Example value
+video_quality = 23  # Example value
+log_level = 'error'  # Example value
 
+def set_global_paths(src: str, tgt: str, out: str) -> None:
+    """
+    Set global paths for source, target, and output.
+
+    Args:
+        src (str): Source path.
+        tgt (str): Target path.
+        out (str): Output path.
+    """
+    global source_path, target_path, output_path
+    source_path = src
+    target_path = tgt
+    output_path = out
 
-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:
+def set_processing_options(encoder: str, quality: int, log: str) -> None:
     """
-    Initialize or reset global variables.
+    Set options for video processing.
 
     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.
+        encoder (str): Video encoder.
+        quality (int): Video quality.
+        log (str): Log level.
     """
+    global video_encoder, video_quality, log_level
+    video_encoder = encoder
+    video_quality = quality
+    log_level = log
+
+def set_frame_processors(processors: list) -> None:
+    """
+    Set the frame processors.
+
+    Args:
+        processors (list): List of frame processors.
+    """
+    global frame_processors
+    frame_processors = processors
+
+def set_keep_options(fps: bool, audio: bool) -> None:
+    """
+    Set options to keep FPS and audio.
+
+    Args:
+        fps (bool): Whether to keep FPS.
+        audio (bool): Whether to keep audio.
+    """
+    global keep_fps, keep_audio
+    keep_fps = fps
+    keep_audio = audio
+
+# Example usage of global variables
+def some_function():
     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
+    # Function implementation using global variables
+    print(f"Source Path: {source_path}")
+    print(f"Target Path: {target_path}")
+    print(f"Output Path: {output_path}")
+    print(f"Frame Processors: {frame_processors}")
+    print(f"Keep FPS: {keep_fps}")
+    print(f"Keep Audio: {keep_audio}")
 
-    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