Update gradio_app.py

gradio_app.py

@@ -2,6 +2,7 @@ import yaml
 import tempfile
 import gradio as gr
 import os
+import shutil
 import torch
 import imageio
 import argparse
@@ -249,8 +250,14 @@ def main(args):
         print(f"relight with bg generation! prompt:{relight_prompt}, light:{bg_source.value}, save in {results_path}.")
         return results_path
     
-def infer(n_prompt, relight_prompt, video_path, bg_source, save_path,
-          width, height, strength, gamma, num_step, text_guide_scale, seed):
+def infer(n_prompt, relight_prompt, video_path, bg_source,
+          width, height, strength, gamma, num_step, text_guide_scale, seed, progress=gr.Progress(track_tqdm=True)):
+
+    save_path = "./output"
+    # Ensure output folder is empty
+    if os.path.exists(save_path):
+        shutil.rmtree(save_path)
+    os.makedirs(save_path, exist_ok=True)
 
     config_data = {
         "n_prompt": n_prompt,
@@ -287,28 +294,60 @@ def infer(n_prompt, relight_prompt, video_path, bg_source, save_path,
     return results_path
 
 with gr.Blocks() as demo:
-    with gr.Row():
-        n_prompt = gr.Textbox(label="Negative Prompt")
-        relight_prompt = gr.Textbox(label="Relight Prompt")
-    with gr.Row():
-        video_path = gr.Textbox(label="Video Path")
-        bg_source = gr.Dropdown(["NONE", "LEFT", "RIGHT", "BOTTOM", "TOP"], label="Background Source")
-    with gr.Row():
-        save_path = gr.Textbox(label="Save Path")
-        width = gr.Number(label="Width", value=512)
-        height = gr.Number(label="Height", value=512)
-    with gr.Row():
-        strength = gr.Slider(minimum=0.0, maximum=1.0, label="Strength", value=0.5)
-        gamma = gr.Slider(minimum=0.0, maximum=1.0, label="Gamma", value=0.5)
-    with gr.Row():
-        num_step = gr.Number(label="Number of Steps", value=25)
-        text_guide_scale = gr.Number(label="Text Guide Scale", value=2)
-        seed = gr.Number(label="Seed", value=2060)
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown("# Light-A-Video")
+        with gr.Row():
+            with gr.Column():
+                video_path = gr.Textbox(label="Video Path")
+                with gr.Row():
+                    relight_prompt = gr.Textbox(label="Relight Prompt")
+                    bg_source = gr.Dropdown(["NONE", "LEFT", "RIGHT", "BOTTOM", "TOP"], label="Background Source")
+                
+                with gr.Accordion(label="Advanced Settings", open=False):
+                    n_prompt = gr.Textbox(label="Negative Prompt", value="bad quality, worse quality")
+                    with gr.Row():
+                        width = gr.Number(label="Width", value=512)
+                        height = gr.Number(label="Height", value=512)
+        
+                    with gr.Row():
+                        strength = gr.Slider(minimum=0.0, maximum=1.0, label="Strength", value=0.5)
+                        gamma = gr.Slider(minimum=0.0, maximum=1.0, label="Gamma", value=0.5)
+                    with gr.Row():
+                        num_step = gr.Number(label="Number of Steps", value=25)
+                        text_guide_scale = gr.Number(label="Text Guide Scale", value=2)
+                        seed = gr.Number(label="Seed", value=2060)
+
+                submit = gr.Button("Run")
+
+                gr.Examples(
+                    examples=[
+                        ["./input_animatediff/bear.mp4", "a bear walking on the rock, nature lighting, key light", "TOP"],
+                        ["./input_animatediff/boat.mp4", "a boat floating on the sea, sunset", "TOP"],
+                        ["./input_animatediff/car.mp4", "a car driving on the street, neon light", "RIGHT"],
+                        ["./input_animatediff/cat.mp4", "a cat, red and blue neon light", "LEFT"],
+                        ["./input_animatediff/cow.mp4", "a cow drinking water in the river, sunset", "RIGHT"],
+                        ["./input_animatediff/flowers.mp4", "A basket of flowers, sunshine, hard light", "LEFT"],
+                        ["./input_animatediff/fox.mp4", "a fox, sunlight filtering through trees, dappled light", "LEFT"],
+                        ["./input_animatediff/girl.mp4", "a girl, magic lit, sci-fi RGB glowing, key lighting", "BOTTOM"],
+                        ["./input_animatediff/girl2.mp4", "an anime girl, neon light", "RIGHT"],
+                        ["./input_animatediff/juice.mp4", "Pour juice into a glass, magic golden lit", "RIGHT"],
+                        ["./input_animatediff/man2.mp4", "handsome man with glasses, shadow from window, sunshine", "RIGHT"],
+                        ["./input_animatediff/man4.mp4", "handsome man with glasses, sunlight through the blinds", "LEFT"],
+                        ["./input_animatediff/plane.mp4", "a plane on the runway, bottom neon light", "BOTTOM"],
+                        ["./input_animatediff/toy.mp4", "a maneki-neko toy, cozy bedroom illumination", "RIGHT"],
+                        ["./input_animatediff/woman.mp4", "a woman with curly hair, natural lighting, warm atmosphere", "LEFT"],
+                    ],
+                    inputs=[video_path, relight_prompt, bg_source]
+                )
+
+            with gr.Column():
+                output = gr.Video(label="Results Path")
     
-    output = gr.Video(label="Results Path")
-    submit = gr.Button("Run")
-    submit.click(infer, inputs=[n_prompt, relight_prompt, video_path, bg_source, save_path,
-                                width, height, strength, gamma, num_step, text_guide_scale, seed], 
-                 outputs=output)
+    submit.click(
+        fn=infer, 
+        inputs=[n_prompt, relight_prompt, video_path, bg_source,
+                width, height, strength, gamma, num_step, text_guide_scale, seed], 
+        outputs=[output]
+    )
 
-demo.launch()
+demo.queue().launch(show_api=False, show_error=True, ssr_mode=False)