Update app.py

app.py

@@ -1,119 +1,122 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+PPTX → JPG → ZIP 在线转换（Hugging Face Spaces / 任意 Linux 环境）
+---------------------------------------------------------------
+- 上传 .pptx → 调用 PptxToJpgZip → 返回 _images.zip
+- 兼容含中文文件名、空格路径
+- 启动时自动安装仓库内字体
+"""
+
 import os
-import subprocess
-import gradio as gr
+import sys        # 解决 “name 'sys' is not defined”
 import shutil
+import subprocess
+import tempfile
+import uuid
 from pathlib import Path
 
-# 程序根目录
-BASE_DIR = os.path.dirname(__file__)
-# 可执行文件路径：app.py 与 PptxToJpgZip 同目录
-BINARY_PATH = os.path.join(BASE_DIR, "PptxToJpgZip")
+import gradio as gr
+
+# ============ 基础路径 ============ #
+BASE_DIR = Path(__file__).parent.resolve()
+BINARY_PATH = BASE_DIR / "PptxToJpgZip"     # 可执行文件与 app.py 同目录
 
-# 启动时确保二进制有执行权限（在 Hugging Face Spaces 里无法手动 chmod）
+# ============ 确保可执行权限 ============ #
 try:
     os.chmod(BINARY_PATH, 0o755)
 except Exception as e:
-    # 如果权限设置失败，也无需中断，只要后续能调用即可
-    print(f"⚠️ 警告：设置执行权限失败：{e}")
-
+    print(f"⚠️ 无法修改 {BINARY_PATH} 执行权限：{e}")
 
-def install_fonts_from_repository():
+# ============ 安装字体 ============ #
+def install_fonts_from_repository() -> None:
     """
-    将仓库中的 fonts 目录下的所有字体文件复制到 ~/.fonts 目录，并刷新字体缓存。
+    将 repo/fonts 内的字体复制到 ~/.fonts，并刷新字体缓存
+    （HF Spaces 容器默认存在 fontconfig）
     """
     try:
-        # 获取当前用户的主目录路径
-        home_dir = Path.home()
-
-        # 定义目标字体目录路径：~/.fonts
-        fonts_dir = home_dir / ".fonts"
-
-        # 如果 ~/.fonts 目录不存在，则创建该目录
-        if not fonts_dir.exists():
-            fonts_dir.mkdir(parents=True)
-            print(f"已创建字体目录：{fonts_dir}")
-        else:
-            print(f"字体目录已存在：{fonts_dir}")
+        home_fonts_dir = Path.home() / ".fonts"
+        home_fonts_dir.mkdir(parents=True, exist_ok=True)
 
-        # 定义仓库中的 fonts 目录路径
-        repo_fonts_dir = Path(__file__).parent / "fonts"
-
-        # 检查仓库中的 fonts 目录是否存在
+        repo_fonts_dir = BASE_DIR / "fonts"
         if not repo_fonts_dir.exists():
-            print(f"错误：仓库中的 fonts 目录不存在：{repo_fonts_dir}")
+            print(f"❌ 未找到字体目录：{repo_fonts_dir}")
             return
 
-        # 支持的字体文件扩展名
-        font_extensions = [".ttf", ".otf", ".ttc"]
-
-        # 统计已复制的字体文件数量
-        copied_count = 0
-
-        # 遍历 fonts 目录中的所有文件
+        exts = {".ttf", ".otf", ".ttc"}
+        copied = 0
         for font_file in repo_fonts_dir.iterdir():
-            if font_file.suffix.lower() in font_extensions:
-                destination = fonts_dir / font_file.name
-                try:
-                    shutil.copy(font_file, destination)
-                    print(f"已复制字体文件：{font_file.name} 到 {destination}")
-                    copied_count += 1
-                except Exception as e:
-                    print(f"复制字体文件时出错：{font_file.name}，错误信息：{e}")
-
-        if copied_count == 0:
-            print("未找到任何可复制的字体文件。")
-            return
-
-        # 刷新字体缓存（仅在 Linux/Mac 下）
-        if not sys.platform.startswith('win'):
-            try:
-                print("正在刷新字体缓存...")
+            if font_file.suffix.lower() in exts:
+                dest = home_fonts_dir / font_file.name
+                shutil.copy(font_file, dest)
+                print(f"✅ 已复制字体：{font_file.name}")
+                copied += 1
+
+        if copied:
+            # 刷新缓存（Linux / Mac）
+            if not sys.platform.startswith("win"):
+                print("🔄 刷新字体缓存 …")
                 subprocess.run(["fc-cache", "-f", "-v"], check=True)
-                print("字体缓存刷新完成。")
-            except subprocess.CalledProcessError as e:
-                print(f"刷新字体缓存时出错：{e}")
+                print("✅ 字体缓存刷新完成")
+        else:
+            print("⚠️ 未找到可复制的字体文件")
     except Exception as e:
-        print(f"安装字体时出错：{e}")
-
+        print(f"⚠️ 安装字体失败：{e}")
 
-# 启动时安装字体
 install_fonts_from_repository()
 
+# ============ PPTX → ZIP 主逻辑 ============ #
+def _generate_safe_name() -> str:
+    """生成仅含 ASCII 的随机文件名（避免中文路径问题）"""
+    return f"input_{uuid.uuid4().hex}.pptx"
 
-def convert_pptx_to_zip(pptx_file):
+def convert_pptx_to_zip(pptx_file) -> str:
     """
-    调用外部 PptxToJpgZip，可执行文件将上传的 PPTX 转为 JPG 并打包 ZIP。
-    pptx_file.name 是 Gradio 上传后的临时文件路径。
+    1. 创建临时工作目录 tmpdir
+    2. 将上传的 PPTX 复制为 ASCII 文件名
+    3. 在 tmpdir 中调用 PptxToJpgZip
+    4. 搜索 *_images.zip 并返回其路径
     """
-    src_path = pptx_file.name  # Gradio 会在后台保存上传文件，并提供 .name 属性
+    # -------- 准备工作目录 -------- #
+    tmpdir = Path(tempfile.mkdtemp())
+    safe_pptx = tmpdir / _generate_safe_name()
+    shutil.copy(pptx_file.name, safe_pptx)
 
-    # 调用外部程序：PptxToJpgZip <输入文件.pptx>
+    # -------- 调用外部程序 -------- #
     proc = subprocess.run(
-        [BINARY_PATH, src_path],
+        [str(BINARY_PATH), str(safe_pptx)],
+        cwd=tmpdir,                 # 关键：工作目录设为 tmpdir
         stdout=subprocess.PIPE,
-        stderr=subprocess.PIPE
+        stderr=subprocess.PIPE,
     )
+
     if proc.returncode != 0:
-        stderr = proc.stderr.decode("utf-8", errors="ignore")
-        raise RuntimeError(f"转换失败：\n{stderr}")
+        raise RuntimeError(
+            f"转换失败，PptxToJpgZip 返回码 {proc.returncode}：\n"
+            f"{proc.stderr.decode(errors='ignore')}"
+        )
 
-    # 可执行程序会在同目录生成 <原文件名>_images.zip
-    zip_path = os.path.splitext(src_path)[0] + "_images.zip"
-    if not os.path.isfile(zip_path):
-        raise FileNotFoundError(f"转换完成，但未找到输出 ZIP：{zip_path}")
+    # -------- 查找输出 ZIP -------- #
+    zips = list(tmpdir.glob("*_images.zip"))
+    if not zips:
+        raise FileNotFoundError("转换成功，但未在临时目录找到 *_images.zip 输出文件")
+    if len(zips) > 1:
+        raise RuntimeError("检测到多个 *_images.zip，无法确定返回哪一个")
 
-    return zip_path  # Gradio 会自动为此路径生成下载链接
+    return str(zips[0])            # Gradio 会自动生成下载链接
 
-# —— 构建 Gradio 界面 ——
+# ============ Gradio UI ============ #
 with gr.Blocks(title="PPTX→JPG→ZIP 转换器") as demo:
-    gr.Markdown("""
+    gr.Markdown(
+        """
 # PPTX→JPG→ZIP  
-上传一个 `.pptx` 文件，后台调用 `PptxToJpgZip` 可执行程序，  
-将每页幻灯片导出为高质量 JPG，并打包成 ZIP，最后提供下载。
-""")
+上传一个 `.pptx` 文件，后台调用 **PptxToJpgZip** 可执行程序，  
+将每页幻灯片导出为高质量 JPG，并打包成 ZIP，最后提供下载。"""
+    )
     with gr.Row():
         pptx_in = gr.File(label="上传 PPTX (.pptx)", file_types=[".pptx"])
         btn = gr.Button("开始转换")
+
     zip_out = gr.File(label="下载 ZIP 文件")
 
     btn.click(fn=convert_pptx_to_zip, inputs=pptx_in, outputs=zip_out)
@@ -122,5 +125,6 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False
+        share=False,
+        # Spaces 如果需要关掉 SSR，可加 ssr_mode=False
     )