synapse-sdk 1.0.0a82__tar.gz → 1.0.0a83__tar.gz

{synapse_sdk-1.0.0a82 → synapse_sdk-1.0.0a83}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0a82
+Version: 1.0.0a83
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0a82 → synapse_sdk-1.0.0a83}/synapse_sdk/plugins/categories/pre_annotation/actions/to_task.py

@@ -1,7 +1,7 @@
 import json
 from datetime import datetime
 from enum import Enum
-from typing import Annotated, Any, Dict, List, Optional
+from typing import Annotated, Any, Dict, List, Optional, Tuple
 
 import requests
 from pydantic import AfterValidator, BaseModel, field_validator
@@ -382,7 +382,7 @@ class ToTaskAction(Action):
         data_file = files.get(target_specification_name)
 
         # Extract primary file URL from task data
-        primary_file_url = self._extract_primary_file_url(task)
+        primary_file_url, primary_file_original_name = self._extract_primary_file_url(task)
         if not primary_file_url:
             error_msg = 'Primary image URL not found in task data'
             self.run.log_message(f'{error_msg} for task {task_id}')
@@ -396,6 +396,13 @@ class ToTaskAction(Action):
             return {'success': False, 'error': error_msg}
 
         data_file_url = data_file.get('url')
+        data_file_original_name = data_file.get('file_name_original')
+        if not data_file_original_name:
+            error_msg = 'File original name not found'
+            self.run.log_message(f'{error_msg} for task {task_id}')
+            self.run.log_annotate_task_data({'task_id': task_id, 'error': error_msg}, AnnotateTaskDataStatus.FAILED)
+            return {'success': False, 'error': error_msg}
+
         if not data_file_url:
             error_msg = 'URL not found'
             self.run.log_message(f'{error_msg} for task {task_id}')
@@ -406,7 +413,9 @@ class ToTaskAction(Action):
         try:
             # Convert data to task object
             annotation_to_task = self.entrypoint(self.run)
-            converted_data = annotation_to_task.convert_data_from_file(primary_file_url, data_file_url)
+            converted_data = annotation_to_task.convert_data_from_file(
+                primary_file_url, primary_file_original_name, data_file_url, data_file_original_name
+            )
         except requests.RequestException as e:
             error_msg = f'Failed to fetch data from URL: {str(e)}'
             self.run.log_message(f'{error_msg} for task {task_id}')
@@ -467,7 +476,7 @@ class ToTaskAction(Action):
                 return pre_processor_status
 
             # Extract primary file URL from task data
-            primary_file_url = self._extract_primary_file_url(task)
+            primary_file_url, _ = self._extract_primary_file_url(task)
             if not primary_file_url:
                 error_msg = 'Primary image URL not found in task data'
                 self.run.log_message(f'{error_msg} for task {task_id}')
@@ -609,23 +618,23 @@ class ToTaskAction(Action):
         except Exception as e:
             return {'success': False, 'error': f'Failed to restart pre-processor: {str(e)}'}
 
-    def _extract_primary_file_url(self, task: Dict[str, Any]) -> Optional[str]:
+    def _extract_primary_file_url(self, task: Dict[str, Any]) -> Tuple[str, str]:
         """Extract the primary file URL from task data.
 
         Args:
             task (Dict[str, Any]): The task data.
 
         Returns:
-            Optional[str]: The primary file URL if found, None otherwise.
+            Tuple[str, str]: The primary file URL and original name.
         """
         data_unit = task.get('data_unit', {})
         files = data_unit.get('files', {})
 
         for file_info in files.values():
             if isinstance(file_info, dict) and file_info.get('is_primary') and file_info.get('url'):
-                return file_info['url']
+                return file_info['url'], file_info['file_name_original']
 
-        return None
+        return None, None
 
     def _run_inference(
         self, client: BackendClient, pre_processor_code: str, pre_processor_version: str, primary_file_url: str

{synapse_sdk-1.0.0a82 → synapse_sdk-1.0.0a83}/synapse_sdk/plugins/categories/pre_annotation/templates/plugin/to_task.py

@@ -7,12 +7,20 @@ class AnnotationToTask:
         """
         self.run = run
 
-    def convert_data_from_file(self, primary_file_url: str, data_file_url: str) -> dict:
+    def convert_data_from_file(
+        self,
+        primary_file_url: str,
+        primary_file_original_name: str,
+        data_file_url: str,
+        data_file_original_name: str,
+    ) -> dict:
         """Convert the data from a file to a task object.
 
         Args:
             primary_file_url (str): primary file url.
+            primary_file_original_name (str): primary file original name.
             data_file_url (str): data file url.
+            data_file_original_name (str): data file original name.
 
         Returns:
             dict: The converted data.

{synapse_sdk-1.0.0a82 → synapse_sdk-1.0.0a83}/synapse_sdk/utils/converters/coco/to_dm.py

@@ -116,12 +116,13 @@ class COCOToDMConverter(ToDMConverter):
             result[img_filename] = (dm_json, img_path)
         return result
 
-    def convert_single_file(self, data: Dict[str, Any], original_file: IO) -> Dict[str, Any]:
+    def convert_single_file(self, data: Dict[str, Any], original_file: IO, original_image_name: str) -> Dict[str, Any]:
         """Convert a single COCO annotation data and corresponding image to DM format.
 
         Args:
             data: COCO format data dictionary (JSON content)
             original_file: File object for the corresponding original image
+            original_image_name: Original image name
 
         Returns:
             Dictionary containing DM format data for the single file
@@ -147,7 +148,7 @@ class COCOToDMConverter(ToDMConverter):
         matched_img = None
         for img in images:
             for key in ['file_name', 'filename', 'name']:
-                if key in img and os.path.basename(img[key]) == img_basename:
+                if key in img and os.path.basename(img[key]) == original_image_name:
                     matched_img = img
                     break
             if matched_img:
@@ -157,7 +158,6 @@ class COCOToDMConverter(ToDMConverter):
             raise ValueError(f'No matching image found in COCO data for file: {img_basename}')
 
         img_id = matched_img['id']
-        print('img_id : ', img_id)
         cat_map = {cat['id']: cat for cat in categories}
         anns = [ann for ann in annotations if ann['image_id'] == img_id]
 

{synapse_sdk-1.0.0a82 → synapse_sdk-1.0.0a83}/synapse_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-sdk
-Version: 1.0.0a82
+Version: 1.0.0a83
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT