Update app.py

app.py

@@ -1,154 +1,141 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 """
-SVG → PNG 在线批量转换（Gradio + ImageMagick 版）
+SVG → PNG 在线批量转换（Inkscape 版）
 作者: ChatGPT 示例
-环境: Python 3.9+ / Gradio 5.x / ImageMagick 6 或 7
+运行环境: Python 3.9+ / Gradio 5.x / Inkscape ≥1.0
 """
 
 import os
 import shutil
 import subprocess
-import sys
 import tempfile
 import zipfile
 from pathlib import Path
 from typing import List, Tuple
 
-import gradio as gr  # pip install gradio>=5.0
-
-# ----------------------------------------------------------------------
-# 全局配置
-# ----------------------------------------------------------------------
-# 默认 DPI（决定 PNG 清晰度），与 Inkscape/浏览器相同：96
-DEFAULT_DPI = 300
-# ImageMagick 可执行文件名字 (Magick v7 用 "magick"，v6 用 "convert")
-IMAGEMAGICK_CMD_CANDIDATES = ["magick", "convert"]
-
-# ----------------------------------------------------------------------
-# 辅助函数
-# ----------------------------------------------------------------------
-def find_imagemagick() -> str:
-    """
-    检测系统中的 ImageMagick 可执行文件名称，并返回绝对路径或命令名。
-    若未找到则抛出异常。
-    """
-    for cmd in IMAGEMAGICK_CMD_CANDIDATES:
-        if shutil.which(cmd):
-            return cmd
-    raise RuntimeError(
-        "未检测到 ImageMagick。请确认已安装并将可执行文件加入 PATH。"
+import gradio as gr
+import cairosvg  # 纯 Python 兜底渲染
+
+# --------------------------- 全局配置 ---------------------------
+DEFAULT_DPI = 300  # 默认导出分辨率 (DPI)
+
+# --------------------------- 工具函数 ---------------------------
+def find_inkscape() -> str:
+    """在 PATH 中查找 inkscape 可执行文件；找不到则抛异常。"""
+    exe = shutil.which("inkscape")
+    if exe:
+        return exe
+    raise FileNotFoundError(
+        "未检测到 Inkscape，请确认 packages.txt 已包含 `inkscape`，"
+        "并在 Spaces 构建日志中检查安装是否成功。"
     )
 
 
-IM_CMD = find_imagemagick()
+INKSCAPE = None  # 运行期检测并缓存路径
 
 
-def svg_to_png(svg_path: Path, out_path: Path, dpi: int = DEFAULT_DPI) -> None:
+def svg_to_png(svg_path: Path, out_path: Path, dpi: int = DEFAULT_DPI):
     """
-    调用 ImageMagick 将单个 SVG 转换为 PNG。
-    参数:
-        svg_path:  SVG 文件路径
-        out_path:  目标 PNG 路径
-        dpi:       渲染密度 (DPI)，数值越大 PNG 越清晰，体积也越大
+    以 Inkscape CLI 为主，将 SVG 转成 PNG。
+    - 优先使用新 CLI(--export-type)；失败后自动尝试旧 CLI(--export-png)。
+    - 若两者均失败，则回退到 CairoSVG。
     """
-    # ImageMagick 标准命令: magick -background none -density <dpi> input.svg PNG32:output.png
-    command = [
-        IM_CMD,
-        "-background",
-        "none",        # 保留透明度&#8203;:contentReference[oaicite:6]{index=6}
-        "-density",
-        str(dpi),      # 设置渲染 DPI
+    global INKSCAPE
+    if INKSCAPE is None:  # 首次调用时初始化
+        INKSCAPE = find_inkscape()
+
+    # Inkscape ≥1.0 推荐写法
+    cmd_new = [
+        INKSCAPE,
         str(svg_path),
-        "PNG32:" + str(out_path),  # PNG32 强制包含 alpha 通道
+        "--export-type=png",
+        f"--export-filename={out_path}",
+        f"--export-dpi={dpi}",
     ]
 
-    # Windows 上若使用 magick，需要子命令 convert；直接传 PNG32:output 同样有效
     try:
-        subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    except subprocess.CalledProcessError as e:
-        raise RuntimeError(
-            f"转换失败：{svg_path.name}\n\n"
-            f"ImageMagick 输出：\n{e.stderr.decode('utf-8', errors='ignore')}"
-        ) from e
-
-
-def batch_convert(
-    files: List[gr.File], dpi: int = DEFAULT_DPI
-) -> Tuple[List[Tuple[str, str]], str]:
-    """
-    Gradio 回调函数：批量转换上传的 SVG。
-    返回:
-        1. [(PNG文件名, PNG路径)] 供 Gallery 展示
-        2. ZIP 文件路径 供用户下载
-    """
+        subprocess.run(cmd_new, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    except subprocess.CalledProcessError as e_new:
+        # 兼容极旧版 (<1.0) 参数：--export-png
+        cmd_old = [
+            INKSCAPE,
+            str(svg_path),
+            f"--export-png={out_path}",
+            f"-d={dpi}",
+        ]
+        try:
+            subprocess.run(cmd_old, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        except subprocess.CalledProcessError as e_old:
+            # Inkscape 全部失败 ➜ 回退 CairoSVG
+            try:
+                cairosvg.svg2png(url=str(svg_path), write_to=str(out_path), dpi=dpi)
+            except Exception as cs_err:
+                # 抛出详细错误，方便前端提示
+                raise RuntimeError(
+                    f"Inkscape 新旧命令均失败：\n{e_new.stderr.decode(errors='ignore')}\n"
+                    f"{e_old.stderr.decode(errors='ignore')}\n"
+                    f"CairoSVG 也失败：{cs_err}"
+                ) from cs_err
+
+
+def batch_convert(files: List[gr.File], dpi: int = DEFAULT_DPI) -> Tuple[List[Tuple[str, str]], str]:
+    """Gradio 回调：批量转换并打包下载。"""
     if not files:
-        raise gr.Error("请先上传至少一个 SVG 文件。")
+        raise gr.Error("请先上传至少一个 SVG 文件！")
 
     tmp_dir = Path(tempfile.mkdtemp(prefix="svg2png_"))
     png_dir = tmp_dir / "png"
     png_dir.mkdir(exist_ok=True)
 
     gallery_items = []
-    for file_obj in files:
-        svg_path = Path(file_obj.name)
-        out_path = png_dir / (svg_path.stem + ".png")
+    for f in files:
+        svg_path = Path(f.name)
+        out_path = png_dir / f"{svg_path.stem}.png"
         svg_to_png(svg_path, out_path, dpi)
-        # Gallery 接收 (display_name, file_path)
+        # Gallery 需要 (标题, 路径)
         gallery_items.append((out_path.name, str(out_path)))
 
-    # 将所有 PNG 打包成 zip
+    # 打包 ZIP
     zip_path = tmp_dir / "result.zip"
     with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
-        for png_item in png_dir.iterdir():
-            zf.write(png_item, arcname=png_item.name)
+        for p in png_dir.iterdir():
+            zf.write(p, arcname=p.name)
 
     return gallery_items, str(zip_path)
 
-
-# ----------------------------------------------------------------------
-# Gradio UI
-# ----------------------------------------------------------------------
-with gr.Blocks(title="SVG 转 PNG 在线工具", theme="soft") as demo:
+# --------------------------- Gradio UI ---------------------------
+with gr.Blocks(title="SVG → PNG (Inkscape 版)", theme="soft") as demo:
     gr.Markdown(
         """
-# 🌟 SVG → PNG 转换器  
-- **支持批量上传**，透明背景，DPI 可调  
-- 使用 **ImageMagick** 后端渲染，品质可靠  
-- 点击「开始转换」后，可在下方画廊预览，也可整体打包下载  
+# 🌈 SVG → PNG 在线转换器（Inkscape 引擎）  
+- **批量上传** + **DPI 可调** + **透明背景**  
+- 首选 Inkscape CLI，高质量矢量光栅化  
+- 若 Inkscape 不可用自动回退 CairoSVG  
         """
     )
 
     with gr.Row():
         uploader = gr.File(
-            label="上传 SVG（可多选）",
-            file_count="multiple",
-            file_types=[".svg"],
+            label="上传 SVG（可多选）", file_count="multiple", file_types=[".svg"]
         )
         dpi_slider = gr.Slider(
-            minimum=72,
-            maximum=600,
-            step=1,
-            value=DEFAULT_DPI,
-            label="输出 DPI（分辨率）",
-            info="数值越大越清晰，文件体积越大",
+            72, 600, step=1, value=DEFAULT_DPI,
+            label="输出 DPI", info="数值越大越清晰，文件体积也越大"
         )
+
     convert_btn = gr.Button("🚀 开始转换")
-    out_gallery = gr.Gallery(label="PNG 结果预览", show_label=True, height="auto")
-    download_zip = gr.File(label="下载全部 PNG (ZIP)")
+    gallery = gr.Gallery(label="PNG 预览", show_label=True, height="auto")
+    zip_file = gr.File(label="下载全部 PNG (ZIP)")
 
     convert_btn.click(
-        fn=batch_convert,
+        batch_convert,
         inputs=[uploader, dpi_slider],
-        outputs=[out_gallery, download_zip],
+        outputs=[gallery, zip_file],
         api_name="convert",
         queue=True,
     )
 
-# ----------------------------------------------------------------------
-# 入口
-# ----------------------------------------------------------------------
 if __name__ == "__main__":
-    # 在本地启动: python svg2png_app.py
-    # 部署到 Hugging Face Spaces / Docker 时，可根据需要修改 host/port
+    # 在 HF Spaces 中启动时会忽略 share 参数；本地调试可设 share=True
     demo.launch(share=False)