Spaces:

HighCWu
/

colorful-ascii-art

Build error

App Files Files Community

HighCWu commited on May 4, 2022

Commit

067ec2e

1 Parent(s): f53177b

init commits.

Browse files

Files changed (3) hide show

README.md +1 -1
app.py +204 -0
requirements.txt +2 -0

README.md CHANGED Viewed

@@ -10,4 +10,4 @@ pinned: false
 license: mit
 ---
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference

 license: mit
 ---
+Largely referenced from [this space](https://huggingface.co/spaces/dt/ascii-art)

app.py ADDED Viewed

	@@ -0,0 +1,204 @@

+'''
+Refer to https://huggingface.co/spaces/dt/ascii-art/blob/main/app.py
+'''
+# Python code to convert an image to ASCII image.
+import sys, random, argparse
+import numpy as np
+import math
+import gradio as gr
+from PIL import Image, ImageFont, ImageDraw
+# 70 levels of gray
+gscale1 = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/\|()1{}[]?-_+~<>i!lI;:,\"^`'. "
+# 10 levels of gray
+gscale2 = '@%#*+=-:. '
+font = ImageFont.load_default()
+def getAverageL(image):
+    """
+    Given PIL Image, return average value of grayscale value
+    """
+    # get image as numpy array
+    im = np.array(image)
+    # get shape
+    w,h = im.shape
+    # get average
+    return np.average(im.reshape(w*h))
+def covertImageToAscii(input_img, cols, scale, moreLevels):
+    """
+    Given Image and dims (rows, cols) returns an m*n list of Images
+    """
+    # declare globals
+    global gscale1, gscale2
+    # open image and convert to grayscale
+    image = input_img.convert('L')
+    # store dimensions
+    # store dimensions
+    W, H = image.size[0], image.size[1]
+    print("input image dims: %d x %d" % (W, H))
+    # compute width of tile
+    w = W/cols
+    # compute tile height based on aspect ratio and scale
+    h = w/scale
+    # compute number of rows
+    rows = int(H/h)
+    print("cols: %d, rows: %d" % (cols, rows))
+    print("tile dims: %d x %d" % (w, h))
+    # check if image size is too small
+    if cols > W or rows > H:
+        print("Image too small for specified cols!")
+        exit(0)
+    # ascii image is a list of character strings
+    aimg = []
+    # generate list of dimensions
+    for j in range(rows):
+        y1 = int(j*h)
+        y2 = int((j+1)*h)
+        # correct last tile
+        if j == rows-1:
+            y2 = H
+        # append an empty string
+        aimg.append("")
+        for i in range(cols):
+            # crop image to tile
+            x1 = int(i*w)
+            x2 = int((i+1)*w)
+            # correct last tile
+            if i == cols-1:
+                x2 = W
+            # crop image to extract tile
+            img = image.crop((x1, y1, x2, y2))
+            # get average luminance
+            avg = int(getAverageL(img))
+            # look up ascii char
+            if moreLevels:
+                gsval = gscale1[int((avg*69)/255)]
+            else:
+                gsval = gscale2[int((avg*9)/255)]
+            # append ascii char to string
+            aimg[j] += gsval
+    # return txt image
+    return aimg
+def colorizeTextImage(input_img, text_img):
+    input_img = np.asarray(input_img)
+    input_img = input_img.reshape((
+        input_img.shape[0]//11,
+        11,
+        input_img.shape[1]//6,
+        6,
+        3
+    ))
+    input_img = np.float32(input_img)
+    text_img = np.asarray(text_img)
+    text_img = text_img.reshape((
+        input_img.shape[0],
+        11,
+        input_img.shape[2],
+        6,
+        3
+    ))
+    alpha = np.float32(text_img)[...,:1] / 255
+    out_img = input_img * alpha
+    out_colors = out_img.sum((1,3), keepdims=True) / (alpha.sum((1,3), keepdims=True) + 1e-12)
+    out_img = out_colors * alpha
+    out_img = np.concatenate([out_img, alpha * 255], -1)
+    out_img = out_img.reshape((
+        out_img.shape[0] * out_img.shape[1],
+        out_img.shape[2] * out_img.shape[3],
+        4
+    ))
+    out_img = np.clip(out_img, 0, 255)
+    out_img = np.uint8(out_img)
+    out_img = Image.fromarray(out_img)
+    our_colors = np.clip(out_colors, 0, 255)
+    our_colors = np.uint8(out_colors)[:,0,:,0]
+    return out_img, our_colors
+def convertTextToHTML(our_colors, aimg):
+    bimg = r'''
+<!DOCTYPE html>
+<html>
+<body>
+<style>text{ font-size:8px; }</style>
+<svg xmlns="http://www.w3.org/2000/svg" version="1.1" style="width: 100vw;height: 100vh;" viewBox="0, 0, 1200, 1200">
+'''
+    for i in range(our_colors.shape[0]):
+        our_colors2 = our_colors[i]
+        aimg2 = aimg[i]
+        for j in range(our_colors2.shape[0]):
+            [r, g, b] = our_colors2[j]
+            p = aimg2[j]
+            if p == ' ': continue
+            aimg3 = f'<text x="{j*6}" y="{i*11}" style="fill:rgb{int(r),int(g),int(b)};">{p}</text>\n'
+            bimg += aimg3
+    bimg += r'''
+</svg>
+</body>
+</html>
+'''
+    return bimg
+def sepia(input_img):
+    input_img = Image.fromarray(input_img).convert('RGB')
+    aimg = covertImageToAscii(input_img, 200, 6/11, True)
+    blank_image = Image.new(mode="RGB", size=(len(aimg[0])*6, len(aimg)*11), color=(0, 0, 0))
+    my_image = blank_image.copy()
+    image_editable = ImageDraw.Draw(my_image)
+    image_editable.text((0, 0), "\n".join(aimg), (255, 255, 255), font=font, spacing=0)
+    input_img_resize = input_img.resize((len(aimg[0])*6, len(aimg)*11), Image.BICUBIC)
+    w, h = input_img.size
+    scale = 200 * 6 / w
+    w = 200 * 6
+    h = int(round(h*scale))
+    input_img = input_img.resize((200 * 6, h), Image.BICUBIC)
+    input_img_resize.paste(input_img, (0, 0, w, h))
+    input_img = input_img_resize
+    my_image, my_colors = colorizeTextImage(input_img, my_image)
+    my_html = convertTextToHTML(my_colors, aimg)
+    return [my_image, my_html]
+iface = gr.Interface(sepia,
+                     gr.inputs.Image(),
+                     ["image", "html"],
+                     title = "Colorful ASCII Art",
+                     description = "Convert an image to colorful ASCII art based on ascii character density. Copy and paste the text to a notepad to see it correctly")
+iface.launch()

requirements.txt ADDED Viewed

	@@ -0,0 +1,2 @@


1	+ Pillow
2	+ gradio