import gradio as gr

from video_diffusion.damo.damo_text2_video import DamoText2VideoGenerator
from video_diffusion.inpaint_zoom.zoom_in_app import StableDiffusionZoomIn
from video_diffusion.inpaint_zoom.zoom_out_app import StableDiffusionZoomOut
from video_diffusion.stable_diffusion_video.stable_video_text2video import StableDiffusionText2VideoGenerator
from video_diffusion.tuneavideo.tuneavideo_text2video import TunaVideoText2VideoGenerator
from video_diffusion.zero_shot.zero_shot_text2video import ZeroShotText2VideoGenerator


def diffusion_app():
    app = gr.Blocks()
    with app:
        gr.HTML(
        """
        <h1 style='text-align: center'>
       Video Diffusion WebUI
        </h1>
        """
    )
        gr.HTML(
            """
            <h3 style='text-align: center'>
            Follow me for more! 
            <a href='https://twitter.com/kadirnar_ai' target='_blank'>Twitter</a> | <a href='https://github.com/kadirnar' target='_blank'>Github</a> | <a href='https://www.linkedin.com/in/kadir-nar/' target='_blank'>Linkedin</a>
            </h3>
            """
    )
        with gr.Row():
            with gr.Column():
                with gr.Tab("Stable Diffusion Video"):
                    StableDiffusionText2VideoGenerator.app()
                with gr.Tab("Tune-a-Video"):
                    TunaVideoText2VideoGenerator.app()
                with gr.Tab("Stable Infinite Zoom"):
                    with gr.Tab("Zoom In"):
                        StableDiffusionZoomIn.app()
                    with gr.Tab("Zoom Out"):
                        StableDiffusionZoomOut.app()
                with gr.Tab("Damo Text2Video"):
                    DamoText2VideoGenerator.app()
                with gr.Tab("Zero Shot Text2Video"):
                    ZeroShotText2VideoGenerator.app()

    app.queue(concurrency_count=1)
    app.launch(debug=True, enable_queue=True)


if __name__ == "__main__":
    diffusion_app()