Fix image dimension calculation and tidy scripts

text/Dockerfile

@@ -4,9 +4,8 @@ WORKDIR /usr/src/app
 
 COPY requirements.txt ./
 RUN pip install --no-cache-dir -r requirements.txt
-RUN python -m nltk.downloader -d /usr/local/share/nltk_data wordnet
 
 WORKDIR /opt/work
 #COPY . .
 
-#CMD [ "python", "./your-daemon-or-script.py" ]
+#CMD [ "python", "./your-daemon-or-script.py" ]

text/build.sh

@@ -1,2 +1,2 @@
 #!/bin/bash
-docker build . -t python-data-art
+docker build . -t python-data-art

text/pil.py

@@ -8,8 +8,7 @@
     - https://stackoverflow.com/questions/704152/how-can-i-convert-a-character-to-a-integer-in-python-and-viceversa
 """
 import math
-from PIL import Image, ImageColor
-import nltk
+from PIL import Image
 
 
 """
@@ -46,10 +45,14 @@ def get_img_dim(text):
 
 
 
-height = int(math.sqrt(len(text))) / 3
-width = height
+num_pixels = math.ceil(len(text) / 3)
+height = width = int(math.ceil(math.sqrt(num_pixels)))
 print("[*] w = {0}, h = {1}".format(width, height))
 
+expected_len = width * height * 3
+if len(text) < expected_len:
+    text = text.ljust(expected_len, '\0')
+
 #get_img_dim(text)
 
 im = Image.new('RGBA', (width, height))
@@ -66,4 +69,4 @@ def get_img_dim(text):
         im.putpixel((x, y), (ord(text[i]), ord(text[i+1]), ord(text[i+2])))
         i = i + 3
 
-im.save('putPixel.png')
+im.save('putPixel.png')

text/requirements.txt

@@ -1,4 +1,2 @@
 ###### Requirements without Version Specifiers ######
-#nose
-#nose-cov
-nltk
+Pillow

text/run.sh

@@ -1,2 +1,2 @@
 #!/bin/bash
-docker container run -it --rm -v $(pwd):/opt/work python-data-art bash
+docker container run -it --rm -v $(pwd):/opt/work python-data-art bash