Merge pull request #11 from supervisely-ecosystem/workflow

Add Workflow and upgrade SDK to v6.73.157

config.json

@@ -20,6 +20,6 @@
     "target": ["images_project", "images_dataset"],
     "context_root": "Download as"
   },
-  "docker_image": "supervisely/import-export:6.73.106",
-  "min_instance_version": "6.9.22"
+  "docker_image": "supervisely/import-export:6.73.157",
+  "min_instance_version": "6.10.0"
 }

dev_requirements.txt

@@ -1 +1 @@
-supervisely==6.73.156
+supervisely==6.73.157

src/main.py

@@ -2,12 +2,16 @@
 
 import supervisely as sly
 
-import src.globals as g
 import src.functions as f
+import src.globals as g
+import src.workflow as w
+
+api = None
 
 
 class MyExport(sly.app.Export):
     def process(self, context: sly.app.Export.Context):
+        global api
         api = sly.Api.from_env()
 
         project = api.project.get_info_by_id(id=context.project_id)
@@ -16,6 +20,8 @@ def process(self, context: sly.app.Export.Context):
         else:
             datasets = api.dataset.get_list(project.id, recursive=True)
 
+        w.workflow_input(api, project.id)
+
         images_count = 0
         for ds in datasets:
             images_count += ds.images_count
@@ -47,9 +53,13 @@ def process(self, context: sly.app.Export.Context):
         progress = sly.Progress("Transformation ...", images_count)
         for ds in datasets:
             train_info, val_info = f.process_images(
-                api, meta, ds,
-                class_names, progress,
-                dir_names, skipped,
+                api,
+                meta,
+                ds,
+                class_names,
+                progress,
+                dir_names,
+                skipped,
                 max_kpts_count,
             )
             train_ids, train_img_paths, train_anns = train_info
@@ -106,6 +116,7 @@ def main():
     try:
         app = MyExport()
         app.run()
+        w.workflow_output(api, app.output_file)
     finally:
         if not sly.is_development():
             sly.logger.info(f"Remove temp directory: {g.TEMP_DIR}")

src/workflow.py

@@ -0,0 +1,29 @@
+# This module contains the functions that are used to configure the input and output of the workflow for the current app.
+
+from typing import Union
+
+import supervisely as sly
+
+
+def workflow_input(api: sly.Api, project_id: int):
+    api.app.workflow.add_input_project(project_id)
+    sly.logger.debug(f"Workflow: Input project - {project_id}")
+
+
+def workflow_output(api: sly.Api, file: Union[int, sly.api.file_api.FileInfo]):
+    try:
+        if isinstance(file, int):
+            file = api.file.get_info_by_id(file)
+        relation_settings = sly.WorkflowSettings(
+            title=file.name,
+            icon="archive",
+            icon_color="#33c94c",
+            icon_bg_color="#d9f7e4",
+            url=f"/files/{file.id}/true/?teamId={file.team_id}",
+            url_title="Download",
+        )
+        meta = sly.WorkflowMeta(relation_settings=relation_settings)
+        api.app.workflow.add_output_file(file, meta=meta)
+        sly.logger.debug(f"Workflow: Output file - {file}")
+    except Exception as e:
+        sly.logger.debug(f"Failed to add output to the workflow: {repr(e)}")