Update app.py

app.py

@@ -28,8 +28,6 @@ from PIL import Image, ImageFilter
 from sam2.build_sam import build_sam2_video_predictor
 
 from moviepy.editor import ImageSequenceClip
-from tqdm import tqdm
-import re
 
 def get_video_fps(video_path):
     # Open the video file
@@ -317,13 +315,13 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
         # Define the output filename and save the figure as a JPEG file
         output_filename = os.path.join(frames_output_dir, f"frame_{out_frame_idx}.jpg")
         plt.savefig(output_filename, format='jpg')
-        
-        # Append the file path to the list
-        jpeg_images.append(output_filename)
     
         # Close the plot
         plt.close()
 
+        # Append the file path to the list
+        jpeg_images.append(output_filename)
+
     
 
     if vis_frame_type == "check":
@@ -337,30 +335,12 @@ def propagate_to_all(video_in, checkpoint, stored_inference_state, stored_frame_
         # Write the result to a file
         final_vid_output_path = "output_video.mp4"
         
-        # Initialize the tqdm progress bar with the total number of frames
-        pbar = tqdm(total=total_frames, desc="Rendering video")
-        
-        # Define a function to update the progress bar
-        def update_progress(txt):
-            # Extract progress from the log message
-            progress_match = re.search(r"frame\s+(\d+)", txt)
-            if progress_match:
-                current_frame = int(progress_match.group(1))
-                pbar.n = current_frame
-                pbar.refresh()
-        
         # Write the result to a file
         clip.write_videofile(
             final_vid_output_path,
-            codec='libx264',
-            logger=lambda txt: update_progress(txt)
+            codec='libx264'
         )
         
-        # Close the progress bar after the process is finished
-        pbar.close()
-        
-        
-       
         return gr.update(value=None), gr.update(value=final_vid_output_path)
 
 def update_ui(vis_frame_type):