OCR Engine Support: Pororo (`kakaobrain/pororo`, potentially used together with EasyOCR)

Sometimes a designer and dreamer, I just like to build—or hack things. Check my website for my work.

🦄 Anything for Creation. Preparing the infrastructure—insight, followers, and capital—for a bigger dream.
⚡️ Someone who can start something right away, when they have an idea.

	for idx, item in enumerate(items):
	box = item["box"]

	# [x,y,w,h]
	if len(box) == 4 and isinstance(box[0], int):
	rect = rectangle_corners(box)
	items[idx]["box"] = rect

	# [[x,y],[w,h]]
	elif len(box) == 2 and isinstance(box[0], list) and len(box[0]) == 2:
	flattened = [i for sublist in box for i in sublist]
	rect = rectangle_corners(flattened)
	items[idx]["box"] = rect

	print("=====")
	print(prompt)

	api_key = os.environ["OPENAI_API_KEY"]
	if "API_KEY" in options["openai"] and options["openai"]["API_KEY"] != "":
	api_key = options["openai"]["API_KEY"]
	client = OpenAI(
	api_key=api_key,
	)

	print("=====")

	completion = client.chat.completions.create(
	messages=[
	{"role": "user", "content": prompt},
	],
	**options["openai"],
	)
	output = completion.choices[0].message.content
	output = output.replace("\n", "")
	print("[*] LLM", output)

junhoyeo / betterocr Goto Github PK