Update app.py

app.py

@@ -16,9 +16,11 @@ from depth_anything.dpt import DepthAnything
 from depth_anything.util.transform import Resize, NormalizeImage, PrepareForNet
 from moviepy.editor import *
 
-def zip_files(files):
+def zip_files(files_in, files_out):
     with ZipFile("depth_result.zip", "w") as zipObj:
-        for idx, file in enumerate(files):
+        for idx, file in enumerate(files_in):
+            zipObj.write(file, file.split("/")[-1])
+        for idx, file in enumerate(files_out):
             zipObj.write(file, file.split("/")[-1])
     return "depth_result.zip"
 
@@ -106,11 +108,13 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
         #out = cv2.VideoWriter(output_path, fourcc, frame_rate, (output_width, frame_height))
         count=0
         depth_frames = []
+        orig_frames = []
         while raw_video.isOpened():
             ret, raw_frame = raw_video.read()
             if not ret:
                 break
-            
+
+            orig_frames.append(raw_frame)
             frame = cv2.cvtColor(raw_frame, cv2.COLOR_BGR2RGB) / 255.0
             frame_pil =  Image.fromarray((frame * 255).astype(np.uint8))
             frame = transform({'image': frame})['image']
@@ -148,7 +152,7 @@ def make_video(video_path, outdir='./vis_video_depth', encoder='vits'):
             count += 1
 
         final_vid = create_video(depth_frames, frame_rate, "depth")
-        final_zip = zip_files(depth_frames)
+        final_zip = zip_files(orig_frames, depth_frames)
         raw_video.release()
         # out.release()
         cv2.destroyAllWindows()