添加了后台运行的功能以及密钥设置

.gitignore

@@ -0,0 +1,2 @@
+
+.env

Dockerfile

@@ -1,3 +1,10 @@
+
+
+ARG USERNAME
+ARG PASSWORD
+ARG OPENAI_API_KEY
+
+
 FROM python:3.12
 
 WORKDIR /app
@@ -22,4 +29,18 @@ USER appuser
 ENV PATH="/home/appuser/.local/bin:${PATH}"
 RUN pip install --user pdf2zh
 
-CMD ["pdf2zh", "-i"]
+WORKDIR /app
+COPY . .
+
+USER root
+RUN mkdir -p /home/appuser/.cache/babeldoc/models/ && \
+    chmod -R 777 /home/appuser/.cache/ && \
+    mv models/doclayout_yolo_docstructbench_imgsz1024.onnx /home/appuser/.cache/babeldoc/models/
+
+# ENV HF_ENDPOINT=https://hf-mirror.com
+# ENV HF_HUB_OFFLINE=1
+
+USER appuser
+EXPOSE 7860
+
+CMD ["/bin/bash", "start.sh"]

README.md

@@ -1,9 +1,12 @@
 ---
-title: PDFMathTranslate Demo
+title: PDFMathTranslate
+emoji: 📚
+colorFrom: blue
+colorTo: pink
 sdk: docker
 pinned: false
-license: agpl-3.0
-short_description: 'Demo for https://github.com/Byaidu/PDFMathTranslate'
+license: MIT
+short_description: https://github.com/Byaidu/PDFMathTranslate
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

awake.py

@@ -0,0 +1,81 @@
+import time
+import random
+import math
+from itertools import cycle
+
+def cpu_intensive_task():
+    """随机选择一个CPU密集型任务执行"""
+    tasks = [
+        _calculate_primes,
+        _matrix_multiplication,
+        _fibonacci_calculation,
+        _pi_calculation
+    ]
+    task = random.choice(tasks)
+    task()
+
+def _calculate_primes():
+    """计算质数"""
+    n = random.randint(100000, 1000000)
+    sieve = [True] * (n + 1)
+    sieve[0:2] = [False, False]
+    for i in range(2, int(math.sqrt(n)) + 1):
+        if sieve[i]:
+            sieve[i*i : n+1 : i] = [False] * len(sieve[i*i : n+1 : i])
+
+def _matrix_multiplication():
+    """矩阵乘法"""
+    size = random.randint(100, 300)
+    matrix = [[random.random() for _ in range(size)] for _ in range(size)]
+    result = [[0] * size for _ in range(size)]
+    for i in range(size):
+        for j in range(size):
+            for k in range(size):
+                result[i][j] += matrix[i][k] * matrix[k][j]
+
+def _fibonacci_calculation():
+    """斐波那契数列计算"""
+    n = random.randint(300000, 500000)
+    a, b = 0, 1
+    for _ in range(n):
+        a, b = b, a + b
+
+def _pi_calculation():
+    """蒙特卡洛法计算π近似值"""
+    iterations = 10000000
+    count = 0
+    for _ in range(iterations):
+        x = random.random()
+        y = random.random()
+        if x*x + y*y <= 1:
+            count += 1
+    pi = 4 * count / iterations
+
+def main():
+    try:
+        # 初始随机延迟 (5~15 分钟)
+        initial_delay = random.randint(5, 15)
+        time.sleep(initial_delay * 60)
+        
+        # 创建间隔时间循环（10-300秒）
+        intervals = cycle([random.randint(10, 300) for _ in range(50)])
+        
+        while True:
+            start_time = time.time()
+            print(f"开始CPU密集型任务 @ {time.strftime('%Y-%m-%d %H:%M:%S')}")
+            
+            cpu_intensive_task()
+            
+            elapsed = time.time() - start_time
+            print(f"任务完成，耗时: {elapsed:.2f}秒")
+            
+            # 动态调整间隔时间
+            delay = next(intervals)
+            print(f"下次任务将在 {delay} 分钟后执行...")
+            time.sleep(delay * 60)
+            
+    except KeyboardInterrupt:
+        print("\n任务调度已停止")
+
+if __name__ == "__main__":
+    main()

build.sh

@@ -0,0 +1,4 @@
+
+docker build -t pdf-math-translator .
+docker run --rm -t -p 7860:7860 --name pdf --env-file .env pdf-math-translator
+# docker exec -it pdf /bin/bash

init_config.py

@@ -0,0 +1,27 @@
+
+import os
+import json
+
+OPENAI_API_KEY: str = os.environ.get("OPENAI_API_KEY", "")
+
+if not OPENAI_API_KEY:
+    print("You Should Set OPENAI_API_KEY Enviroment.")
+    exit(-1)
+
+config = {
+    "USE_MODELSCOPE": "0",
+    "translators": [
+        {
+            "name": "openai",
+            "envs": {
+                # 设置环境变量
+                "OPENAI_BASE_URL": "https://megatrump-oneapi.hf.space/v1",
+                "OPENAI_API_KEY": OPENAI_API_KEY,
+                "OPENAI_MODEL": "gpt-4o-mini",
+            }
+        }
+    ]
+}
+
+with open("config.json", "w") as f:
+    json.dump(config, f)

start.sh

@@ -0,0 +1,12 @@
+
+# 根据环境变量初始化用户名以及密码
+echo "${USERNAME},${PASSWORD}" > users.txt
+
+# 启动一个后台任务避免 Space 进入 Sleep 状态
+python3 awake.py &
+
+# 初始化环境变量
+python3 init_config.py
+
+# 启动 Server
+pdf2zh -i --config config.json --serverport 7860 -t 5 --authorized users.txt