Update handler.py

handler.py

@@ -71,16 +71,18 @@ class EndpointHandler():
 
         pc = sampler.output_to_point_clouds(samples)[0]
         print('type of pc: ', type(pc))
+
+        pc_dict = {}
         
         data_list = pc.coords.tolist()
         json_string = json.dumps(data_list)
-        return {"data": json_string}
-        #print(pc)
+        pc_dict['data'] = json_string
 
-        # encode image as base 64
-        #buffered = BytesIO()
-        #image.save(buffered, format="JPEG")
-        #img_str = base64.b64encode(buffered.getvalue())
+        # Convert NumPy arrays to Python lists for serializing
+        serializable_channels = {key: value.tolist() for key, value in pc.channels.items()}
 
-        # postprocess the prediction
-        #return {"image": img_str.decode()}
+        # Serialize the dictionary to a JSON-formatted string
+        channel_data = json.dumps(serializable_channels)
+        pc_dict['channels'] = channel_data
+        
+        return pc_dict