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(
"""
Video Diffusion WebUI
"""
)
gr.HTML(
"""
"""
)
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()