Update app.py

app.py

@@ -64,6 +64,8 @@ cleanup_thread.start()
 @spaces.GPU
 def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", fps=0, video_handling="slow_down", fast_mode=False):
  try:
+ start_time = time.time() # Start the timer
+
  # Load the video using moviepy
  video = mp.VideoFileClip(vid)
 
@@ -79,7 +81,7 @@ def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", fps=
 
  # Process each frame for background removal
  processed_frames = []
- yield gr.update(visible=True), gr.update(visible=False)
+ yield gr.update(visible=True), gr.update(visible=False), f"Processing started... Elapsed time: 0 seconds"
 
  if bg_type == "Video":
  background_video = mp.VideoFileClip(bg_video)
@@ -115,7 +117,8 @@ def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", fps=
  processed_image = pil_image # Default to original image if no background is selected
 
  processed_frames.append(np.array(processed_image))
- yield processed_image, None
+ elapsed_time = time.time() - start_time
+ yield processed_image, None, f"Processing frame {i+1}... Elapsed time: {elapsed_time:.2f} seconds"
 
  # Create a new video from the processed frames
  processed_video = mp.ImageSequenceClip(processed_frames, fps=fps)
@@ -130,14 +133,16 @@ def fn(vid, bg_type="Color", bg_image=None, bg_video=None, color="#00FF00", fps=
  temp_filepath = os.path.join(temp_dir, unique_filename)
  processed_video.write_videofile(temp_filepath, codec="libx264")
 
- yield gr.update(visible=False), gr.update(visible=True)
+ elapsed_time = time.time() - start_time
+ yield gr.update(visible=False), gr.update(visible=True), f"Processing complete! Elapsed time: {elapsed_time:.2f} seconds"
  # Return the path to the temporary file
- yield processed_image, temp_filepath
+ yield processed_image, temp_filepath, f"Processing complete! Elapsed time: {elapsed_time:.2f} seconds"
 
  except Exception as e:
  print(f"Error: {e}")
- yield gr.update(visible=False), gr.update(visible=True)
- yield None, f"Error processing video: {e}"
+ elapsed_time = time.time() - start_time
+ yield gr.update(visible=False), gr.update(visible=True), f"Error processing video: {e}. Elapsed time: {elapsed_time:.2f} seconds"
+ yield None, f"Error processing video: {e}", f"Error processing video: {e}. Elapsed time: {elapsed_time:.2f} seconds"
 
 
 def process(image, bg, fast_mode=False):
@@ -192,6 +197,7 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
  video_handling_radio = gr.Radio(["slow_down", "loop"], label="Video Handling", value="slow_down", interactive=True)
  fast_mode_checkbox = gr.Checkbox(label="Fast Mode (Use BiRefNet_lite)", value=False, interactive=True)
 
+ time_textbox = gr.Textbox(label="Time Elapsed", interactive=False) # Add time textbox
 
  def update_visibility(bg_type):
  if bg_type == "Color":
@@ -214,7 +220,7 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
  ["rickroll-2sec.mp4", "Color", None, None],
  ],
  inputs=[in_video, bg_type, bg_image, bg_video],
- outputs=[stream_image, out_video],
+ outputs=[stream_image, out_video, time_textbox],
  fn=fn,
  cache_examples=True,
  cache_mode="eager",
@@ -224,7 +230,7 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
  submit_button.click(
  fn,
  inputs=[in_video, bg_type, bg_image, bg_video, color_picker, fps_slider, video_handling_radio, fast_mode_checkbox],
- outputs=[stream_image, out_video],
+ outputs=[stream_image, out_video, time_textbox],
  )
 
 if __name__ == "__main__":