import os
import gradio as gr
from demo_img import demo_img
from demo_vid import demo_vid

SPACE_ID = os.getenv('SPACE_ID')
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>]
        [<a href="https://github.com/MCG-NKU/AMT" style="color:blue;">Colab</a>]
        [<a href="https://github.com/MCG-NKU/AMT" style="color:blue;">Replicate</a>]
        </h2>
        <p>For faster inference without waiting in queue, you may duplicate the space and upgrade to GPU in settings. 
        """  
        f'<a href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true">'
        """ 
        <img style="display: inline; margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space" /></a>
        </p>
        </div>
        """
        )

    with gr.Tab('Img2Vid'):
        demo_img()
    with gr.Tab('VFI'):
        demo_vid()
        
    gr.HTML(
        """
        <div style="text-align: center; max-width: 1200px; margin: 20px auto;">
        <h2 style="font-weight: 450; font-size: 1rem; margin: 0rem">
        Licensed under the Creative Commons Attribution-NonCommercial 4.0 International for Non-commercial use only. 
        Any commercial use should get a formal permission first.
        </h2>
        </div>
        """
        )

demo.launch(debug=False)