| import tokenize |
| import io |
| import base64 |
| import numpy as np |
| from PIL import Image |
|
|
| def remove_inline_comments_safe(code: str) -> str: |
| result = [] |
| tokens = tokenize.generate_tokens(io.StringIO(code).readline) |
|
|
| last_line = -1 |
| current_line = '' |
| for tok_type, tok_string, (srow, scol), (_, _), _ in tokens: |
| if srow != last_line: |
| if current_line: |
| result.append(current_line.rstrip()) |
| current_line = '' |
| last_line = srow |
|
|
| if tok_type == tokenize.COMMENT: |
| |
| continue |
|
|
| current_line += tok_string |
|
|
| if current_line: |
| result.append(current_line.rstrip()) |
|
|
| return '\n'.join(result) |
|
|
|
|
| def image_to_jpg_base64_url(image: Image.Image | np.ndarray) -> str: |
| """Return a base64 *JPEG* data‑URL from a PIL image or NumPy array.""" |
| if isinstance(image, np.ndarray): |
| image = Image.fromarray(image) |
| if image.mode in {"RGBA", "LA"}: |
| image = image.convert("RGB") |
| with io.BytesIO() as buffer: |
| image.save(buffer, format="JPEG") |
| encoded: str = base64.b64encode(buffer.getvalue()).decode() |
| return f"data:image/jpeg;base64,{encoded}" |
|
|
