add grid size parameter

alpr_api.py

@@ -119,10 +119,28 @@ def create_rest_server_flask():
 
     @app.route('/v1/image/alpr', methods=['POST'])
     def alpr():
+        """
+        This function is called when a POST request is made to the /v1/image/alpr endpoint.
+        The function receives an image and processes it using the ultimateALPR SDK.
+
+        Parameters:
+            - upload: The image to be processed
+            - grid_size: The number of cells to split the image into (e.g. 4)
+            - wanted_cells: The cells to process in the grid separated by commas (e.g. 1,2,3,4) (max: grid_size²)
+        """
         interference = time.time()
 
         if 'upload' not in request.files:
             return jsonify({'error': 'No image found'})
+        if 'grid_size' in request.form and request.form['grid_size'].isdigit():
+            grid_size = int(request.form['grid_size'])
+        else:
+            grid_size = None
+        if 'wanted_cells' in request.form and request.form['wanted_cells']:
+            wanted_cells = request.form['wanted_cells'].split(',')
+            wanted_cells = [int(cell) for cell in wanted_cells]
+        else:
+            wanted_cells = None
 
         image = request.files['upload']
         if image.filename == '':
@@ -135,7 +153,7 @@ def create_rest_server_flask():
         if not result['predictions']:
             print("No plate found in the image, attempting to split the image")
 
-            predictions_found = find_best_plate_with_split(image)
+            predictions_found = find_best_plate_with_split(image, grid_size, wanted_cells)
 
             if predictions_found:
                 result['predictions'].append(max(predictions_found, key=lambda x: x['confidence']))
@@ -195,19 +213,21 @@ def convert_to_cpai_compatible(result):
     return response
 
 
-def find_best_plate_with_split(image, split_size=4, wanted_cells=None):
+def find_best_plate_with_split(image: Image, grid_size: int = None, wanted_cells: str = None):
+    if grid_size is None:
+        grid_size = 3
     if wanted_cells is None:
-        wanted_cells = [5, 6, 7, 9, 10, 11, 14, 15] # TODO: use params not specifc to my use case
+        wanted_cells = list(range(1, grid_size * grid_size + 1))
 
     predictions_found = []
 
     width, height = image.size
-    cell_width = width // split_size
-    cell_height = height // split_size
+    cell_width = width // grid_size
+    cell_height = height // grid_size
 
-    for cell_index in range(1, split_size * split_size + 1):
-        row = (cell_index - 1) // split_size
-        col = (cell_index - 1) % split_size
+    for cell_index in range(1, grid_size * grid_size + 1):
+        row = (cell_index - 1) // grid_size
+        col = (cell_index - 1) % grid_size
         left = col * cell_width
         upper = row * cell_height
         right = left + cell_width

build_alpr_api.sh

@@ -1 +1,2 @@
-pyinstaller --noconfirm --onefile --console --add-data libs:. --add-data assets:assets --add-data static:static --add-data templates:templates --name easy-local-alpr-1.1.0-openvinocpu_linux_x86_64 "alpr_api.py"
+pyinstaller --noconfirm --console --add-data libs:. --add-data assets:assets --add-data static:static --add-data templates:templates --name easy-local-alpr-1.1.0-openvinocpu_linux_x86_64 "alpr_api.py"
+# optional: --onefile

templates/index.html

@@ -14,6 +14,25 @@
             background-size: 20px 20px;
             font-family: 'Google Sans', sans-serif;
         }
+        .loading-circle {
+            border: 4px solid rgba(0, 0, 0, 0.1);
+            border-left-color: #000;
+            border-radius: 50%;
+            width: 24px;
+            height: 24px;
+            animation: spin 1s linear infinite;
+        }
+        @keyframes spin {
+            to { transform: rotate(360deg); }
+        }
+        input[type="text"], input[type="number"] {
+            background-color: white;
+            color: black;
+        }
+        input[type="text"].dark, input[type="number"].dark {
+            background-color: #3b3b3b;
+            color: white;
+        }
     </style>
 </head>
 <body class="bg-neutral-100 dark:bg-neutral-900 dark:text-white flex items-center justify-center min-h-screen p-4">
@@ -36,10 +55,21 @@
                     <span id="fileName" class="ml-2 text-sm text-gray-600 dark:text-gray-300"></span>
                 </div>
             </div>
+            <div>
+                <label for="grid_size" class="block text-sm font-medium text-gray-700 dark:text-gray-300">Grid Size:</label>
+                <input type="number" id="grid_size" name="grid_size" value="3" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm dark">
+            </div>
+            <div>
+                <label for="wanted_cells" class="block text-sm font-medium text-gray-700 dark:text-gray-300">Wanted Cells:</label>
+                <input type="text" id="wanted_cells" name="wanted_cells" class="mt-1 block w-full px-3 py-2 border border-gray-300 rounded-md shadow-sm focus:outline-none focus:ring-indigo-500 focus:border-indigo-500 sm:text-sm dark">
+            </div>
             <div id="imagePreview" class="mt-4 hidden">
                 <img id="previewImage" src="#" alt="Preview" class="max-w-full h-auto rounded-lg">
             </div>
-            <button type="submit" class="w-full py-2 px-4 bg-black text-white font-semibold rounded-md shadow-sm hover:bg-neutral-900 dark:bg-neutral-900 dark:hover:bg-neutral-950">Upload</button>
+            <button id="submitButton" type="submit" class="w-full py-2 px-4 bg-black text-white font-semibold rounded-md shadow-sm hover:bg-neutral-900 dark:bg-neutral-900 dark:hover:bg-neutral-950 relative flex justify-center items-center">
+                <span>Upload</span>
+                <div id="loadingCircle" class="loading-circle absolute hidden"></div>
+            </button>
         </form>
         <div class="mt-6">
             <h2 class="text-xl font-semibold mb-2 dark:text-gray-200">Response</h2>
@@ -73,12 +103,22 @@
         function toggleLogo() {
             const logo = document.getElementById('logo');
             const logoDark = document.getElementById('logoDark');
+            const upload = document.getElementById('upload');
+            const gridSize = document.getElementById('grid_size');
+            const wantedCells = document.getElementById('wanted_cells');
+
             if (prefersDarkScheme.matches) {
                 logo.style.display = 'none';
                 logoDark.style.display = 'block';
+                upload.classList.add('dark');
+                gridSize.classList.add('dark');
+                wantedCells.classList.add('dark');
             } else {
                 logo.style.display = 'block';
                 logoDark.style.display = 'none';
+                upload.classList.remove('dark');
+                gridSize.classList.remove('dark');
+                wantedCells.classList.remove('dark');
             }
         }
 
@@ -89,6 +129,9 @@
             $('#uploadForm').on('submit', function (e) {
                 e.preventDefault();
                 var formData = new FormData(this);
+                $('#submitButton').prop('disabled', true);
+                $('#loadingCircle').removeClass('hidden');
+
                 $.ajax({
                     url: '/v1/image/alpr',
                     type: 'POST',
@@ -101,6 +144,10 @@
                     error: function (xhr, status, error) {
                         var err = JSON.parse(xhr.responseText);
                         $('#responseBox').text(JSON.stringify(err, null, 2));
+                    },
+                    complete: function () {
+                        $('#submitButton').prop('disabled', false);
+                        $('#loadingCircle').addClass('hidden');
                     }
                 });
             });