Update main.py

main.py

@@ -1,3 +1,82 @@
+import sys, random, argparse
+import numpy as np
+import math
+from PIL import Image, ImageFont, ImageDraw
+import gradio as gr
+
+# 70 levels of gray
+gscale1 = "$@B%8&WM#*oahkbdpqwmZO0QLCJUYXzcvunxrjft/|()1{}[]?-_+~<>i!lI;:,\\"^`'. "
+
+# 10 levels of gray
+gscale2 = '@%#*+=-:. '
+
+def getAverageL(image):
+    im = np.array(image)
+    w, h = im.shape
+    return np.average(im.reshape(w*h))
+
+def covertImageToAscii(input_img, cols, scale, moreLevels):
+    global gscale1, gscale2
+
+    image = Image.fromarray(input_img).convert('L')
+    W, H = image.size[0], image.size[1]
+
+    w = W/cols
+    h = w/scale
+    rows = int(H/h)
+
+    if cols > W or rows > H:
+        print("Image too small for specified cols!")
+        exit(0)
+
+    aimg = []
+    for j in range(rows):
+        y1 = int(j*h)
+        y2 = int((j+1)*h)
+
+        if j == rows-1:
+            y2 = H
+
+        aimg.append("")
+
+        for i in range(cols):
+            x1 = int(i*w)
+            x2 = int((i+1)*w)
+
+            if i == cols-1:
+                x2 = W
+
+            img = image.crop((x1, y1, x2, y2))
+            avg = int(getAverageL(img))
+
+            if moreLevels:
+                gsval = gscale1[int((avg*69)/255)]
+            else:
+                gsval = gscale2[int((avg*9)/255)]
+
+            aimg[j] += gsval
+    
+    return aimg
+
+def sepia(input_img):
+    aimg = covertImageToAscii(input_img, 200, 0.43, False)
+
+    my_image = Image.new(mode="RGB", size=(2000, 2000), color=(0, 0, 0))
+    image_editable = ImageDraw.Draw(my_image)
+    image_editable.text((10, 10), "\n".join(aimg), (237, 230, 211))
+
+    return [my_image, "\n".join(aimg)]
+
+iface = gr.Interface(sepia, 
+                     gr.inputs.Image(shape=(200, 200)), 
+                     ["image", "text"],
+                     title = "ASCII Art",
+                     description = "Convert an image to ASCII art based on ascii character density. Copy and paste the text to a notepad to see it correctly")
+
+
+iface.launch(server_name="0.0.0.0", server_port=7860)
+
+'''
 import gradio as gr
 import numpy as np
 from PIL import Image
@@ -38,8 +117,7 @@ iface = gr.Interface(
 )
 
 iface.launch(server_name="0.0.0.0", server_port=7860)
-
-
+'''
 
 '''
 import gradio as gr