import torch
import gradio as gr
from demo_img import demo_img
from demo_vid import demo_vid
with gr.Blocks(css='style.css') as demo:
    gr.HTML(
        """
        <div style="text-align: center; max-width: 1200px; margin: 20px auto;">
        <h1 style="font-weight: 900; font-size: 2rem; margin: 0rem">
            AMT: All-Pairs Multi-Field Transforms for Efficient Frame Interpolation
        </h1>
        <h2 style="font-weight: 450; font-size: 1rem; margin: 0rem">
        <a href="https://paper99.github.io" style="color:blue;">Zhen Li</a><sup>1*</sup>, 
        <a href="https://github.com/NK-CS-ZZL" style="color:blue;">Zuo-Liang Zhu</a><sup>1*</sup>, 
        <a href="https://github.com/hlh981029" style="color:blue;">Ling-Hao Han</a><sup>1*</sup>, 
        <a href="https://houqb.github.io" style="color:blue;">Qibin Hou</a><sup>1*</sup>, 
        <a href="https://github.com" style="color:blue;">Chun-Le Guo</a><sup>1*</sup>, 
        <a href="https://mmcheng.net" style="color:blue;">Ming-Ming Cheng</a><sup>1*</sup>, 
        </h2>
        <h2 style="font-weight: 450; font-size: 1rem; margin: 0rem">
        <sup>1</sup>Nankai University <sup>*</sup> represents the equal contribution and <sup>#</sup> represents the corresponding author.
        </h2>
        <h2 style="font-weight: 450; font-size: 1rem; margin: 0rem">
        [<a href="https://arxiv.org/abs/2303.13439" style="color:blue;">arXiv</a>] 
        [<a href="https://github.com/MCG-NKU/AMT" style="color:blue;">GitHub</a>]
        </h2>
        </div>
        """)

    with gr.Tab('Img2Vid'):
        demo_img()
    with gr.Tab('VFI'):
        demo_vid()

demo.launch(debug=False)