From e666384792d2825bdb92605d572cdabc2c895dec Mon Sep 17 00:00:00 2001
From: kilroythethird <44308116+kilroythethird@users.noreply.github.com>
Date: Tue, 27 Aug 2019 15:58:24 +0200
Subject: [PATCH] Fixed convert bug in preview (#846)

---
 lib/convert.py   |  2 ++
 tools/preview.py | 11 ++++++-----
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/lib/convert.py b/lib/convert.py
index c09703e5..3f782f2e 100644
--- a/lib/convert.py
+++ b/lib/convert.py
@@ -86,6 +86,8 @@ class Converter():
                 in_queue.put(items)
                 break
 
+            if isinstance(items, dict):
+                items = [items]
             for item in items:
                 logger.trace("Patch queue got: '%s'", item["filename"])
                 try:
diff --git a/tools/preview.py b/tools/preview.py
index 78a925df..570ed26b 100644
--- a/tools/preview.py
+++ b/tools/preview.py
@@ -245,13 +245,14 @@ class Samples():
             idx = 0
             while idx < self.sample_size:
                 logger.debug("Predicting face %s of %s", idx + 1, self.sample_size)
-                item = self.predictor.out_queue.get()
-                if item == "EOF":
+                items = self.predictor.out_queue.get()
+                if items == "EOF":
                     logger.debug("Received EOF")
                     break
-                self.predicted_images.append(item)
-                logger.debug("Predicted face %s of %s", idx + 1, self.sample_size)
-                idx += 1
+                for item in items:
+                    self.predicted_images.append(item)
+                    logger.debug("Predicted face %s of %s", idx + 1, self.sample_size)
+                    idx += 1
         logger.debug("Predicted faces")