@leejungkiin/awkit 1.3.8 → 1.4.2

package/skills/short-maker/scripts/google-flow-cli/gflow/api/models.py

@@ -0,0 +1,64 @@
+"""
+Data models for Google Flow API.
+
+These models represent the request/response structures for Flow's
+image and video generation endpoints, based on the reverse-engineered
+aisandbox-pa.googleapis.com API.
+"""
+
+from __future__ import annotations
+
+from enum import Enum
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+
+class AssetType(str, Enum):
+    """Type of generated asset."""
+    IMAGE = "image"
+    VIDEO = "video"
+    UNKNOWN = "unknown"
+
+
+class Asset(BaseModel):
+    """A generated image or video asset from Google Flow."""
+    id: str = ""
+    name: str = ""
+    asset_type: AssetType = AssetType.UNKNOWN
+    prompt: str = ""
+    url: str = ""  # CDN URL (for videos)
+    width: int = 0
+    height: int = 0
+    duration_seconds: float = 0.0  # For videos
+    model: str = ""
+    raw: dict[str, Any] = Field(default_factory=dict)  # Raw API response
+
+
+class GenerateImageRequest(BaseModel):
+    """Request to generate images using Imagen 4."""
+    prompt: str
+    aspect_ratio: str = "landscape"  # square, portrait, landscape, 4:3
+    seed: int | None = None
+    num_images: int = 1  # Number of images to generate (1-8)
+
+
+class GenerateVideoRequest(BaseModel):
+    """Request to generate a video using Veo 3.1."""
+    prompt: str
+    aspect_ratio: str = "landscape"
+    duration: str = "short"  # "short", "medium", "long"
+    seed: int | None = None
+    start_image: str | None = None  # Path to image for image-to-video (starting frame)
+
+
+class ExtendVideoRequest(BaseModel):
+    """Request to extend a video using Veo 3.1 extend.
+
+    Takes an existing video's media ID and generates a continuation.
+    """
+    prompt: str  # Describe what happens next
+    media_id: str  # Media ID of the video to extend
+    aspect_ratio: str = "landscape"
+    workflow_id: str = ""  # Flow workflow ID (optional, for project continuity)
+    seed: int | None = None

package/skills/short-maker/scripts/google-flow-cli/gflow/api/rpc_ids.py

@@ -0,0 +1,98 @@
+"""
+RPC endpoint IDs for Google Flow.
+
+These IDs are used in the BatchExecute protocol to identify which
+server-side method to call. They must be discovered by inspecting
+network traffic when using flow.google in a browser.
+
+HOW TO DISCOVER THESE IDs:
+1. Open flow.google in Chrome
+2. Open DevTools > Network tab
+3. Filter by "batchexecute"
+4. Perform an action (e.g., generate an image)
+5. Look at the request payload for the "rpcids" parameter
+6. The value is the RPC ID for that action
+
+The IDs below are PLACEHOLDERS. Replace them with the real values
+you discover from network inspection. The format is typically
+a short alphanumeric string like "xYz123" or "AbCdEf".
+
+tmc/nlm uses the same approach — their RPC IDs were reverse-engineered
+from NotebookLM's web traffic.
+"""
+
+# =============================================================
+# IMPORTANT: These are PLACEHOLDER IDs.
+# You MUST replace them with real IDs from network inspection.
+# =============================================================
+
+# --- Image Generation ---
+GENERATE_IMAGE = "PLACEHOLDER_GENERATE_IMAGE"
+# Generates images using Imagen 4
+
+# --- Video Generation ---
+GENERATE_VIDEO = "PLACEHOLDER_GENERATE_VIDEO"
+# Generates videos using Veo 3.1
+
+# --- Asset Management ---
+LIST_ASSETS = "PLACEHOLDER_LIST_ASSETS"
+# Lists all assets in the user's library
+
+GET_ASSET = "PLACEHOLDER_GET_ASSET"
+# Gets details for a single asset
+
+DELETE_ASSET = "PLACEHOLDER_DELETE_ASSET"
+# Deletes an asset
+
+# --- Collections ---
+LIST_COLLECTIONS = "PLACEHOLDER_LIST_COLLECTIONS"
+CREATE_COLLECTION = "PLACEHOLDER_CREATE_COLLECTION"
+ADD_TO_COLLECTION = "PLACEHOLDER_ADD_TO_COLLECTION"
+REMOVE_FROM_COLLECTION = "PLACEHOLDER_REMOVE_FROM_COLLECTION"
+DELETE_COLLECTION = "PLACEHOLDER_DELETE_COLLECTION"
+
+# --- Editing ---
+EDIT_IMAGE = "PLACEHOLDER_EDIT_IMAGE"
+# Lasso tool / inpainting edits
+
+EXTEND_VIDEO = "PLACEHOLDER_EXTEND_VIDEO"
+# Extend a video clip
+
+FRAMES_TO_VIDEO = "PLACEHOLDER_FRAMES_TO_VIDEO"
+# Create video bridging two images
+
+# --- Account ---
+GET_ACCOUNT = "PLACEHOLDER_GET_ACCOUNT"
+# Get account info and quota
+
+# --- App Config ---
+GET_APP_CONFIG = "PLACEHOLDER_GET_APP_CONFIG"
+# Loaded during initial page load, contains feature flags and model info
+
+# =============================================================
+# Discovery helper
+# =============================================================
+
+# Mapping of friendly names to RPC IDs for CLI discovery mode
+ALL_RPC_IDS = {
+    "generate_image": GENERATE_IMAGE,
+    "generate_video": GENERATE_VIDEO,
+    "list_assets": LIST_ASSETS,
+    "get_asset": GET_ASSET,
+    "delete_asset": DELETE_ASSET,
+    "list_collections": LIST_COLLECTIONS,
+    "create_collection": CREATE_COLLECTION,
+    "add_to_collection": ADD_TO_COLLECTION,
+    "remove_from_collection": REMOVE_FROM_COLLECTION,
+    "delete_collection": DELETE_COLLECTION,
+    "edit_image": EDIT_IMAGE,
+    "extend_video": EXTEND_VIDEO,
+    "frames_to_video": FRAMES_TO_VIDEO,
+    "get_account": GET_ACCOUNT,
+    "get_app_config": GET_APP_CONFIG,
+}
+
+
+def is_placeholder(rpc_id: str) -> bool:
+    """Check if an RPC ID is still a placeholder."""
+    return rpc_id.startswith("PLACEHOLDER_")

package/skills/short-maker/scripts/google-flow-cli/gflow/auth/__init__.py

@@ -0,0 +1,15 @@
+"""Browser-based authentication for Google Flow."""
+
+from .browser_auth import (
+    BrowserAuth, AuthData, AuthError,
+    load_env, save_env, refresh_access_token,
+    refresh_cookies_from_cdp,
+    kill_auth_browser, get_saved_cdp_port,
+)
+
+__all__ = [
+    "BrowserAuth", "AuthData", "AuthError",
+    "load_env", "save_env", "refresh_access_token",
+    "refresh_cookies_from_cdp",
+    "kill_auth_browser", "get_saved_cdp_port",
+]