Upload 14 files

app.py

@@ -25,7 +25,7 @@ with gr.Blocks(title="Nova Reel — Text → Video (Single-shot)") as demo:
         with gr.Column(scale=1):
             s3_uri = gr.Textbox(
                 label="S3 output base URI",
-                value="s3://bucketvideogenerationwest",
+                value="s3://r2-video-test",
                 placeholder="s3://your-bucket[/optional/prefix]",
             )
             region = gr.Textbox(label="AWS Region", value=DEFAULT_REGION)
@@ -70,4 +70,8 @@ with gr.Blocks(title="Nova Reel — Text → Video (Single-shot)") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue().launch(share=True, ssr_mode=False)
+    auth_user = os.getenv("USER")
+    auth_pass = os.getenv("PASS")
+    demo.queue().launch(
+        auth=(auth_user, auth_pass), share=True,  ssr_mode=False
+    )

drivers/nova_reel.py

@@ -1,7 +1,27 @@
+"""
+Module for interacting with AWS Bedrock and S3 to generate videos from text prompts.
+
+Functions:
+    bedrock_and_s3(region_name: str) -> tuple:
+        Creates and returns AWS Bedrock Runtime and S3 clients for the specified region.
+
+    build_single_shot_input(prompt: str, seed: int | None, reference_image_path: str | None) -> dict:
+        Builds the input payload for a single-shot text-to-video generation request.
+        Optionally includes a seed for reproducibility and a reference image.
+
+    start_async_single_shot(bedrock_client, prompt: str, seed: int | None,
+                            reference_image_path: str | None, s3_output_uri: str) -> str:
+        Initiates an asynchronous text-to-video generation request using Bedrock.
+        Returns the invocation ARN for tracking the request.
+
+    get_async(bedrock_client, invocation_arn: str) -> dict:
+        Retrieves the status and result of an asynchronous Bedrock invocation using the provided ARN.
+"""
 import boto3
 from config import MODEL_ID, FIXED_FPS, FIXED_WIDTH, FIXED_HEIGHT, FIXED_DURATION
 from utils.images import image_path_to_image_source
 
+
 def bedrock_and_s3(region_name: str):
     return (
         boto3.client("bedrock-runtime", region_name=region_name),

drivers/polling.py

@@ -1,8 +1,29 @@
+"""
+Polls the status of an asynchronous Bedrock job and fetches the output video upon completion.
+
+Args:
+    bedrock_runtime: The Bedrock runtime client used to poll the job status.
+    s3_client: The S3 client used to download the output video.
+    invocation_arn (str): The ARN of the asynchronous Bedrock job to poll.
+
+Yields:
+    tuple:
+        - local_mp4 (str or None): Path to the downloaded MP4 file if the job is completed successfully, otherwise None.
+        - message (str): Status or error message describing the current state or any issues encountered.
+
+Behavior:
+    - Continuously polls the job status at intervals defined by POLL_INTERVAL_SECS.
+    - Yields status updates when the job status changes.
+    - On completion, attempts to download the output MP4 from S3 and yields the local file path.
+    - Handles and yields error messages for polling failures, download failures, or job failures.
+"""
+
 import time
 from botocore.exceptions import BotoCoreError, ClientError
 from config import POLL_INTERVAL_SECS
 from VideoBR.utils.s3_io import download_output_mp4
 
+
 def poll_and_fetch(bedrock_runtime, s3_client, invocation_arn: str):
     last_status = None
     while True:

handlers/video_generation_type.py

@@ -1,3 +1,38 @@
+
+"""
+Module for handling video generation using Bedrock and S3.
+
+Functions:
+    generate_single_shot(prompt: str, output_s3_uri: str, seed_in: str, region_name: str,
+                         reference_image_path: str | None, custom_name: str)
+        Asynchronously generates a single-shot video based on the provided prompt and optional reference image.
+        The video is saved to the specified S3 URI, and the function yields status updates and results.
+        Handles job submission, polling for status, renaming the output file, and generating a presigned URL for playback.
+
+        Args:
+            prompt (str): The text prompt for video generation.
+            output_s3_uri (str): The S3 URI where the output video will be stored.
+            seed_in (str): Seed value for deterministic generation.
+            region_name (str): AWS region name for Bedrock and S3 clients.
+            reference_image_path (str | None): Optional path to a reference image.
+            custom_name (str): Custom name for the output video file.
+
+        Yields:
+            tuple: (vid_path, status_msg, rows) where
+                vid_path (str | None): Presigned URL for the generated video (if available).
+                status_msg (str): Status or error message.
+                rows (list): List of video metadata rows from S3.
+
+    refresh_listing(output_s3_uri: str, region_name: str)
+        Retrieves and returns a list of video metadata rows from the specified S3 URI.
+
+        Args:
+            output_s3_uri (str): The S3 URI to list videos from.
+            region_name (str): AWS region name for S3 client.
+
+        Returns:
+            list: List of video metadata rows from S3, or an empty list on error.
+"""
 import time
 from datetime import datetime
 from botocore.exceptions import BotoCoreError, ClientError

utils/images.py

@@ -1,3 +1,16 @@
+"""
+Converts an image file path to a dictionary containing the image format and a base64-encoded source.
+
+Args:
+    image_path (str | None): The file path to the image. Can be None.
+
+Returns:
+    dict | None: A dictionary with keys 'format' (either 'jpeg' or 'png') and 'source' (containing base64-encoded image bytes),
+                 or None if image_path is None.
+
+Raises:
+    ValueError: If the image file is not a PNG or JPEG.
+"""
 import base64
 from pathlib import Path