Update app.py

app.py

@@ -213,22 +213,36 @@ class AnimateController:
             self.savedir_sample, f"{sample_idx}.mp4")
         save_videos_grid(sample, save_sample_path)
 
-        sample_config = {
-            "prompt": prompt_textbox,
-            "n_prompt": negative_prompt_textbox,
-            "sampler": sampler_dropdown,
-            "num_inference_steps": sample_step_slider,
-            "guidance_scale": cfg_scale_slider,
-            "width": width_slider,
-            "height": height_slider,
-            "video_length": length_slider,
-            "seed": seed
-        }
-        json_str = json.dumps(sample_config, indent=4)
-        with open(os.path.join(self.savedir, "logs.json"), "a") as f:
-            f.write(json_str)
-            f.write("\n\n")
-        return gr.Video.update(value=save_sample_path)
+        # sample_config = {
+        #     "prompt": prompt_textbox,
+        #     "n_prompt": negative_prompt_textbox,
+        #     "sampler": sampler_dropdown,
+        #     "num_inference_steps": sample_step_slider,
+        #     "guidance_scale": cfg_scale_slider,
+        #     "width": width_slider,
+        #     "height": height_slider,
+        #     "video_length": length_slider,
+        #     "seed": seed
+        # }
+        # json_str = json.dumps(sample_config, indent=4)
+        # with open(os.path.join(self.savedir, "logs.json"), "a") as f:
+        #     f.write(json_str)
+        #     f.write("\n\n")
+        # return gr.Video.update(value=save_sample_path)
+
+
+        # Read the content of the video file and encode it to base64
+        with open(save_sample_path, "rb") as video_file:
+            video_base64 = base64.b64encode(video_file.read()).decode('utf-8')
+    
+        # Prepend the appropriate data URI header with MIME type
+        video_data_uri = 'data:video/mp4;base64,' + video_base64
+        
+        # clean-up (otherwise there is a risk of "ghosting", eg. someone seeing the previous generated video",
+        # of one of the steps go wrong)
+        os.remove(video_path)
+    
+        return video_data_uri
 
 
 controller = AnimateController()
@@ -332,8 +346,7 @@ def ui():
                     generate_button = gr.Button(
                         value="Generate", variant='primary')
 
-                result_video = gr.Video(
-                    label="Generated Animation", interactive=False)
+                result_video_base_64 = gr.Text()
 
             generate_button.click(
                 fn=controller.animate,
@@ -351,7 +364,7 @@ def ui():
                     cfg_scale_slider,
                     seed_textbox,
                 ],
-                outputs=[result_video]
+                outputs=[result_video_base64]
             )
             
     return demo