import numpy as np
import matplotlib.pyplot as plt

segments = ["Enterprise", "SMB", "Startup", "Consumer", "Partner"]
revenue = np.array([320, 180, 90, 70, 150])
margin = np.array([28, 22, 15, 12, 20])
customers = np.array([45, 110, 260, 520, 160])

fig, ax = plt.subplots(figsize=(6, 4))
scatter = ax.scatter(
    revenue,
    margin,
    s=customers,
    c=margin,
    cmap="Blues",
    alpha=0.7,
    edgecolors="#1f2937",
)

for x, y, label in zip(revenue, margin, segments):
    ax.text(x, y, label, fontsize=10, ha="center", va="center")

ax.set_xlabel("売上（百万円）")
ax.set_ylabel("利益率（%）")
ax.set_title("セグメント別 売上・利益率・顧客数")
ax.grid(alpha=0.3)

cbar = fig.colorbar(scatter, ax=ax)
cbar.set_label("利益率（%）")

fig.tight_layout()
fig.savefig("static/images/visualize/scatter/bubble.svg")