Create utils_viz.py

utils_viz.py

@@ -0,0 +1,17 @@
+# utils_viz.py
+import io
+import matplotlib.pyplot as plt
+
+def bar_topk(top_rows):
+    labels = [r["name"] for r in top_rows]
+    vals = [r["score"] for r in top_rows]
+    fig, ax = plt.subplots(figsize=(7,3))
+    ax.barh(labels[::-1], vals[::-1])
+    ax.set_xlabel("Score")
+    ax.set_xlim(0,100)
+    fig.tight_layout()
+    bio = io.BytesIO()
+    fig.savefig(bio, format="png", dpi=160, bbox_inches="tight")
+    plt.close(fig)
+    bio.seek(0)
+    return bio