Update gradio_app.py

gradio_app.py

@@ -18,7 +18,7 @@ from torch.hub import download_url_to_file
 from src.ic_light import BGSource
 from src.animatediff_pipe import AnimateDiffVideoToVideoPipeline
 from src.ic_light_pipe import StableDiffusionImg2ImgPipeline
-from utils.tools import read_video
+from utils.tools import read_video, set_all_seed
 
 from huggingface_hub import snapshot_download, hf_hub_download
 
@@ -247,6 +247,7 @@ def main(args):
         results_path = f"{save_path}/relight_{video_name}"
         imageio.mimwrite(results_path, frames, fps=8)
         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):
@@ -280,10 +281,7 @@ def infer(n_prompt, relight_prompt, video_path, bg_source, save_path,
             self.config = config_path
     
     args = Args()
-    main(args)
-
-    video_name = os.path.basename(video_path)
-    results_path = f"{save_path}/relight_{video_name}"
+    results_path= main(args)
     os.remove(config_path)
     
     return results_path
@@ -307,7 +305,7 @@ with gr.Blocks() as demo:
         text_guide_scale = gr.Number(label="Text Guide Scale", value=2)
         seed = gr.Number(label="Seed", value=2060)
     
-    output = gr.Textbox(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],