add 'folder' to responses

Dockerfile

@@ -36,4 +36,4 @@ COPY --chown=user . $HOME/app
 
 RUN bash $HOME/app/download_models.sh
 
-CMD python ./src/main.py
+CMD ["python", "./src/main.py"]

src/itr/dtb_cursor.py

@@ -34,11 +34,13 @@ class DatabaseCursor:
     ):
         with open(keyframes_groups_json_path) as file:
             keyframes_group_info = json.loads(file.read())
+            self.no_keyframes = len(keyframes_group_info)
         with open(subframes_groups_json_path) as file:
             subframes_groups_info = json.loads(file.read())
+            self.no_subframes = len(subframes_groups_info)
+
         self.frames_groups_info = keyframes_group_info
         self.frames_groups_info.extend(subframes_groups_info)
-
         assert self.index.ntotal == len(
             self.frames_groups_info
         ), "dtb_cursos::Index length and map lenght mismatch"
@@ -49,5 +51,8 @@ class DatabaseCursor:
         for i in range(len(ids[0])):
             frame_detail = self.frames_groups_info[ids[0][i]]
             frame_detail["distance"] = str(distances[0][i])
+            frame_detail["folder"] = (
+                "Keyframes" if ids[0][i] < self.no_keyframes else "Subframes"
+            )
             results.append(frame_detail)
         return results