Merge pull request #5 from navanchauhan/deepsource-transform-8bc97824

Format code with Black

1 files changed, 48 insertions, 23 deletions

diff --git a/src/VHSImage.py b/src/VHSImage.py
index ce6411a..d80af92 100644
--- a/src/VHSImage.py
+++ b/src/VHSImage.py
@@ -2,7 +2,7 @@ import requests
 import datetime
 from PIL import Image
 from PIL import ImageFont
-from PIL import ImageDraw 
+from PIL import ImageDraw
 import colorsys
 import json
 import numpy as np
@@ -21,7 +21,7 @@ def generate_offsets(array_size, max_offset):
     periodicity = random.random() * periodicity
     offsets = []
     for i in range(array_size):
-        offsets.append(floor(max_offset*np.sin(periodicity*(i*np.pi/180))))
+        offsets.append(floor(max_offset * np.sin(periodicity * (i * np.pi / 180))))
     return offsets
 
 
@@ -35,11 +35,11 @@ def hueChange(img, offset):
     g_data = []
     b_data = []
 
-    offset = offset/100.
+    offset = offset / 100.0
     for rd, gr, bl in zip(r.getdata(), g.getdata(), b.getdata()):
-        h, s, v = colorsys.rgb_to_hsv(rd/255.0, bl/255.0, gr/255.0)
-        rgb = colorsys.hsv_to_rgb(h, s+offset, v)
-        rd, bl, gr = [int(x*255.) for x in rgb]
+        h, s, v = colorsys.rgb_to_hsv(rd / 255.0, bl / 255.0, gr / 255.0)
+        rgb = colorsys.hsv_to_rgb(h, s + offset, v)
+        rd, bl, gr = [int(x * 255.0) for x in rgb]
         r_data.append(rd)
         g_data.append(gr)
         b_data.append(bl)
@@ -47,12 +47,21 @@ def hueChange(img, offset):
     r.putdata(r_data)
     g.putdata(g_data)
     b.putdata(b_data)
-    return Image.merge('RGB',(r,g,b))
+    return Image.merge("RGB", (r, g, b))
+
 
 def decision(probability):
     return random.random() < probability
 
-def mod_image_repeat_rows(imgname, chance_of_row_repeat=0, max_row_repeats=0, min_row_repeats=0, save=True, out_name="image.jpg"):
+
+def mod_image_repeat_rows(
+    imgname,
+    chance_of_row_repeat=0,
+    max_row_repeats=0,
+    min_row_repeats=0,
+    save=True,
+    out_name="image.jpg",
+):
     img = Image.open(imgname)
     pixels = img.load()
     width, height = img.size
@@ -80,7 +89,7 @@ def mod_image_repeat_rows(imgname, chance_of_row_repeat=0, max_row_repeats=0, mi
                     pixels[x, y] = row_to_repeat[x - offsets[num_repeats]]
             else:
                 pixels[x, y] = (r, g, b)
-        
+
         if repeat:
             num_repeats += 1
             if num_repeats >= times_to_repeat:
@@ -102,53 +111,68 @@ def add_date(img_path, out_name="image.jpg", bottom_offset=0):
     width, height = img.size
     draw = ImageDraw.Draw(img)
     font = ImageFont.truetype("src/VCR_OSD_MONO_1.001.ttf", 64)
-    draw.text((corner_offset, (height-150-bottom_offset)), date_str_1, (255, 255, 255), font=font)
-    draw.text((corner_offset, (height-75)-bottom_offset), date_str_2, (255, 255, 255), font=font)
+    draw.text(
+        (corner_offset, (height - 150 - bottom_offset)),
+        date_str_1,
+        (255, 255, 255),
+        font=font,
+    )
+    draw.text(
+        (corner_offset, (height - 75) - bottom_offset),
+        date_str_2,
+        (255, 255, 255),
+        font=font,
+    )
     draw.text((corner_offset, 25), "|| PAUSE", (255, 255, 255), font=font)
     img.save(out_name)
 
+
 def add_img_noise(imgpath, intensity=1, out_name="image.jpg"):
-    img = imageio.imread(imgpath, pilmode='RGB')
+    img = imageio.imread(imgpath, pilmode="RGB")
     noise1 = img + intensity * img.std() * np.random.random(img.shape)
     imageio.imwrite(out_name, noise1)
 
+
 def offset_hue(image, out_name="image.jpg"):
     if isinstance(image, str):
         image = Image.open(image)
         image = hueChange(image, 25)
         image.save(out_name)
 
+
 def build_background(out_name, taskbar_offset):
-    #getImage(out_name="start.jpg")
+    # getImage(out_name="start.jpg")
     offset_hue("start.jpg", out_name="saturated.jpg")
     mod_image_repeat_rows("saturated.jpg", 0.012, 50, 10, out_name="shifted.jpg")
     add_img_noise("shifted.jpg", out_name="noisy.jpg")
     add_date("noisy.jpg", out_name=out_name, bottom_offset=taskbar_offset)
 
+
 """
 if __name__ == "__main__":
     build_background("bkg.jpg", 25)
-"""    
+"""
+
 
 def generateVHSStyle(infile, outfile, silence=False):
     if silence:
         cut_rows = bool(random.getrandbits(1))
-        offset = random.choice([0,5,10])
-        offset_hue(infile,"saturated.jpg")
+        offset = random.choice([0, 5, 10])
+        offset_hue(infile, "saturated.jpg")
         if cut_rows:
             mod_image_repeat_rows("saturated.jpg", 0.012, 50, 10, True, "shifted.jpg")
         else:
             mod_image_repeat_rows("saturated.jpg", 0, 0, 0, True, "shifted.jpg")
-        add_date("shifted.jpg","noisy.jpg")
-        add_date("noisy.jpg",outfile, bottom_offset=offset)
+        add_date("shifted.jpg", "noisy.jpg")
+        add_date("noisy.jpg", outfile, bottom_offset=offset)
         os.remove("shifted.jpg")
         os.remove("saturated.jpg")
         os.remove("noisy.jpg")
     else:
         cut_rows = bool(random.getrandbits(1))
-        offset = random.choice([0,5,10,15,20,25])
+        offset = random.choice([0, 5, 10, 15, 20, 25])
         logger.info("Saturating the image")
-        offset_hue(infile,"saturated.jpg")
+        offset_hue(infile, "saturated.jpg")
         if cut_rows:
             logger.info("Shifting the image")
             mod_image_repeat_rows("saturated.jpg", 0.012, 50, 10, True, "shifted.jpg")
@@ -156,13 +180,14 @@ def generateVHSStyle(infile, outfile, silence=False):
             logger.info("Not applying lines effect")
             mod_image_repeat_rows("saturated.jpg", 0, 0, 0, True, "shifted.jpg")
         logger.info("Adding noise")
-        add_date("shifted.jpg","noisy.jpg")
+        add_date("shifted.jpg", "noisy.jpg")
         logger.info("Adding text")
-        add_date("noisy.jpg",outfile, bottom_offset=offset)
+        add_date("noisy.jpg", outfile, bottom_offset=offset)
         logger.info("Generated Image: out.jpg")
         logger.info("Removing residual files")
         os.remove("shifted.jpg")
         os.remove("saturated.jpg")
         os.remove("noisy.jpg")
 
-#generateVHSStyle("s.jpg","o.jpg")
\ No newline at end of file
+
+# generateVHSStyle("s.jpg","o.jpg")