Update sankey_plot.py

sankey_plot.py

@@ -46,9 +46,25 @@ def plot_sankey_from_df(sankey_df: pd.DataFrame, title="问题 → 关键词共
 
     fig.update_layout(title_text=title, font=dict(family="Microsoft YaHei", size=18), width=900, height=600,
                       margin=dict(l=50, r=50, t=80, b=50),autosize=False )   # 关闭自动缩放
+    print(f"✅ 桑基图{fig}生成成功")
+    
     # === 5. 导出高清 PNG（关键！）===
-    img_bytes = fig.to_image( format="png", width=900, height=600, scale=2, engine="kaleido")  # 2倍 DPI → 超清！
-    # return base64.b64encode(img_bytes).decode()
-    b64 = base64.b64encode(img_bytes).decode("utf-8")
-    print(f"✅ 桑基图base64生成成功，长度：{len(b64)} 字符")
+    # 6. 关键：生成base64（增加异常捕获，确保生成成功）
+    try:
+        # 生成高清PNG字节流（scale=2 → 2倍DPI，避免模糊）
+        img_bytes = fig.to_image(
+            format="png",
+            width=900,
+            height=600,
+            scale=2,
+            engine="kaleido"  # 显式指定引擎，避免依赖自动检测
+        )
+        # 转base64字符串
+        b64 = base64.b64encode(img_bytes).decode("utf-8")
+        print(f"✅ 桑基图base64生成成功，长度：{len(b64)} 字符")  # 正常长度约10万+字符
+    except Exception as img_err:
+        # 图片生成失败时，用空图表的base64兜底
+        print(f"⚠️  桑基图转base64失败：{str(img_err)}")
+        empty_img_bytes = fig.to_image(format="png", width=900, height=600, scale=1)
+        b64 = base64.b64encode(empty_img_bytes).decode("utf-8")
     return fig, b64