From e5e9b6e3acd1f23e05f34659a1dcbf3c4d4153f3 Mon Sep 17 00:00:00 2001 From: puke Date: Mon, 27 Oct 2025 23:44:07 +0800 Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E5=91=BD=E5=90=8Dframe=5Fprocessor.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- reelforge/service.py | 6 +++--- reelforge/services/__init__.py | 4 ++-- .../{storyboard_processor.py => frame_processor.py} | 9 +++++---- reelforge/services/video_generator.py | 2 +- 4 files changed, 11 insertions(+), 10 deletions(-) rename reelforge/services/{storyboard_processor.py => frame_processor.py} (98%) diff --git a/reelforge/service.py b/reelforge/service.py index 49473d6..9a268bd 100644 --- a/reelforge/service.py +++ b/reelforge/service.py @@ -15,7 +15,7 @@ from reelforge.services.image import ImageService from reelforge.services.narration_generator import NarrationGeneratorService from reelforge.services.image_prompt_generator import ImagePromptGeneratorService from reelforge.services.title_generator import TitleGeneratorService -from reelforge.services.storyboard_processor import StoryboardProcessorService +from reelforge.services.frame_processor import FrameProcessor from reelforge.services.video_generator import VideoGeneratorService @@ -70,7 +70,7 @@ class ReelForgeCore: self.title_generator: Optional[TitleGeneratorService] = None # Frame processing services - self.storyboard_processor: Optional[StoryboardProcessorService] = None + self.frame_processor: Optional[FrameProcessor] = None # Video generation service (named as verb for direct calling) self.generate_video: Optional[VideoGeneratorService] = None @@ -101,7 +101,7 @@ class ReelForgeCore: self.title_generator = TitleGeneratorService(self) # 3. Initialize frame processing services - self.storyboard_processor = StoryboardProcessorService(self) + self.frame_processor = FrameProcessor(self) # 4. Initialize video generation service self.generate_video = VideoGeneratorService(self) diff --git a/reelforge/services/__init__.py b/reelforge/services/__init__.py index b749fdb..77f65da 100644 --- a/reelforge/services/__init__.py +++ b/reelforge/services/__init__.py @@ -12,7 +12,7 @@ from reelforge.services.video import VideoService from reelforge.services.narration_generator import NarrationGeneratorService from reelforge.services.image_prompt_generator import ImagePromptGeneratorService from reelforge.services.title_generator import TitleGeneratorService -from reelforge.services.storyboard_processor import StoryboardProcessorService +from reelforge.services.frame_processor import FrameProcessor from reelforge.services.video_generator import VideoGeneratorService __all__ = [ @@ -24,7 +24,7 @@ __all__ = [ "NarrationGeneratorService", "ImagePromptGeneratorService", "TitleGeneratorService", - "StoryboardProcessorService", + "FrameProcessor", "VideoGeneratorService", ] diff --git a/reelforge/services/storyboard_processor.py b/reelforge/services/frame_processor.py similarity index 98% rename from reelforge/services/storyboard_processor.py rename to reelforge/services/frame_processor.py index 09ffde2..f298b9d 100644 --- a/reelforge/services/storyboard_processor.py +++ b/reelforge/services/frame_processor.py @@ -1,5 +1,5 @@ """ -Storyboard processor - Process single frame through complete pipeline +Frame processor - Process single frame through complete pipeline Orchestrates: TTS → Image Generation → Frame Composition → Video Segment """ @@ -14,8 +14,8 @@ from reelforge.models.storyboard import Storyboard, StoryboardFrame, StoryboardC from reelforge.utils.os_util import get_temp_path -class StoryboardProcessorService: - """Storyboard processor service""" +class FrameProcessor: + """Frame processor""" def __init__(self, reelforge_core): """ @@ -26,7 +26,7 @@ class StoryboardProcessorService: """ self.core = reelforge_core - async def process_frame( + async def __call__( self, frame: StoryboardFrame, storyboard: 'Storyboard', @@ -45,6 +45,7 @@ class StoryboardProcessorService: Args: frame: Storyboard frame to process + storyboard: Storyboard instance config: Storyboard configuration total_frames: Total number of frames in storyboard progress_callback: Optional callback for progress updates (receives ProgressEvent) diff --git a/reelforge/services/video_generator.py b/reelforge/services/video_generator.py index c8bb59c..31bc094 100644 --- a/reelforge/services/video_generator.py +++ b/reelforge/services/video_generator.py @@ -325,7 +325,7 @@ class VideoGeneratorService: frame_total=len(storyboard.frames) ) - processed_frame = await self.core.storyboard_processor.process_frame( + processed_frame = await self.core.frame_processor( frame=frame, storyboard=storyboard, config=config,