modules/im2bin.py at master · KeyZenD/modules · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
import asyncio
import logging
from PIL import Image, ImageDraw, ImageFont, ImageOps
import io
from requests import get
from string import digits
from random import choice
from .. import loader, utils

logger = logging.getLogger(__name__)

font_ = get("https://github.com/KeyZenD/l/blob/master/mono.otf?raw=true").content

@loader.tds
class Im2BinaryMod(loader.Module):
	"""Картинки в текст. что?"""
	strings = {"name": "Im2Bin"}

	@loader.unrestricted
	async def bincmd(self, message):
		""".bin <картинка  или реплай> + слова (дефолт на рандоме) (не мешает слова)"""
		img, words, me = await prepare(message)
		if not img:
			await message.delete()
			return await message.client.send_file(message.chat.id, get("https://thiscatdoesnotexist.com").content, caption=choice(["<b>Тебе картинок мало?</b>"]+[None]*100))
		await message.edit("<b>Processing...</b>")
		img = await image_to_text(words, img, False)
		[await message.delete(), await (await message.get_reply_message()).reply(file=img)] if not me else await message.edit(file=img, text="")
	@loader.unrestricted
	async def rbincmd(self, message):
		""".rbin <картинка  или реплай> + слова (дефолт на рандоме) (мешает слова)"""
		img, words, me = await prepare(message)
		if not img:
			await message.delete()
			return await message.client.send_file(message.chat.id, get("https://thiscatdoesnotexist.com").content, caption=choice(["<b>Тебе картинок мало?</b>"]+[None]*100))
		await message.edit("<b>Processing...</b>")
		img = await image_to_text(words, img, True)
		[await message.delete(), await (await message.get_reply_message()).reply(file=img)] if not me else await message.edit(file=img, text="")

async def getimg(m):
	if not m.file:
		return False
	if not "image" in m.file.mime_type.lower():
		return False
	return True

async def prepare(message):
	if not await getimg(message):
		reply = await message.get_reply_message()
		if not reply or not await getimg(reply):
			return False, False, False
		else:
			me = False
			img = await reply.download_media(bytes)
	else:
		me = True
		img = await message.download_media(bytes)
	args = utils.get_args(message)
	words = [f"{x} " for x in args] if args else list("01")
	return img, words, me

async def image_to_text(words, img, rand):
	inp = Image.open(io.BytesIO(img))
	img = Image.new("RGBA", inp.size, "#000")
	res = img.copy()
	img.paste(inp, (0, 0), inp if inp.mode == "RGBA" else None)
	w, h = img.size
	font = ImageFont.truetype(io.BytesIO(font_), 15)
	mw = min(map(lambda x: font.getsize(x)[0], "".join(words)))
	mh = min(map(lambda x: font.getsize(x)[1], "".join(words)))
	rand_ = 0
	text = []
	while len(text)*mh <= h:
		row = []
		while len("".join(row))*mw <= w:
			word = choice(words) if rand else words[rand_%len(words)]
			rand_ += 1
			row.append(word)
		rand_ -= 1
		text.append("".join(row))
	text = "\n".join(text)
	wt, ht = ImageDraw.Draw(Image.new("L", (0, 0))).multiline_textsize(font=font, text=text, spacing=0)
	im = Image.new("L", (wt, ht), 0)
	ImageDraw.Draw(im).multiline_text((0, -3), font=font, text=text, spacing=0, fill=255)
	im = im.crop((0, 0, w, h))
	im = Image.frombytes("L", (w, h), bytes([255 if x > 150 else 0 for x in im.tobytes()]))
	img.putalpha(im)
	res.paste(img, (0, 0), img)
	out = io.BytesIO()
	out.name = words[0] + ".png"
	res.save(out)
	out.seek(0)
	return out