Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .python-version
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
3.12
11 changes: 11 additions & 0 deletions agents/camera_image_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -116,6 +116,14 @@ def __init__(
self.chat_model = chat_model
self.image_generator = image_generator
self.video_generator = video_generator
self._resolution = "1080p"
self._aspect_ratio = "16:9"
self._shot_duration = 8

def set_video_params(self, resolution: str, aspect_ratio: str, shot_duration: int) -> None:
self._resolution = resolution
self._aspect_ratio = aspect_ratio
self._shot_duration = shot_duration


async def construct_camera_tree(
Expand Down Expand Up @@ -164,6 +172,9 @@ async def generate_transition_video(
video_output = await self.video_generator.generate_single_video(
prompt=prompt,
reference_image_paths=reference_image_paths,
resolution=self._resolution,
aspect_ratio=self._aspect_ratio,
duration=self._shot_duration,
)
return video_output

Expand Down
Loading