let5see/AI-Video
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

重命名frame_processor.py

Browse Source
This commit is contained in:
puke
2025-10-27 23:44:07 +08:00
committed by puke
parent be5b749913
commit e5e9b6e3ac
4 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
reelforge/service.py
View File

@@ -15,7 +15,7 @@ from reelforge.services.image import ImageService
from reelforge.services.narration_generator import NarrationGeneratorService from reelforge.services.narration_generator import NarrationGeneratorService
from reelforge.services.image_prompt_generator import ImagePromptGeneratorService from reelforge.services.image_prompt_generator import ImagePromptGeneratorService
from reelforge.services.title_generator import TitleGeneratorService 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 from reelforge.services.video_generator import VideoGeneratorService
@@ -70,7 +70,7 @@ class ReelForgeCore:
self.title_generator: Optional[TitleGeneratorService] = None self.title_generator: Optional[TitleGeneratorService] = None
# Frame processing services # Frame processing services
self.storyboard_processor: Optional[StoryboardProcessorService] = None self.frame_processor: Optional[FrameProcessor] = None
# Video generation service (named as verb for direct calling) # Video generation service (named as verb for direct calling)
self.generate_video: Optional[VideoGeneratorService] = None self.generate_video: Optional[VideoGeneratorService] = None
@@ -101,7 +101,7 @@ class ReelForgeCore:
self.title_generator = TitleGeneratorService(self) self.title_generator = TitleGeneratorService(self)
# 3. Initialize frame processing services # 3. Initialize frame processing services
self.storyboard_processor = StoryboardProcessorService(self) self.frame_processor = FrameProcessor(self)
# 4. Initialize video generation service # 4. Initialize video generation service
self.generate_video = VideoGeneratorService(self) self.generate_video = VideoGeneratorService(self)

4
reelforge/services/__init__.py
View File

@@ -12,7 +12,7 @@ from reelforge.services.video import VideoService
from reelforge.services.narration_generator import NarrationGeneratorService from reelforge.services.narration_generator import NarrationGeneratorService
from reelforge.services.image_prompt_generator import ImagePromptGeneratorService from reelforge.services.image_prompt_generator import ImagePromptGeneratorService
from reelforge.services.title_generator import TitleGeneratorService 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 from reelforge.services.video_generator import VideoGeneratorService
__all__ = [ __all__ = [
@@ -24,7 +24,7 @@ __all__ = [
"NarrationGeneratorService", "NarrationGeneratorService",
"ImagePromptGeneratorService", "ImagePromptGeneratorService",
"TitleGeneratorService", "TitleGeneratorService",
"StoryboardProcessorService", "FrameProcessor",
"VideoGeneratorService", "VideoGeneratorService",
] ]

9
reelforge/services/storyboard_processor.py → reelforge/services/frame_processor.py
View File

@@ -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 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 from reelforge.utils.os_util import get_temp_path
class StoryboardProcessorService: class FrameProcessor:
"""Storyboard processor service""" """Frame processor"""
def __init__(self, reelforge_core): def __init__(self, reelforge_core):
""" """
@@ -26,7 +26,7 @@ class StoryboardProcessorService:
""" """
self.core = reelforge_core self.core = reelforge_core
async def process_frame( async def __call__(
self, self,
frame: StoryboardFrame, frame: StoryboardFrame,
storyboard: 'Storyboard', storyboard: 'Storyboard',
@@ -45,6 +45,7 @@ class StoryboardProcessorService:
Args: Args:
frame: Storyboard frame to process frame: Storyboard frame to process
storyboard: Storyboard instance
config: Storyboard configuration config: Storyboard configuration
total_frames: Total number of frames in storyboard total_frames: Total number of frames in storyboard
progress_callback: Optional callback for progress updates (receives ProgressEvent) progress_callback: Optional callback for progress updates (receives ProgressEvent)

2
reelforge/services/video_generator.py
View File

@@ -325,7 +325,7 @@ class VideoGeneratorService:
frame_total=len(storyboard.frames) frame_total=len(storyboard.frames)
) )
processed_frame = await self.core.storyboard_processor.process_frame( processed_frame = await self.core.frame_processor(
frame=frame, frame=frame,
storyboard=storyboard, storyboard=storyboard,
config=config, config=config,