| import matplotlib.pyplot as plt | |
| import numpy as np | |
| def string_to_image(text): | |
| text = text.replace("_", " ").lower().replace(", ", "\n") | |
| # Create a blank white square image | |
| img = np.ones((220, 75, 3)) | |
| fig, ax = plt.subplots(figsize=(6, 2.25)) | |
| ax.imshow(img, extent=[0, 1, 0, 1]) | |
| ax.text(0.5, 0.75, text, fontsize=18, ha="center", va="center") | |
| ax.set_xticks([]) | |
| ax.set_yticks([]) | |
| ax.set_xticklabels([]) | |
| ax.set_yticklabels([]) | |
| for spine in ax.spines.values(): | |
| spine.set_visible(False) | |
| return fig | |