Update app.py

app.py

@@ -1,130 +1,70 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
-"""
-PPTX → JPG → ZIP 在线转换（Hugging Face Spaces / 任意 Linux 环境）
----------------------------------------------------------------
-- 上传 .pptx → 调用 PptxToJpgZip → 返回 _images.zip
-- 兼容含中文文件名、空格路径
-- 启动时自动安装仓库内字体
-"""
-
-import os
-import sys        # 解决 “name 'sys' is not defined”
-import shutil
-import subprocess
-import tempfile
-import uuid
+import os, sys, shutil, subprocess, tempfile, uuid
 from pathlib import Path
-
 import gradio as gr
 
-# ============ 基础路径 ============ #
 BASE_DIR = Path(__file__).parent.resolve()
-BINARY_PATH = BASE_DIR / "PptxToJpgZip"     # 可执行文件与 app.py 同目录
-
-# ============ 确保可执行权限 ============ #
-try:
-    os.chmod(BINARY_PATH, 0o755)
-except Exception as e:
-    print(f"⚠️ 无法修改 {BINARY_PATH} 执行权限：{e}")
-
-# ============ 安装字体 ============ #
-def install_fonts_from_repository() -> None:
-    """
-    将 repo/fonts 内的字体复制到 ~/.fonts，并刷新字体缓存
-    （HF Spaces 容器默认存在 fontconfig）
-    """
-    try:
-        home_fonts_dir = Path.home() / ".fonts"
-        home_fonts_dir.mkdir(parents=True, exist_ok=True)
-
-        repo_fonts_dir = BASE_DIR / "fonts"
-        if not repo_fonts_dir.exists():
-            print(f"❌ 未找到字体目录：{repo_fonts_dir}")
-            return
-
-        exts = {".ttf", ".otf", ".ttc"}
-        copied = 0
-        for font_file in repo_fonts_dir.iterdir():
-            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("✅ 字体缓存刷新完成")
-        else:
-            print("⚠️ 未找到可复制的字体文件")
-    except Exception as e:
-        print(f"⚠️ 安装字体失败：{e}")
+BINARY_PATH = BASE_DIR / "PptxToJpgZip"
+os.chmod(BINARY_PATH, 0o755)  # 忽略异常
 
-install_fonts_from_repository()
+# -------- 省略 install_fonts_from_repository()，保持之前版本 -------- #
 
-# ============ PPTX → ZIP 主逻辑 ============ #
-def _generate_safe_name() -> str:
-    """生成仅含 ASCII 的随机文件名（避免中文路径问题）"""
+def _safe_name() -> str:
+    """生成仅 ASCII 的临时文件名"""
     return f"input_{uuid.uuid4().hex}.pptx"
 
 def convert_pptx_to_zip(pptx_file) -> str:
-    """
-    1. 创建临时工作目录 tmpdir
-    2. 将上传的 PPTX 复制为 ASCII 文件名
-    3. 在 tmpdir 中调用 PptxToJpgZip
-    4. 搜索 *_images.zip 并返回其路径
-    """
-    # -------- 准备工作目录 -------- #
+    """调用 PptxToJpgZip，把 PPTX 转成图片 ZIP；若仅有文件夹则自动 zip。"""
     tmpdir = Path(tempfile.mkdtemp())
-    safe_pptx = tmpdir / _generate_safe_name()
-    shutil.copy(pptx_file.name, safe_pptx)
+    src = tmpdir / _safe_name()
+    shutil.copy(pptx_file.name, src)
 
-    # -------- 调用外部程序 -------- #
+    # 调用外部可执行文件
     proc = subprocess.run(
-        [str(BINARY_PATH), str(safe_pptx)],
-        cwd=tmpdir,                 # 关键：工作目录设为 tmpdir
+        [str(BINARY_PATH), str(src)],
+        cwd=tmpdir,
         stdout=subprocess.PIPE,
         stderr=subprocess.PIPE,
     )
-
     if proc.returncode != 0:
         raise RuntimeError(
-            f"转换失败，PptxToJpgZip 返回码 {proc.returncode}：\n"
-            f"{proc.stderr.decode(errors='ignore')}"
+            f"PptxToJpgZip 失败，退出码 {proc.returncode}\n"
+            f"stderr:\n{proc.stderr.decode(errors='ignore')}"
         )
 
-    # -------- 查找输出 ZIP -------- #
+    # ---- ① 优先找 *_images.zip ----
     zips = list(tmpdir.glob("*_images.zip"))
-    if not zips:
-        raise FileNotFoundError("转换成功，但未在临时目录找到 *_images.zip 输出文件")
-    if len(zips) > 1:
-        raise RuntimeError("检测到多个 *_images.zip，无法确定返回哪一个")
-
-    return str(zips[0])            # Gradio 会自动生成下载链接
+    if zips:
+        return str(zips[0])
+
+    # ---- ② 若无 ZIP，再找 *_images 目录，自动打包 ----
+    img_dirs = [p for p in tmpdir.glob("*_images") if p.is_dir()]
+    if img_dirs:
+        folder = img_dirs[0]
+        zip_path = folder.with_suffix(".zip")      # <name>_images.zip
+        shutil.make_archive(zip_path.with_suffix(""), "zip", folder)
+        return str(zip_path)
+
+    # ---- ③ 仍没找到：列出 tmpdir 内容方便调试 ----
+    contents = "\n".join(["  • " + p.name for p in tmpdir.iterdir()])
+    raise FileNotFoundError(
+        "未找到输出 ZIP，也未发现 *_images 文件夹。\n"
+        f"临时目录内容：\n{contents}"
+    )
 
-# ============ Gradio UI ============ #
+# ----------------- Gradio UI 保持不变 ----------------- #
 with gr.Blocks(title="PPTX→JPG→ZIP 转换器") as demo:
     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)
 
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        # Spaces 如果需要关掉 SSR，可加 ssr_mode=False
-    )
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)