| import numpy as np | |
| from PIL import Image | |
| import io | |
| import base64 | |
| def generate_image(text): | |
| # Generate a random image | |
| image = np.random.randint(0, 256, size=(1028, 1028, 3), dtype=np.uint8) | |
| image = Image.fromarray(image) | |
| # Save the image to a byte buffer | |
| buffer = io.BytesIO() | |
| image.save(buffer, format="JPEG") | |
| image_bytes = buffer.getvalue() | |
| # Convert the image bytes to a base64-encoded string | |
| image_string = base64.b64encode(image_bytes).decode("utf-8") | |
| return image_string | |
| # # Assume that `image_string` is the base64-encoded image string returned by `generate_image` | |
| # image_bytes = base64.b64decode(generate_image("")) | |
| # image = Image.open(io.BytesIO(image_bytes)) | |
| # image.show() # Show the image in a new window | |