Update cluster_insight.py

cluster_insight.py

@@ -147,6 +147,12 @@ def auto_select_k(embeddings, max_k=10):
     return best_k
 
 
+def fig_to_base64(fig):
+    buffer = BytesIO()
+    fig.write_image(buffer, format="png", engine="kaleido")  # 此时在Hugging Face环境，kaleido能找到字体
+    buffer.seek(0)
+    return base64.b64encode(buffer.read()).decode("ascii")
+
 def cluster_and_visualize(
         excel_path: str,
         questions=['s1', 's2', 's3', 's4'],
@@ -259,7 +265,8 @@ def cluster_and_visualize(
         height=600,
         scale=2
         )
-    b64 = base64.b64encode(img_bytes).decode('utf-8')
+    # b64 = base64.b64encode(img_bytes).decode('utf-8')
+    b64 = fig_to_base64(fig)
     # print(f"{b64}解析成功！")
     # return b64, stats
     return fig, b64, stats