chore:updating autoplay

app.py

@@ -36,8 +36,8 @@ def run_render_final(npy_file_path):
     command = ["python", "render_final.py", npy_file_path]
     try:
         gif_res = subprocess.run(command, check=True, text=True, capture_output=True)
-        gif_file_path = find_latest_file('output', 'gif')
-        return gif_file_path
+        vid_file_path = find_latest_file('output', 'mp4')
+        return vid_file_path
     except subprocess.CalledProcessError as e:
         st.error(f"Error: {e.stderr}")
         return None
@@ -113,10 +113,10 @@ with button_col1:
 with button_col2:
     if st.button("Render Skeleton"):
         if 'npy_file_path' in st.session_state and st.session_state.npy_file_path:
-            gif_file_path = run_render_final(st.session_state.npy_file_path)
-            if gif_file_path:
-                st.session_state.gif_file_path = gif_file_path
-                st.session_state.gif_base64 = gif_to_base64(gif_file_path)
+            vid_file_path = run_render_final(st.session_state.npy_file_path)
+            if vid_file_path:
+                st.session_state.vid_file_path = vid_file_path
+                # st.session_state.gif_base64 = gif_to_base64(gif_file_path)
         else:
             st.error("No npy file found. Please generate HTML first.")
 
@@ -124,7 +124,7 @@ with button_col2:
 # Display the results side by side using HTML components
 if 'html_content' in st.session_state or 'gif_base64' in st.session_state:
     html_content = st.session_state.html_content if 'html_content' in st.session_state else ""
-    gif_base64 = st.session_state.gif_base64 if 'gif_base64' in st.session_state else ""
+    video_path = st.session_state.vid_file_path if 'vid_file_path' in st.session_state else ""
     
     disp_col1, disp_col2 = st.columns([1, 1])
     
@@ -132,6 +132,6 @@ if 'html_content' in st.session_state or 'gif_base64' in st.session_state:
         components.html(html_content, height=800, scrolling=True)
     
     with disp_col2:
-        if gif_base64:
-            gif_html = f'<img src="data:image/gif;base64,{gif_base64}" style="width:100%;" loop="infinite">'
+        if video_path:
+            gif_html = f'<video controls width="800" height="800"><source src="file:///{video_path}" type="video/mp4"></video autoplay>'
             components.html(gif_html, height=800, scrolling=True)

render_final.py

@@ -18,7 +18,7 @@ import math
 # import ffmpeg
 from PIL import Image
 import argparse
-
+import moviepy.editor as mp
 class WeakPerspectiveCamera(pyrender.Camera):
     def __init__(self,
                  scale,
@@ -162,9 +162,10 @@ def render(motions, outdir='test_vis', device_id=0, name=None, pred=True):
 
     out = np.stack(vid, axis=0)
     gif_path = os.path.join(outdir, f'{name}.gif')
-    imageio.mimsave(gif_path, out, fps=20)
-    
+    imageio.mimwrite(gif_path, out, fps=20)
     
+    out_video = mp.VideoFileClip(gif_path)
+    out_video.write_videofile(gif_path.replace('.gif', '.mp4'), codec='libx264', fps=20)
 
 
 if __name__ == "__main__":