valknar
f24d138ab4
Sophisticated Python CLI for generating and manipulating images and video via the Freepik API, built with typer + rich. Commands: - generate-image: text-to-image with 8 models (flux-2-pro, mystic, seedream, etc.) - generate-video: image-to-video with 7 models (kling, minimax, runway, etc.) - generate-icon: text-to-icon in solid/outline/color/flat/sticker styles - upscale-image: 3 modes (precision-v2, precision, creative) + 2x/4x scale - upscale-video: standard/turbo modes - expand-image: outpainting with per-side pixel offsets - relight: AI-controlled relighting (Premium) - style-transfer: artistic style application (Premium) - describe-image: reverse-engineer an image into a prompt - config set/get/show/reset: configuration management Features: Rich Live polling panel, exponential backoff, --wait/--no-wait, auto-timestamped output filenames, streaming download with progress bar, FREEPIK_API_KEY env var support, venv-based setup. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
"""Video generation API methods."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any, Optional, Tuple
|
|
|
|
from freepik_cli.api.client import FreepikClient
|
|
from freepik_cli.api.models import (
|
|
VIDEO_POST_ENDPOINTS,
|
|
VIDEO_STATUS_ENDPOINTS,
|
|
VideoModel,
|
|
get_output_urls,
|
|
get_status,
|
|
get_task_id,
|
|
)
|
|
|
|
|
|
class VideoAPI:
|
|
def __init__(self, client: FreepikClient) -> None:
|
|
self._client = client
|
|
|
|
def generate(
|
|
self,
|
|
model: VideoModel,
|
|
image_b64: str,
|
|
prompt: Optional[str] = None,
|
|
duration: int = 5,
|
|
aspect_ratio: str = "16:9",
|
|
seed: Optional[int] = None,
|
|
) -> str:
|
|
"""Submit an image-to-video task. Returns task_id."""
|
|
payload: dict[str, Any] = {
|
|
"image": image_b64,
|
|
}
|
|
if prompt:
|
|
payload["prompt"] = prompt
|
|
if duration:
|
|
payload["duration"] = duration
|
|
if aspect_ratio:
|
|
payload["aspect_ratio"] = aspect_ratio
|
|
if seed is not None:
|
|
payload["seed"] = seed
|
|
|
|
endpoint = VIDEO_POST_ENDPOINTS[model]
|
|
raw = self._client.post(endpoint, json=payload)
|
|
return get_task_id(raw)
|
|
|
|
def get_status(self, model: VideoModel, task_id: str) -> Tuple[str, dict[str, Any]]:
|
|
"""Poll status. Returns (status_str, raw_response)."""
|
|
endpoint = VIDEO_STATUS_ENDPOINTS[model].format(task_id=task_id)
|
|
raw = self._client.get(endpoint)
|
|
return get_status(raw), raw
|
|
|
|
def get_output_urls(self, raw: dict[str, Any]) -> list[str]:
|
|
return get_output_urls(raw)
|