import io import tempfile import subprocess import requests import base64 from flask import Flask, render_template_string, request, send_from_directory, session from PIL import Image, ImageDraw, ImageFont import os import re import argparse app = Flask(__name__) app.secret_key = 'TpfxYMxUJxJMtCCTraBSg1rbd6NLz38JTmIfpBLsotcI47EqXU' FONT_PATH = "/usr/share/fonts/truetype/DejaVuSansMono.ttf" FONT_PATH_BOLD = "/usr/share/fonts/truetype/DejaVuSansMono-Bold.ttf" FONT_PATH_OBLIQUE = "/usr/share/fonts/truetype/DejaVuSansMono-Oblique.ttf" FONT_PATH_BOLDITALIC = "/usr/share/fonts/truetype/DejaVuSansMono-BoldOblique.ttf" FONT_SIZE = 24 HEADER_SIZE_1 = 56 HEADER_SIZE_2 = 34 HEADER_SIZE_3 = 28 IMAGE_WIDTH = 384 BULLET_CHAR = "• " HTML_FORM = ''' CatNote

😺 CatNote 🖨️




{% if printed %}
✅ Enviado a impresora
{% endif %} {% if error %}
⚠️ {{ error }}
{% endif %}
{% if img %}

Vista previa

{% else %}

Su vista previa aparecerá aquí

{% endif %}

Referencia rápida de Markdown

''' def remove_uploaded_img(): path = session.pop('uploaded_img_path', None) if path and os.path.exists(path): try: os.remove(path) except Exception: pass def bleh_image_from_url(url): resp = requests.get(url, stream=True) resp.raise_for_status() bleh = subprocess.Popen( ["./bleh", "-o", "-", "-mode", "1bpp", "-d", "floyd", "-"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE) out, err = bleh.communicate(resp.content) if bleh.returncode != 0: raise RuntimeError(f"Driver failed: {err.decode()}") img = Image.open(io.BytesIO(out)).convert("L") # Optionally check width, pad/resize if needed if img.width != IMAGE_WIDTH: img = img.resize((IMAGE_WIDTH, img.height), Image.LANCZOS) return img def bleh_image_from_bytes(image_bytes): bleh = subprocess.Popen( ["./bleh", "-o", "-", "-mode", "1bpp", "-d", "floyd", "-"], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE ) out, err = bleh.communicate(image_bytes) if bleh.returncode != 0: raise RuntimeError(f"Driver failed: {err.decode()}") img = Image.open(io.BytesIO(out)).convert("L") if img.width != IMAGE_WIDTH: img = img.resize((IMAGE_WIDTH, img.height), Image.LANCZOS) return img def parse_line(line): if re.match(r"!\(img\)", line) or re.match(r"!\[.*\]\(img\)", line): return ('userimage', None) image_match = re.match(r"^!\[(.*?)\]\((.+?)\)", line) if image_match: alt = image_match.group(1) url = image_match.group(2) return ('image', url, alt) if re.match(r'^\s*---\s*$', line): return ('hr', []) header_match = re.match(r"^(#{1,3}) +(.*)", line) if header_match: header_level = len(header_match.group(1)) line = header_match.group(2) return ('header', header_level, parse_segments(line)) bullet_match = re.match(r"^\s*([-*\u2022]) +(.*)", line) if bullet_match: return ('bullet', parse_segments(bullet_match.group(2))) ordered_match = re.match(r"^\s*(\d+)\. +(.*)", line) if ordered_match: return ('ordered', int(ordered_match.group(1)), parse_segments(ordered_match.group(2))) return ('text', parse_segments(line)) def parse_segments(line): # Handle escaped asterisks: replace them with a placeholder line = line.replace(r'\*', '\x07') # Apply formatting bi = re.findall(r"\*\*\*(.+?)\*\*\*", line) for x in bi: line = line.replace(f"***{x}***", f"\x01{x}\x02") b = re.findall(r"\*\*(.+?)\*\*", line) for x in b: line = line.replace(f"**{x}**", f"\x03{x}\x04") i = re.findall(r"\*(.+?)\*", line) for x in i: line = line.replace(f"*{x}*", f"\x05{x}\x06") # Split into styled segments segments = [] i = 0 while i < len(line): if line[i] == '\x01': i += 1 start = i while i < len(line) and line[i] != '\x02': i += 1 segments.append(('bolditalic', line[start:i])) i += 1 elif line[i] == '\x03': i += 1 start = i while i < len(line) and line[i] != '\x04': i += 1 segments.append(('bold', line[start:i])) i += 1 elif line[i] == '\x05': i += 1 start = i while i < len(line) and line[i] != '\x06': i += 1 segments.append(('italic', line[start:i])) i += 1 else: start = i while i < len(line) and line[i] not in '\x01\x03\x05': i += 1 if i > start: segments.append(('text', line[start:i])) # Restore literal asterisks segments = [(style, text.replace('\x07', '*')) for style, text in segments] return segments def font_for_style(style, font, font_bold, font_italic, font_bolditalic): if style == 'bolditalic': return font_bolditalic or font_bold or font_italic or font elif style == 'bold': return font_bold elif style == 'italic': return font_italic else: return font def wrap_segments(segments, font, font_bold, font_italic, font_bolditalic, max_width, start_x=0): line = [] x = start_x for style, text in segments: words = re.split(r'(\s+)', text) for word in words: if word == '': continue f = font_for_style(style, font, font_bold, font_italic, font_bolditalic) w = f.getbbox(word)[2] - f.getbbox(word)[0] if x + w > max_width and line: yield line line = [] x = start_x line.append((style, word)) x += w if line: yield line def render(md, uploaded_img_bytes=None): font = ImageFont.truetype(FONT_PATH, FONT_SIZE) font_bold = ImageFont.truetype(FONT_PATH_BOLD, FONT_SIZE) font_italic = ImageFont.truetype(FONT_PATH_OBLIQUE, FONT_SIZE) try: font_bolditalic = ImageFont.truetype(FONT_PATH_BOLDITALIC, FONT_SIZE) except: font_bolditalic = None font_h1 = ImageFont.truetype(FONT_PATH_BOLD, HEADER_SIZE_1) font_h2 = ImageFont.truetype(FONT_PATH_BOLD, HEADER_SIZE_2) font_h3 = ImageFont.truetype(FONT_PATH_BOLD, HEADER_SIZE_3) lines_out = [] for src_line in md.splitlines(): if src_line.strip() == '': lines_out.append(('blank', [])) continue tag = parse_line(src_line) if tag[0] == 'image': try: image = bleh_image_from_url(tag[1]) lines_out.append(('image', image)) except Exception as e: lines_out.append(('text', [('text', f"[Imagen inválida: {e}]")], font, FONT_SIZE)) elif tag[0] == 'userimage': if uploaded_img_bytes: try: image = bleh_image_from_bytes(uploaded_img_bytes) lines_out.append(('image', image)) except Exception as e: lines_out.append(('text', [('text', f"[Error al procesar imagen]")], font, FONT_SIZE)) print(f"Image processing error: {e}") else: lines_out.append(('text', [('text', "[No se subió una imagen]")], font, FONT_SIZE)) elif tag[0] == 'hr': lines_out.append(('hr',)) elif tag[0] == 'header': header_level = tag[1] segments = tag[2] if header_level == 1: font_h = font_h1 size_h = HEADER_SIZE_1 elif header_level == 2: font_h = font_h2 size_h = HEADER_SIZE_2 else: font_h = font_h3 size_h = HEADER_SIZE_3 for wrapped in wrap_segments(segments, font_h, font_h, font_h, font_h, IMAGE_WIDTH): lines_out.append(('header', wrapped, font_h, size_h)) elif tag[0] == 'bullet': segments = tag[1] bullet_font = font bullet_w = bullet_font.getbbox(BULLET_CHAR)[2] - bullet_font.getbbox(BULLET_CHAR)[0] wrapped_lines = list(wrap_segments(segments, font, font_bold, font_italic, font_bolditalic, IMAGE_WIDTH - bullet_w, start_x=0)) for i, wrapped in enumerate(wrapped_lines): if i == 0: lines_out.append(('bullet', wrapped, bullet_font, FONT_SIZE, True, bullet_w)) else: lines_out.append(('bullet', wrapped, bullet_font, FONT_SIZE, False, bullet_w)) elif tag[0] == 'ordered': idx, segments = tag[1], tag[2] num_str = f"{idx}. " number_font = font num_w = number_font.getbbox(num_str)[2] - number_font.getbbox(num_str)[0] wrapped_lines = list(wrap_segments(segments, font, font_bold, font_italic, font_bolditalic, IMAGE_WIDTH - num_w, start_x=0)) for i, wrapped in enumerate(wrapped_lines): if i == 0: lines_out.append(('ordered', wrapped, number_font, FONT_SIZE, num_str, True, num_w)) else: lines_out.append(('ordered', wrapped, number_font, FONT_SIZE, num_str, False, num_w)) else: # normal text segments = tag[1] for wrapped in wrap_segments(segments, font, font_bold, font_italic, font_bolditalic, IMAGE_WIDTH): lines_out.append(('text', wrapped, font, FONT_SIZE)) # Compute total height, including images height = 10 # Top margin for item in lines_out: if item[0] in ('header', 'text', 'bullet', 'ordered'): height += item[3] elif item[0] == 'hr': height += 10 elif item[0] == 'blank': height += FONT_SIZE elif item[0] == 'image': img = item[1] height += img.height + 10 # add margin below image image = Image.new("L", (IMAGE_WIDTH, height), 255) draw = ImageDraw.Draw(image) y = 0 for item in lines_out: if item[0] == 'blank': y += FONT_SIZE elif item[0] == 'hr': draw.line((0, y + 5, IMAGE_WIDTH, y + 5), fill=0, width=2) y += 10 elif item[0] == 'header': segments, fnt, sz = item[1], item[2], item[3] x = 0 for style, text in segments: draw.text((x, y), text, font=fnt, fill=0) x += fnt.getbbox(text)[2] - fnt.getbbox(text)[0] y += sz elif item[0] == 'bullet': segments, bullet_font, sz, show_bullet, bullet_w = item[1], item[2], item[3], item[4], item[5] x = 0 if show_bullet: draw.text((x, y), BULLET_CHAR, font=bullet_font, fill=0) x += bullet_w else: x += bullet_w for style, text in segments: f = font_for_style(style, bullet_font, font_bold, font_italic, font_bolditalic) draw.text((x, y), text, font=f, fill=0) x += f.getbbox(text)[2] - f.getbbox(text)[0] y += sz elif item[0] == 'ordered': segments, number_font, sz, num_str, show_num, num_w = item[1], item[2], item[3], item[4], item[5], item[6] x = 0 if show_num: draw.text((x, y), num_str, font=number_font, fill=0) x += num_w else: x += num_w for style, text in segments: f = font_for_style(style, number_font, font_bold, font_italic, font_bolditalic) draw.text((x, y), text, font=f, fill=0) x += f.getbbox(text)[2] - f.getbbox(text)[0] y += sz elif item[0] == 'text': segments, fnt, sz = item[1], item[2], item[3] x = 0 for style, text in segments: f = font_for_style(style, font, font_bold, font_italic, font_bolditalic) draw.text((x, y), text, font=f, fill=0) x += f.getbbox(text)[2] - f.getbbox(text)[0] y += sz elif item[0] == 'image': img = item[1] # Center image horizontally if narrower img_x = (IMAGE_WIDTH - img.width) // 2 if img.width < IMAGE_WIDTH else 0 image.paste(img, (img_x, y)) y += img.height + 10 # vertical margin after image return image @app.route("/", methods=["GET", "POST"]) def index(): img_data = None md = "" printed = False error = None if request.method == "POST": userimg = request.files.get("userimg") uploaded_img_bytes = None if userimg and userimg.filename: # Remove old temp file if present remove_uploaded_img() # Save new file to a temp location with tempfile.NamedTemporaryFile(delete=False, suffix=".catnote") as tmpf: userimg.save(tmpf) tmpf.flush() session['uploaded_img_path'] = tmpf.name tmpf.seek(0) uploaded_img_bytes = tmpf.read() elif 'uploaded_img_path' in session: path = session['uploaded_img_path'] if os.path.exists(path): with open(path, "rb") as f: uploaded_img_bytes = f.read() else: # If file was deleted or missing, remove from session session.pop('uploaded_img_path', None) else: remove_uploaded_img() md = request.form["md"] image = render(md, uploaded_img_bytes) buf = io.BytesIO() image.save(buf, format="PNG") buf.seek(0) img_data = base64.b64encode(buf.getvalue()).decode() # If print button pressed, send to driver if "print" in request.form: try: with tempfile.NamedTemporaryFile(suffix=".png", delete=True) as tmpfile: image.save(tmpfile, format="PNG") tmpfile.flush() # Run the bleh command # You can set dither method here if desired result = subprocess.run([ "./bleh", "-mode", "1bpp", "-intensity", "100", tmpfile.name ], capture_output=True, text=True, timeout=90) if result.returncode != 0: error = f"Printer error: {result.stderr or result.stdout}" else: printed = True except Exception as e: error = f"Failed to print: {e}" return render_template_string(HTML_FORM, img=img_data, default_md=md, printed=printed, error=error) @app.route('/manifest.json') def manifest(): return send_from_directory('static', 'manifest.json') @app.route('/icon512_maskable.png') def icon_maskable(): return send_from_directory('static', 'icon512_maskable.png') @app.route('/icon512_rounded.png') def icon_rounded(): return send_from_directory('static', 'icon512_rounded.png') if __name__ == "__main__": parser = argparse.ArgumentParser(description='CatNote: Server for Markdown to MXW01 Cat Printer') parser.add_argument('-p', '--port', type=int, default=5000, help='Port to run the server on (default: 5000)') args = parser.parse_args() app.run(host='0.0.0.0', port=args.port)