Better video

app.py

@@ -7,6 +7,7 @@ import imageio
 import numpy as np
 import math
 import argparse
+import tempfile
 
 import torch
 import base64
@@ -61,6 +62,14 @@ def decode_latent_images_foo(
     )
     return decoded
 
+def to_video(frames: list[Image.Image], fps: int = 5) -> str:
+        out_file = tempfile.NamedTemporaryFile(suffix='.mp4', delete=False)
+        writer = imageio.get_writer(out_file.name, format='FFMPEG', fps=fps)
+        for frame in frames:
+            writer.append_data(np.asarray(frame))
+        writer.close()
+        return out_file.name
+
 def generate_3D(input, grid_size=64):
     set_state('Entered generate function...')
 
@@ -129,11 +138,8 @@ def generate_3D(input, grid_size=64):
 
     images = decode_latent_images(xm, latents[0], cameras, rendering_mode=render_mode)
 
-    # Convert images to gif
-    images[0].save(f'/tmp/mesh.gif', save_all=True, append_images=images[1:], duration=100, loop=0)
-
 
-    return ply_to_glb('/tmp/mesh.ply', '/tmp/mesh.glb'), '/tmp/mesh.gif', gr.update(value=['/tmp/mesh.glb', '/tmp/mesh.ply'], visible=True)
+    return ply_to_glb('/tmp/mesh.ply', '/tmp/mesh.glb'), to_video(images), gr.update(value=['/tmp/mesh.glb', '/tmp/mesh.ply'], visible=True)
 
 def prepare_img(img):
 
@@ -257,20 +263,20 @@ with block:
     if torch.cuda.is_available():
         gr.Examples(
             examples=[
-                ["images/pumpkin.png"],
-                ["images/fantasy_world.png"],
+                ["a shark"],
+                ["an avocado"],
             ],
-            inputs=[input_image],
+            inputs=[prompt],
             outputs=[model_3d, model_gif, file_out],
             fn=generate_3D,
             cache_examples=True
         )
         gr.Examples(
             examples=[
-                ["a shark"],
-                ["an avocado"],
+                ["images/pumpkin.png"],
+                ["images/fantasy_world.png"],
             ],
-            inputs=[prompt],
+            inputs=[input_image],
             outputs=[model_3d, model_gif, file_out],
             fn=generate_3D,
             cache_examples=True