Update app.py

app.py

@@ -78,7 +78,7 @@ def preprocess_video(video_path, target_fps=24, max_resolution=(640, 360)):
     
     return video
 
-def process_frame(frame, seed=0):
+def process_frame(frame, seed=0, start_time=None):
     """
     Process a single frame through the depth model.
     Returns the discriminative depth map.
@@ -113,6 +113,9 @@ def process_video(video_path, fps=0, seed=0, max_workers=2):
     """
     temp_dir = None
     try:
+        # Initialize start_time here for use in process_frame
+        start_time = time.time()
+        
         # Preprocess the video
         video = preprocess_video(video_path)
         
@@ -134,7 +137,7 @@ def process_video(video_path, fps=0, seed=0, max_workers=2):
         processed_frames = []
         with ThreadPoolExecutor(max_workers=max_workers) as executor:
             for i in range(0, total_frames, 10):  # Process 10 frames at a time
-                futures = [executor.submit(process_frame, frames[j], seed) for j in range(i, min(i + 10, total_frames))]
+                futures = [executor.submit(process_frame, frames[j], seed, start_time) for j in range(i, min(i + 10, total_frames))]
                 for j, future in enumerate(futures):
                     try:
                         result = future.result()