fix: Fixed a bug on fake positives

backend/src/crispy/main.py

@@ -40,6 +40,9 @@ def main(videos: List[str]) -> None:
             l.debug(kill_array)
             vid.segment_video_with_kill_array(video, kill_array)
 
+    if not args.no_merge:
+        vid.merge_cuts()
+
 
 if __name__ == "__main__":
     l = logging.getLogger()
@@ -50,7 +53,7 @@ def main(videos: List[str]) -> None:
 
     l.debug(f"Arguments: {args}")
 
-    videos_path = ["4.mp4"]
+    videos_path = ["4.mp4", "quadra chrlie mice.mp4"]
 
     # FIXME: should be sort with the frontend ?
     videos_path.sort()

backend/src/crispy/utils/arguments.py

@@ -21,6 +21,11 @@
                      help="Do not extract frames",
                      action="store_true")
 
+_parser.add_argument("--no-merge",
+                     default=False,
+                     help="Do not merge final videos",
+                     action="store_true")
+
 args = _parser.parse_args()
 
 if args.debug:

backend/src/crispy/utils/ffmpeg_utils.py

@@ -1,6 +1,7 @@
 import os
 import random
 import string
+import shutil
 from typing import Optional, Any, List, Tuple
 
 import ffmpeg
@@ -148,3 +149,23 @@ def create_new_path(video_path: str) -> str:
     res = os.path.join(drive, cur_name + ext)
 
     return res
+
+
+# FIXME: audio
+def merge_videos(videos_path: List[str], save_path: str) -> None:
+    """
+    Merge videos together.
+    """
+    if len(videos_path) > 1:
+        videos: List[Any] = []
+        for video_path in videos_path:
+            videos.append(ffmpeg.input(video_path))
+        (
+            ffmpeg
+            .concat(*videos)
+            .output(save_path)
+            .overwrite_output()
+            .run(quiet=True)
+        ) # yapf: disable
+    else:
+        shutil.copyfile(videos_path[0], save_path)

backend/src/crispy/video/video.py

@@ -128,10 +128,29 @@ def get_kill_array_from_query_array(
 
     result = []
     for kill in kill_array:
-        if len(kill) < framerate // 2:
+        # Remove fake-positives
+        if len(kill) <= 2:
             continue
 
         start = kill[0] - frames_before
         end = kill[-1] + frames_after
         result.append((start, end))
     return result
+
+
+def merge_cuts() -> None:
+    """
+    Merge the cuts
+    """
+    folders = os.listdir(TMP_PATH)
+    folders = [f for f in folders if os.path.isdir(os.path.join(TMP_PATH, f))]
+    folders.sort()
+    cuts: List[str] = []
+    for folder in folders:
+        cut = os.listdir(os.path.join(TMP_PATH, folder, CUT))
+        cut.sort()
+        for i in range(len(cut)):
+            cut[i] = os.path.join(TMP_PATH, folder, CUT, cut[i])
+        cuts.extend(cut)
+
+    ff.merge_videos(cuts, os.path.join(TMP_PATH, "merged.mp4"))