Errors sending to UI

notifying client of AI model conversion

Azaion.Inference/api_client.pyx

@@ -1,4 +1,5 @@
 import json
+import os
 from http import HTTPStatus
 from os import path
 from uuid import UUID
@@ -6,6 +7,7 @@ import jwt
 import requests
 cimport constants
 import yaml
+from requests import HTTPError
 
 from cdn_manager cimport CDNManager, CDNCredentials
 from hardware_service cimport HardwareService
@@ -23,6 +25,9 @@ cdef class ApiClient:
 
     cdef set_credentials(self, Credentials credentials):
         self.credentials = credentials
+        if self.cdn_manager is not None:
+            return
+
         yaml_bytes = self.load_bytes(constants.CDN_CONFIG, <str>'')
         yaml_config = yaml.safe_load(yaml_bytes)
         creds = CDNCredentials(yaml_config["host"],
@@ -34,11 +39,18 @@ cdef class ApiClient:
         self.cdn_manager = CDNManager(creds)
 
     cdef login(self):
-        response = requests.post(f"{self.api_url}/login",
-            json={"email": self.credentials.email, "password": self.credentials.password})
-        response.raise_for_status()
-        token = response.json()["token"]
-        self.set_token(token)
+        response = None
+        try:
+            response = requests.post(f"{self.api_url}/login",
+                json={"email": self.credentials.email, "password": self.credentials.password})
+            response.raise_for_status()
+            token = response.json()["token"]
+            self.set_token(token)
+        except HTTPError as e:
+            print(response.json())
+            if response.status_code == HTTPStatus.CONFLICT:
+                res = response.json()
+                raise Exception(res['Message'])
 
 
     cdef set_token(self, str token):
@@ -123,11 +135,21 @@ cdef class ApiClient:
         return data
 
     cdef load_big_small_resource(self, str resource_name, str folder, str key):
-        cdef str big_part = path.join(<str>folder, f'{resource_name}.big')
+        cdef str big_part = f'{resource_name}.big'
         cdef str small_part = f'{resource_name}.small'
 
-        with open(<str>big_part, 'rb') as binary_file:
-            encrypted_bytes_big = binary_file.read()
+        print(f'checking on existence for {folder}\\{big_part}')
+        if os.path.exists(os.path.join(<str> folder, big_part)):
+            with open(path.join(<str> folder, big_part), 'rb') as binary_file:
+                encrypted_bytes_big = binary_file.read()
+            print(f'local file {folder}\\{big_part} is found!')
+        else:
+            print(f'downloading file {folder}\\{big_part} from cdn...')
+            if self.cdn_manager.download(folder, big_part):
+                with open(path.join(<str> folder, big_part), 'rb') as binary_file:
+                    encrypted_bytes_big = binary_file.read()
+            else:
+                return None
 
         encrypted_bytes_small = self.load_bytes(small_part, folder)
 
@@ -145,7 +167,7 @@ cdef class ApiClient:
 
         part_big = resource_encrypted[part_small_size:]
 
-        self.cdn_manager.upload(<str>constants.MODELS_FOLDER, <str>big_part_name, part_big)
+        self.cdn_manager.upload(folder, <str>big_part_name, part_big)
         with open(path.join(<str>folder, <str>big_part_name), 'wb') as f:
             f.write(part_big)
-        self.upload_file(small_part_name, part_small, constants.MODELS_FOLDER)
+        self.upload_file(small_part_name, part_small, folder)