#!/usr/bin/env python

import gradio as gr

from gradio_model3dgs import Model3DGS

DESCRIPTION = """
- [GitHub](https://github.com/dylanebert/gradio-splatting)
- [PyPI](https://pypi.org/project/gradio-model3dgs/)
- Examples were made with [this Space](https://huggingface.co/spaces/vast-ai/TriplaneGaussian).
"""

with gr.Blocks() as demo:
    model3d = Model3DGS()
    gr.Examples(
        examples=[
            "assets/green_parrot.splat",
            "assets/a_cat_dressed_as_the_pope.splat",
            "assets/a_purple_winter_jacket.splat",
            "assets/an_otter_wearing_sunglasses.splat",
        ],
        inputs=model3d,
    )

    gr.Markdown(DESCRIPTION)


if __name__ == "__main__":
    demo.queue().launch()