Connect to ML worker for classification.

arkindex/documents/models.py

@@ -3,6 +3,7 @@ from django.contrib.postgres.indexes import GinIndex
 from django.urls import reverse
 from enumfields import EnumField, Enum
 from arkindex.project.models import IndexableModel
+from arkindex.project.celery import app as celery_app
 from arkindex.documents.managers import ElementManager
 from arkindex.project.fields import ArrayField
 import uuid
@@ -329,6 +330,15 @@ class Page(Element):
             out.append(self.direction.value)
         return ' '.join(out)
 
+    def ml_classify(self):
+        '''
+        Use a machine learning worker to classify the page
+        using its image
+        Celery is triggered through an external signature
+        '''
+        signature = celery_app.signature('arkindex_ml.tasks.classify')
+        return signature.delay(self.zone.image.get_thumbnail_url(max_width=None))
+
 
 class Act(Element):
     """

arkindex/project/settings.py

@@ -312,6 +312,12 @@ CELERY_ONCE = {
         'default_timeout': 3600,
     }
 }
+CELERY_TASK_ROUTES = {
+    # Route all machine learning work to specific workers
+    'arkindex_ml.*': {
+        'queue': 'ml',
+    }
+}
 CELERY_WORKING_DIR = os.environ.get('CELERY_WORKING_DIR', os.path.join(BASE_DIR, 'workers'))
 
 # Email