Update app.py

app.py

@@ -68,9 +68,33 @@ def generate(input_video):
             check=True
         )
 
-        output_videos = glob(os.path.join(f"{workdir}", "*.mp4"))
-        print(output_videos)
-        return output_videos[4]
+        # Get all .mp4 files in the workdir
+        output_videos = glob(os.path.join(workdir, "*.mp4"))
+        print("Found videos:", output_videos)
+    
+        # Check if the 5th video exists
+        if len(output_videos) < 5:
+            raise IndexError("Less than 5 .mp4 files found in the workdir.")
+    
+        # Get the 5th video
+        selected_video = output_videos[4]
+        print("Selected video:", selected_video)
+    
+        # Create a new temporary directory
+        temp_dir = tempfile.mkdtemp()
+        print("Temporary directory created:", temp_dir)
+    
+        # Copy the selected video to the temporary directory
+        new_video_path = os.path.join(temp_dir, os.path.basename(selected_video))
+        shutil.copy(selected_video, new_video_path)
+        print(f"Copied {selected_video} to {new_video_path}")
+    
+        # Delete the workdir folder
+        shutil.rmtree(workdir)
+        print(f"Deleted workdir: {workdir}")
+    
+        # Return the new path of the copied video
+        return new_video_path
    
     except subprocess.CalledProcessError as e:
         raise gr.Error(f"Error during inference: {str(e)}")
@@ -78,7 +102,7 @@ def generate(input_video):
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# L4GM: Large 4D Gaussian Reconstruction Model")
-            gr.HTML("""
+        gr.HTML("""
             <div style="display:flex;column-gap:4px;">
                 <a href="https://github.com/nv-tlabs/L4GM-official/tree/main">
                     <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
@@ -96,7 +120,7 @@ with gr.Blocks() as demo:
     				<img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/follow-me-on-HF-sm-dark.svg" alt="Follow me on HF">
     			</a>
             </div>
-            """)
+         """)
         with gr.Row():
             with gr.Column():
                 input_video = gr.Video(label="Input Video", interactive=False)