Update tools/video_encode_tool.py

tools/video_encode_tool.py

@@ -2,24 +2,28 @@
 #
 # Copyright (C) August 4, 2025  Carlos Rodrigues dos Santos
 #
-# Version: 1.1.0
+# Version: 1.1.1
 #
 # This file defines the VideoEncodeTool specialist. Its purpose is to abstract away
 # the underlying command-line tools (like FFmpeg) used for video manipulation tasks
 # such as concatenation and creating transitions. By encapsulating this logic, the core
-# Deformes4D engine can remain agnostic to the specific tool being used.
+# Deformes4D engine can remain agnostic to the specific tool being used, allowing for easier
+# maintenance and future replacement with other libraries or tools.
 
 import os
 import subprocess
 import logging
-import gradio as gr
-from typing import List, Optional, Tuple
 import random
 import time
 import shutil
+from typing import List, Optional, Tuple
 
 logger = logging.getLogger(__name__)
 
+class VideoToolError(Exception):
+    """Custom exception for errors originating from the VideoEncodeTool."""
+    pass
+
 class VideoEncodeTool:
     """
     A specialist for handling video encoding and manipulation tasks.
@@ -45,6 +49,9 @@ class VideoEncodeTool:
 
         Returns:
             str: The file path to the generated transition video clip.
+            
+        Raises:
+            VideoToolError: If the FFmpeg command fails.
         """
         output_path = os.path.join(workspace_dir, f"bridge_{int(time.time())}.mp4")
         width, height = target_resolution
@@ -57,19 +64,8 @@ class VideoEncodeTool:
         
         selected_effect = effect if effect and effect.strip() else random.choice(fade_effects)
         
-        # The duration of each image loop and the xfade itself should match the total desired duration.
-        transition_duration = max(0.1, duration) # Ensure duration is not zero
-
-        # Construct the FFmpeg command
-        # -v error: Suppress all console output except for errors.
-        # -loop 1: Loop the input image.
-        # -t {duration}: Set the duration for the looped image.
-        # -filter_complex: Defines a complex filtergraph.
-        #   - scale,setsar: Pre-process each image to the target resolution and aspect ratio.
-        #   - xfade: Apply the crossfade transition.
-        #     - offset=0: Start the transition immediately.
-        # -map "[out]": Map the output of the filtergraph to the final video.
-        # -c:v libx264 -r {fps} -pix_fmt yuv420p: Standard high-compatibility video encoding settings.
+        transition_duration = max(0.1, duration)
+
         cmd = (
             f"ffmpeg -y -v error -loop 1 -t {transition_duration} -i \"{start_image_path}\" -loop 1 -t {transition_duration} -i \"{end_image_path}\" "
             f"-filter_complex \"[0:v]scale={width}:{height},setsar=1[v0];[1:v]scale={width}:{height},setsar=1[v1];"
@@ -85,7 +81,7 @@ class VideoEncodeTool:
             logger.error(f"FFmpeg bridge creation failed. Return code: {e.returncode}")
             logger.error(f"FFmpeg command: {cmd}")
             logger.error(f"FFmpeg stderr: {e.stderr}")
-            raise gr.Error(f"Failed to create transition video. Details: {e.stderr}")
+            raise VideoToolError(f"Failed to create transition video. Details: {e.stderr}")
             
         return output_path
 
@@ -97,13 +93,15 @@ class VideoEncodeTool:
             video_paths (List[str]): A list of absolute paths to the video clips to be concatenated.
             output_path (str): The absolute path for the final output video.
             workspace_dir (str): The directory to use for temporary files, like the concat list.
+            
+        Raises:
+            VideoToolError: If no video paths are provided or if the FFmpeg command fails.
         """
         if not video_paths:
-            raise gr.Error("VideoEncodeTool: No video fragments provided for concatenation.")
+            raise VideoToolError("VideoEncodeTool: No video fragments provided for concatenation.")
 
         if len(video_paths) == 1:
             logger.info("Only one video clip found. Skipping concatenation and just copying the file.")
-            # If there's only one clip, a simple copy is much faster.
             shutil.copy(video_paths[0], output_path)
             return
 
@@ -125,14 +123,15 @@ class VideoEncodeTool:
         except subprocess.CalledProcessError as e:
             logger.error(f"FFmpeg concatenation failed. Return code: {e.returncode}")
             logger.error(f"FFmpeg stderr: {e.stderr}")
-            raise gr.Error(f"Failed to assemble the final video using FFmpeg. Details: {e.stderr}")
+            raise VideoToolError(f"Failed to assemble the final video using FFmpeg. Details: {e.stderr}")
         except Exception as e:
             logger.error(f"An unexpected error occurred during video concatenation: {e}", exc_info=True)
-            raise gr.Error("An unexpected error occurred during the final video assembly.")
+            raise VideoToolError("An unexpected error occurred during the final video assembly.")
         finally:
             if os.path.exists(list_file_path):
                 os.remove(list_file_path)
 
 
 # --- Singleton Instance ---
-# We create a single instance of the tool to be imported by other modules.
+# We create a single instance of the tool to be imported by other modules.
+video_encode_tool_singleton = VideoEncodeTool()