import os
import subprocess
import gradio as gr
import shutil

# 程序根目录
BASE_DIR = os.path.dirname(__file__)
# 可执行文件路径：app.py 与 PptxToJpgZip 同目录
BINARY_PATH = os.path.join(BASE_DIR, "PptxToJpgZip")

# 启动时确保二进制有执行权限（在 Hugging Face Spaces 里无法手动 chmod）
try:
    os.chmod(BINARY_PATH, 0o755)
except Exception as e:
    # 如果权限设置失败，也无需中断，只要后续能调用即可
    print(f"⚠️ 警告：设置执行权限失败：{e}")

# 1. 定位源目录：仓库根目录（当前工作目录）
src_dir = os.getcwd()

# 2. 目标目录：用户字体目录（可写）
dst_dir = os.path.expanduser("~/.local/share/fonts")
os.makedirs(dst_dir, exist_ok=True)

# 3. 复制根目录下的所有 .ttf 和 .otf 文件
for fname in os.listdir(src_dir):
    if fname.lower().endswith((".ttf", ".otf")):
        shutil.copy(
            os.path.join(src_dir, fname),
            os.path.join(dst_dir, fname)
        )

# 4. 刷新指定目录的字体缓存，确保新字体可用
subprocess.run(
    ["fc-cache", "-f", "-v", dst_dir],
    check=True
)

def convert_pptx_to_zip(pptx_file):
    """
    调用外部 PptxToJpgZip，可执行文件将上传的 PPTX 转为 JPG 并打包 ZIP。
    pptx_file.name 是 Gradio 上传后的临时文件路径。
    """
    src_path = pptx_file.name  # Gradio 会在后台保存上传文件，并提供 .name 属性

    # 调用外部程序：PptxToJpgZip <输入文件.pptx>
    proc = subprocess.run(
        [BINARY_PATH, src_path],
        stdout=subprocess.PIPE,
        stderr=subprocess.PIPE
    )
    if proc.returncode != 0:
        stderr = proc.stderr.decode("utf-8", errors="ignore")
        raise RuntimeError(f"转换失败：\n{stderr}")

    # 可执行程序会在同目录生成 <原文件名>_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}")

    return zip_path  # Gradio 会自动为此路径生成下载链接

# —— 构建 Gradio 界面 ——
with gr.Blocks(title="PPTX→JPG→ZIP 转换器") as demo:
    gr.Markdown("""
# PPTX→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
    )