so limiting

1 files changed, 12 insertions, 8 deletions

diff --git a/rpc/rpc.py b/rpc/rpc.py
index 3ea7cd6..82db460 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -66,15 +66,19 @@ class CortexRPC(object):
     print('Get params call received!')
     return self._get_fn()
 
-  def send_pil_image(self, fn, meta, im):
+  def send_pil_image(self, fn, meta, im, format='JPEG'):
     output = BytesIO()
-    im = im.convert('RGB')
-    # im.save(output, format='JPEG')
-    im.save(output, format='PNG')
-    frame = output.getvalue()
-    output.close()
-    # self.send_frame(fn.replace('png', 'jpg'), meta, frame)
-    self.send_frame(fn, meta, frame)
+    if format == 'PNG':
+      im = im.convert('RGB')
+      im.save(output, format='PNG')
+      frame = output.getvalue()
+      output.close()
+      self.send_frame(fn, meta, frame)
+  else:
+      im.save(output, format='JPEG')
+      frame = output.getvalue()
+      output.close()
+      self.send_frame(fn.replace('png', 'jpg'), meta, frame)
 
   def send_status(self, key, value):
     if self.relay is not None: