from pathlib import Path

from gradio_client import Client


def main():
    client = Client("yingzhac/LingBot_Video_demo")
    video, info, caption = client.predict(
        "A humanoid robot walks through a sunlit garden, realistic motion and materials",
        "",
        "",
        "Landscape 16:9 (832 x 480)",
        "Tracking",
        2,
        16,
        30,
        3.0,
        42,
        False,
        api_name="/generate_video",
    )
    output = Path("lingbot_output.mp4")
    output.write_bytes(Path(video).read_bytes())
    print(info)
    print(caption)
    print(f"Saved: {output.resolve()}")


if __name__ == "__main__":
    main()