import os
import gradio as gr
from glob import glob


def get_inputs_components():
    return [
            gr.Video(
                label      = 'INPUT VIDEO',
                show_label = True,
            ),
            gr.Radio(
                choices = ['Static Camera', 'Dynamic Camera'],
                label   = 'Camera Status',
                info    = 'If the camera is static, DPVO will be skipped.'
            ),
            gr.Number(
                value = 60,
                label = 'GPU quota',
                info  = 'Decrease this value if you have insufficient GPU quota left. (Allocating too less may cause wasting of GPU quota while allocating too much won\'t. We suggest to set it as much as possible.)',
            ),
        ]


def get_outputs_components():
    return [
            gr.PlayableVideo(
                label      = 'INCAM RESULT',
                show_label = True,
            ),
            gr.PlayableVideo(
                label      = 'GLOBAL RESULT',
                show_label = True,
            ),
        ]


def get_examples():
    REPO_ROOT = str(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

    return [
        [os.path.join(REPO_ROOT, 'examples/cxk.mp4'), 'Static Camera', 120],
        [os.path.join(REPO_ROOT, 'examples/tennis.mp4'), 'Static Camera', 120],
    ]


def get_desc():
    return '''
        <center><b><font size='10'><font color='78aa58'>G</font><font color='6589bf'>V</font>HMR</font></b></center>

        <center><b><font size='5'> World-Grounded Human Motion Recovery via <font color='78aa58'>Gravity</font>-<font color='6589bf'>View</font> <font color='c68b5b'>Coordinates</font></font></b></center>

        <center><b>
        <a href=https://zju3dv.github.io/gvhmr>Project Page</a>
        |
        <a href=https://arxiv.org/abs/2409.06662>Paper</a>
        </b></center>

        > World-Grounded Human Motion Recovery via Gravity-View Coordinates  
        > [Zehong Shen](https://zehongs.github.io/)<sup>\*</sup>,
        [Huaijin Pi](https://phj128.github.io/)<sup>\*</sup>,
        [Yan Xia](https://isshikihugh.github.io/scholar),
        [Zhi Cen](https://scholar.google.com/citations?user=Xyy-uFMAAAAJ),
        [Sida Peng](https://pengsida.net/)<sup>†</sup>,
        [Zechen Hu](https://zju3dv.github.io/gvhmr),
        [Hujun Bao](http://www.cad.zju.edu.cn/home/bao/),
        [Ruizhen Hu](https://csse.szu.edu.cn/staff/ruizhenhu/),
        [Xiaowei Zhou](https://xzhou.me/)  
        > SIGGRAPH Asia 2024
        
        **Tips: since the GPU quota is limited, we suggest to use short videos (< 5s) for demo.** For longer videos or moving camera videos, please refer to the [colab demo](https://colab.research.google.com/drive/1N9WSchizHv2bfQqkE9Wuiegw_OT7mtGj?usp=sharing).
        '''