|
1580 | 1580 | "import base64\n",
|
1581 | 1581 | "import io\n",
|
1582 | 1582 | "\n",
|
| 1583 | + "\n", |
1583 | 1584 | "def encode_image_to_base64(image):\n",
|
1584 | 1585 | " with io.BytesIO() as output_bytes:\n",
|
1585 | 1586 | " if isinstance(image, dict):\n",
|
1586 |
| - " image = image['image']\n", |
1587 |
| - " format = \"PNG\" if image.mode == 'RGBA' else \"JPEG\"\n", |
| 1587 | + " image = image[\"image\"]\n", |
| 1588 | + " format = \"PNG\" if image.mode == \"RGBA\" else \"JPEG\"\n", |
1588 | 1589 | " image.save(output_bytes, format=format)\n",
|
1589 | 1590 | " bytes_data = output_bytes.getvalue()\n",
|
1590 | 1591 | "\n",
|
1591 | 1592 | " encoded_string = base64.b64encode(bytes_data)\n",
|
1592 | 1593 | "\n",
|
1593 | 1594 | " base64_str = str(encoded_string, \"utf-8\")\n",
|
1594 |
| - " mimetype = \"image/jpeg\" if format == 'JPEG' else 'image/png'\n", |
| 1595 | + " mimetype = \"image/jpeg\" if format == \"JPEG\" else \"image/png\"\n", |
1595 | 1596 | " image_encoded_in_base64 = (\n",
|
1596 | 1597 | " \"data:\" + (mimetype if mimetype is not None else \"\") + \";base64,\" + base64_str\n",
|
1597 | 1598 | " )\n",
|
1598 | 1599 | " return image_encoded_in_base64\n",
|
1599 | 1600 | "\n",
|
| 1601 | + "\n", |
1600 | 1602 | "def decode_base64_to_image(encoding):\n",
|
1601 | 1603 | " if encoding.startswith(\"data:image/\"):\n",
|
1602 | 1604 | " encoding = encoding.split(\";\")[1].split(\",\")[1]\n",
|
|
0 commit comments