Update main.py

main.py

@@ -1,33 +1,35 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
 """
 FastAPI 版 SVG ➜ PNG 转换服务
 ──────────────────────────────────────────
-与 Gradio 版本保持同等功能：
+功能：
 1. 自动检测 / 定位 Inkscape
-2. 自动下载并嵌入 <image href="http/https"> 外链图片（Base64）
+2. 下载并嵌入 <image href="http/https"> 外链图片（Base64，支持中文 URL）
 3. 支持 DPI / 宽高 / 背景色 / 透明 / 导出区域 / 指定元素 ID / 简化 SVG
 4. 返回生成的 PNG 图片文件
-──────────────────────────────────────────
 运行：
     python main.py
 接口：
-    POST /convert   ➜ Multipart 表单上传 svg_file + 各参数，返回 image/png
-    GET  /health    ➜ 简单健康检查
+    POST /convert   ➜ Multipart 上传 svg_file + 参数，返回 image/png
+    GET  /health    ➜ 健康检查
 """
-
 import os
-import subprocess
 import platform
-import tempfile
+import re
 import shutil
+import subprocess
+import tempfile
 import uuid
+from pathlib import Path
+from typing import Optional
+
 import base64
 import mimetypes
 import urllib.request
-import re
-from pathlib import Path
-from typing import Optional
+import urllib.parse
 
-from fastapi import FastAPI, UploadFile, File, Form, HTTPException
+from fastapi import FastAPI, File, Form, HTTPException, UploadFile
 from fastapi.responses import FileResponse, JSONResponse
 
 ###############################################################################
@@ -48,10 +50,10 @@ def ensure_inkscape_available() -> tuple[bool, str]:
         common_paths: list[str] = []
         if system == "Windows":
             common_paths = [
-                r"C:\Program Files\Inkscape\bin\inkscape.exe",
-                r"C:\Program Files (x86)\Inkscape\bin\inkscape.exe",
                 r"C:\Program Files\Inkscape\inkscape.exe",
                 r"C:\Program Files (x86)\Inkscape\inkscape.exe",
+                r"C:\Program Files\Inkscape\bin\inkscape.exe",
+                r"C:\Program Files (x86)\Inkscape\bin\inkscape.exe",
             ]
         elif system == "Linux":
             common_paths = [
@@ -64,44 +66,58 @@ def ensure_inkscape_available() -> tuple[bool, str]:
                 "/Applications/Inkscape.app/Contents/MacOS/inkscape",
                 "/usr/local/bin/inkscape",
             ]
-
-        for path in common_paths:
-            if os.path.exists(path):
-                os.environ["PATH"] += os.pathsep + os.path.dirname(path)
-                return True, f"在 {path} 找到 Inkscape"
-
+        for p in common_paths:
+            if os.path.exists(p):
+                os.environ["PATH"] += os.pathsep + os.path.dirname(p)
+                return True, f"在 {p} 找到 Inkscape"
         return False, "未找到 Inkscape，请安装或手动加入 PATH"
 
 ###############################################################################
-# 2. 下载并嵌入外链图片
+# 2. URL 规范化（解决中文 / 空格 / () 等字符）
+###############################################################################
+def normalize_url(url: str) -> str:
+    """
+    将包含非 ASCII 字符的 URL 转为合法百分号编码格式
+    """
+    parsed = urllib.parse.urlparse(url)
+    # 域名 IDNA（如有中文）
+    netloc = parsed.netloc.encode("idna").decode("ascii")
+    # 路径 / 查询 / 片段 百分号转义
+    path     = urllib.parse.quote(parsed.path, safe="/")
+    params   = urllib.parse.quote(parsed.params, safe=":&=")
+    query    = urllib.parse.quote_plus(parsed.query, safe="=&")
+    fragment = urllib.parse.quote(parsed.fragment, safe="")
+    return urllib.parse.urlunparse((parsed.scheme, netloc, path, params, query, fragment))
+
+###############################################################################
+# 3. 下载并嵌入外链图片
 ###############################################################################
-# 匹配 <image ... href="http(s)://..."> 或 xlink:href
 IMG_TAG_PATTERN = re.compile(
-    r'<image[^>]+(?:href|xlink:href)\s*=\s*["\']([^"\']+)["\']', re.I
+    r'<image[^>]+(?:href|xlink:href)\s*=\s*["\']([^"\']+)["\']',
+    re.I,
 )
 
 def download_and_encode(url: str) -> str:
     """下载远程图片并转为 data URI（Base64）"""
     try:
-        req = urllib.request.Request(url, headers={"User-Agent": "Mozilla/5.0"})
-        with urllib.request.urlopen(req, timeout=15) as resp:
+        safe_url = normalize_url(url)  # ⭐ 关键改动
+        req = urllib.request.Request(safe_url, headers={"User-Agent": "Mozilla/5.0"})
+        with urllib.request.urlopen(req, timeout=20) as resp:
             data = resp.read()
-        mime, _ = mimetypes.guess_type(url)
+        mime, _ = mimetypes.guess_type(safe_url)
         mime = mime or "image/png"
         b64 = base64.b64encode(data).decode("ascii")
         return f"data:{mime};base64,{b64}"
     except Exception as e:
-        # 失败则返回原链接（Inkscape 可能渲染空白 / 红叉）
         print(f"[WARN] 下载外链图片失败 {url}: {e}")
-        return url
+        return url  # 失败则返回原链接
 
 def embed_external_images(svg_path: str) -> str:
     """把 SVG 中的远程图片下载并嵌入（Base64），返回新的临时 SVG 路径"""
     with open(svg_path, "r", encoding="utf-8") as f:
         svg_text = f.read()
-
     if "http://" not in svg_text and "https://" not in svg_text:
-        return svg_path  # 无外链，直接返回
+        return svg_path  # 无外链直接返回
 
     def replacer(match: re.Match) -> str:
         url = match.group(1)
@@ -125,7 +141,7 @@ def embed_external_images(svg_path: str) -> str:
     return new_path
 
 ###############################################################################
-# 3. 核心转换函数
+# 4. 核心转换函数
 ###############################################################################
 def convert_svg_to_png(
     src_svg_path: str,
@@ -138,12 +154,10 @@ def convert_svg_to_png(
     export_plain_svg: bool = False,
 ) -> tuple[Optional[str], str]:
     """调用 Inkscape CLI 将 SVG 转为 PNG，返回 (文件路径 or None, 消息)"""
-
     ok, msg = ensure_inkscape_available()
     if not ok:
         return None, msg
 
-    # 预处理外链
     processed_svg = embed_external_images(src_svg_path)
 
     tmp_dir = tempfile.mkdtemp()
@@ -164,29 +178,20 @@ def convert_svg_to_png(
         cmd += ["--export-height", str(height)]
     if background_color != "transparent":
         cmd += ["--export-background", background_color]
-
     if export_area == "drawing":
         cmd.append("--export-area-drawing")
     elif export_area == "page":
         cmd.append("--export-area-page")
     elif export_area == "id" and export_id:
         cmd += ["--export-id", export_id]
-
     if export_plain_svg:
         cmd.append("--export-plain-svg")
 
     try:
-        subprocess.run(
-            cmd,
-            stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE,
-            text=True,
-            check=True,
-        )
+        subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
         if not os.path.exists(output_path):
             return None, "未生成 PNG 文件"
 
-        # 把结果移动到项目根目录 output 目录
         out_dir = Path(__file__).parent / "output"
         out_dir.mkdir(exist_ok=True)
         final_path = out_dir / output_name
@@ -199,13 +204,13 @@ def convert_svg_to_png(
         return None, f"未知错误：{e}"
 
 ###############################################################################
-# 4. FastAPI 应用
+# 5. FastAPI 应用
 ###############################################################################
-app = FastAPI(title="SVG ➜ PNG 转换 API", version="1.0.0")
+app = FastAPI(title="SVG ➜ PNG 转换 API", version="1.1.0")
 
 @app.get("/health", summary="健康检查")
 def health():
-    return {"status": "ok"}
+    return JSONResponse({"status": "ok"})
 
 @app.post(
     "/convert",
@@ -229,9 +234,8 @@ async def convert_endpoint(
 ):
     """
     使用 Inkscape 将上传的 SVG 转换为 PNG 并返回。
-    所有参数与 Gradio 版本保持一致。
     """
-    # 保存上传文件到临时目录
+    # 保存上传文件
     temp_dir = tempfile.mkdtemp()
     uploaded_path = os.path.join(temp_dir, svg_file.filename)
     with open(uploaded_path, "wb") as f:
@@ -247,13 +251,11 @@ async def convert_endpoint(
         export_id=export_id if export_area == "id" else None,
         export_plain_svg=export_plain_svg,
     )
-
     shutil.rmtree(temp_dir, ignore_errors=True)
 
     if png_path is None:
         raise HTTPException(status_code=400, detail=msg)
 
-    # 返回 PNG 文件流
     return FileResponse(
         png_path,
         media_type="image/png",
@@ -261,12 +263,10 @@ async def convert_endpoint(
     )
 
 ###############################################################################
-# 5. 入口
+# 6. 入口
 ###############################################################################
 if __name__ == "__main__":
     import uvicorn
 
-    # 确保 output 目录存在
     (Path(__file__).parent / "output").mkdir(exist_ok=True)
-
     uvicorn.run("main:app", host="0.0.0.0", port=7860, reload=False)