modules/ascii.py

@@ -16,7 +16,6 @@ import module
 from tools import web
 
 ASCII_CHARS = "$@%#*+=-:. "
-BRAIL_CHARS = "⠿⠾⠼⠸⠰⠠ "
 HEADERS = {'User-Agent': 'Gimme the ascii.'}
 
 
@@ -49,22 +48,21 @@ def scale_image(image, size=(100,100)):
 	return image
 
 
-def pixels_to_chars(image, scale="ascii"):
+def pixels_to_chars(image, reverse=False):
 	"""
 	Maps each pixel to an ascii char based on where it falls in the range
 	0-255 normalized to the length of ASCII_CHARS.
 	"""
-	scales = {"ascii": ASCII_CHARS,
-		"ascii_reverse": "".join(list(reversed(ASCII_CHARS))),
-		"brail": BRAIL_CHARS,
-		"brail_reverse": "".join(list(reversed(BRAIL_CHARS)))}
-	range_width = int(255 / len(scales[scale])) + (255 % len(scales[scale]) > 0)
+	range_width = int(255 / len(ASCII_CHARS)) + (255 % len(ASCII_CHARS) > 0)
 
 	pixels = list(image.getdata())
 	chars = []
 	for pixel in pixels:
+		if reverse:
+			index = -int(pixel/range_width)-1
+		else:
 			index = int(pixel/range_width)
-		chars.append(scales[scale][index])
+		chars.append(ASCII_CHARS[index])
 
 	chars = "".join(chars)
 	chars = [chars[i:i + image.size[0]] for i in range(0, len(chars),
@@ -151,71 +149,16 @@ def colorize(chars, image, code):
 	return "\n".join(chars)
 
 
-def alpha_composite(front, back):
-	"""Alpha composite two RGBA images.
-
-	Source: http://stackoverflow.com/a/9166671/284318
-
-	Keyword Arguments:
-	front -- PIL RGBA Image object
-	back -- PIL RGBA Image object
-
-	"""
-	front = np.asarray(front)
-	back = np.asarray(back)
-	result = np.empty(front.shape, dtype='float')
-	alpha = np.index_exp[:, :, 3:]
-	rgb = np.index_exp[:, :, :3]
-	falpha = front[alpha] / 255.0
-	balpha = back[alpha] / 255.0
-	result[alpha] = falpha + balpha * (1 - falpha)
-	old_setting = np.seterr(invalid='ignore')
-	result[rgb] = (front[rgb] * falpha + back[rgb] * balpha * (1 - falpha)) / result[alpha]
-	np.seterr(**old_setting)
-	result[alpha] *= 255
-	np.clip(result, 0, 255)
-	# astype('uint8') maps np.nan and np.inf to 0
-	result = result.astype('uint8')
-	result = Image.fromarray(result, 'RGBA')
-	return result
-
-
-def alpha_composite_with_color(image, color=(255, 255, 255)):
-	"""Alpha composite an RGBA image with a single color image of the
-	specified color and the same size as the original image.
-
-	Keyword Arguments:
-	image -- PIL RGBA Image object
-	color -- Tuple r, g, b (default 255, 255, 255)
-
-	"""
-	back = Image.new('RGBA', size=image.size, color=color + (255,))
-	return alpha_composite(image, back)
-
-
-def image_to_ascii(image, reverse=False, colors=None, brail=False):
+def image_to_ascii(image, reverse=False, colors=None):
 	"""
 	Reads an image file and converts it to ascii art. Returns a
 	newline-delineated string. If reverse is True, the ascii scale is
 	reversed.
 	"""
-	if image.mode == "RGBA":
-		image = alpha_composite_with_color(image).convert("RGB")
-
 	image = scale_image(image)
 	image_grey = image.convert('L')	# convert to grayscale
 
-	if reverse:
-		if brail:
-			scale = "brail_reverse"
-		else:
-			scale = "ascii_reverse"
-	else:
-		if brail:
-			scale = "brail"
-		else:
-			scale = "ascii"
-	chars = pixels_to_chars(image_grey, scale)
+	chars = pixels_to_chars(image_grey, reverse)
 
 	if colors:
 		chars = colorize(chars, image, colors)
@@ -245,7 +188,7 @@ def handle_gif(image, output, reverse=False):
 	"""
 	Handle gifs seperately.
 	"""
-	# image = open_image(args.imagePath)
+	image = open_image(args.imagePath)
 	ascii_seq = []
 	new_image = ascii_to_image(image_to_ascii(image, reverse))
 	image.seek(1)
@@ -273,14 +216,11 @@ def ascii(bot, trigger):
 	"""
 	Downloads an image and converts it to ascii.
 	"""
-	if not trigger.group(2):
-		return bot.say()
 	parser = argparse.ArgumentParser()
 	parser.add_argument("imagePath")
 	parser.add_argument("-r", "--reverse", action="store_true")
 	parser.add_argument("-c", "--color", action="store_true")
-	parser.add_argument("-b", "--brail", action="store_true")
-	parser.add_argument("-a", "--animated", action="store_true")
+	parser.set_defaults(reverse=False, color=False)
 	args = parser.parse_args(trigger.group(2).split())
 
 	if args.color:
@@ -294,14 +234,7 @@ def ascii(bot, trigger):
 		return
 
 	image = open_image(args.imagePath)
-	if args.animated:
-		handle_gif(image, "temp.png", args.reverse)
-		file = {"file": open("temp.png", "rb")}
-		res = requests.post("https://uguu.se/api.php?d=upload-tool", files=file)
-		# print(res.text)
-		bot.say(res.text)
-	else:
-		image_ascii = image_to_ascii(image, args.reverse, args.color, args.brail)
+	image_ascii = image_to_ascii(image, args.reverse, args.color)
 	bot.say(image_ascii)
 
 

modules/wiktionary.py

@@ -28,8 +28,7 @@ def text(html):
 
 def wikt(word):
     bytes = requests.get(uri.format(word))
-#    bytes = r_ul.sub('', bytes)
-    bytes = bytes.text
+    bytes = r_ul.sub('', bytes)
 
     mode = None
     etymology = None