"""InfiniteTalk — powered by WaveSpeed AI.

Generated from _shared/apps.py — edit the spec there, not this file.
"""

import os

# Set before gradio is imported: its telemetry client reads these at import time.
os.environ.setdefault("GRADIO_ANALYTICS_ENABLED", "False")
os.environ.setdefault("HF_HUB_DISABLE_TELEMETRY", "1")

from pathlib import Path  # noqa: E402

import spaceui  # noqa: E402

SPEC = {
    "title": "InfiniteTalk",
    "model": "wavespeed-ai/infinitetalk",
    "output": "video",
    "tagline": "Drive a portrait from an audio track — lip sync and head motion.",
    "fields": [
        {
            "kind": "image",
            "key": "image",
            "label": "Input image"
        },
        {
            "kind": "audio",
            "key": "audio",
            "label": "Driving audio"
        },
        {
            "kind": "choice",
            "key": "resolution",
            "label": "Resolution",
            "choices": [
                "480p",
                "720p"
            ],
            "default": "480p"
        },
        {
            "kind": "seed",
            "key": "seed"
        }
    ],
    "campaign": "infinitetalk"
}

SPEC["css"] = Path(__file__).with_name("style.css").read_text(encoding="utf-8")

if __name__ == "__main__":
    spaceui.launch(spaceui.build(SPEC), SPEC["css"])