Fixed QtWidgets not imported on some components.

4 files changed, 5 insertions, 5 deletions

diff --git a/src/components/color.py b/src/components/color.py
index 2e3902a..44ed82e 100644
--- a/src/components/color.py
+++ b/src/components/color.py
@@ -1,5 +1,5 @@
 from PIL import Image, ImageDraw
-from PyQt5 import uic, QtGui, QtCore
+from PyQt5 import uic, QtGui, QtCore, QtWidgets
 from PyQt5.QtGui import QColor
 from PIL.ImageQt import ImageQt
 import os
diff --git a/src/components/original.py b/src/components/original.py
index 61f463d..0d5001c 100644
--- a/src/components/original.py
+++ b/src/components/original.py
@@ -1,6 +1,6 @@
 import numpy
 from PIL import Image, ImageDraw
-from PyQt5 import uic, QtGui, QtCore
+from PyQt5 import uic, QtGui, QtCore, QtWidgets
 from PyQt5.QtGui import QColor
 import os
 from . import __base__
diff --git a/src/components/text.py b/src/components/text.py
index 0f599ed..76961c9 100644
--- a/src/components/text.py
+++ b/src/components/text.py
@@ -1,6 +1,6 @@
 from PIL import Image, ImageDraw
 from PyQt5.QtGui import QPainter, QColor, QFont
-from PyQt5 import uic, QtGui, QtCore
+from PyQt5 import uic, QtGui, QtCore, QtWidgets
 from PIL.ImageQt import ImageQt
 import os
 import io
@@ -138,7 +138,7 @@ class Component(__base__.Component):
         painter.drawText(x, y, self.title)
         painter.end()
 
-        imBytes = image.bits().asstring(image.numBytes())
+        imBytes = image.bits().asstring(image.byteCount())
 
         return Image.frombytes('RGBA', (width, height), imBytes)
 
diff --git a/src/components/video.py b/src/components/video.py
index 0090426..70247e1 100644
--- a/src/components/video.py
+++ b/src/components/video.py
@@ -1,5 +1,5 @@
 from PIL import Image, ImageDraw
-from PyQt5 import uic, QtGui, QtCore
+from PyQt5 import uic, QtGui, QtCore, QtWidgets
 import os
 import subprocess
 import threading