PULL request

filter.py

@@ -1,28 +1,37 @@
-from PIL import Image
 import numpy as np
-img = Image.open("img2.jpg")
-arr = np.array(img)
-a = len(arr)
-a1 = len(arr[1])
-i = 0
-while i < a - 11:
-    j = 0
-    while j < a1 - 11:
-        s = 0
-        for n in range(i, i + 10):
-            for n1 in range(j, j + 10):
-                n1 = arr[n][n1][0]
-                n2 = arr[n][n1][1]
-                n3 = arr[n][n1][2]
-                M = n1 + n2 + n3
-                s += M
-        s = int(s // 100)
-        for n in range(i, i + 10):
-            for n1 in range(j, j + 10):
-                arr[n][n1][0] = int(s // 50) * 50
-                arr[n][n1][1] = int(s // 50) * 50
-                arr[n][n1][2] = int(s // 50) * 50
-        j = j + 10
-    i = i + 10
-res = Image.fromarray(arr)
-res.save('res.jpg')
+from PIL import Image
+
+
+def find_step(n):
+    return int(255 / (n - 1) - 1 if 255 % (n - 1) == 0 else 255 / (n - 1))
+
+
+def find_av_brightness(i, j, arr, m_h, m_w, step):
+    br = 0.
+    br += arr[i: i + m_h, j: j + m_w, :].sum() / 3
+    br //= m_h * m_w
+    br -= br % step
+    return br
+
+
+def do_mosaic(arr, m_h, m_w, step):
+    for i in range(0, len(arr), m_h):
+        for j in range(0, len(arr[1]), m_w):
+            brightness = find_av_brightness(i, j, arr, m_h, m_w, step)
+            arr[i: i + m_h, j: j + m_w, :] = brightness
+
+
+inp_im_names = input('Enter the names of the source image (jpg) and the '
+                     'result separated by comma (for example: img2,res): ') \
+    .split(',')
+inp_sizes = input('Enter the height and width of the mosaic element '
+                  'separated by comma (for example: 10,10): ')
+num_grad = int(input('Enter the number of gradations(for example: 6): '))
+
+inp_im = np.array(Image.open(inp_im_names[0] + ".jpg"))
+
+e_h, e_w = map(int, inp_sizes.split(','))
+do_mosaic(inp_im, e_h, e_w, find_step(num_grad))
+
+res = Image.fromarray(inp_im)
+res.save(inp_im_names[1] + ".jpg")