fix: Error viewing images in Explore Data page (#936)

code/backend/batch/utilities/search/azure_search_handler.py

@@ -29,8 +29,9 @@ def process_results(self, results):
         if results is None:
             return []
         data = [
-            [json.loads(result["metadata"])["chunk"], result["content"]]
-            for result in results
+            # Note that images uploaded with advanced image processing do not have a chunk ID
+            [json.loads(result["metadata"]).get("chunk", i), result["content"]]
+            for i, result in enumerate(results)
         ]
         return data
 

code/tests/search_utilities/test_azure_search_handler.py

@@ -64,6 +64,20 @@ def test_process_results(handler):
     assert data[0] == [1, "Content 1"]
 
 
+def test_process_results_without_chunk(handler):
+    # given
+    results = [
+        {"metadata": "{}", "content": "Content 1"},
+        {"metadata": "{}", "content": "Content 2"},
+    ]
+
+    # when
+    data = handler.process_results(results)
+
+    # then
+    assert data == [[0, "Content 1"], [1, "Content 2"]]
+
+
 def test_process_results_null(handler):
     # given
     results = []