重命名frame_processor.py
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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",
|
||||
]
|
||||
|
||||
|
||||
@@ -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)
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user