Update app.py

app.py

@@ -1,42 +1,61 @@
 import os
 import sys
+import subprocess
 from loguru import logger
 
-if __name__ == '__main__':
-    logger.info("准备启动 Infinity-Parser2-Pro 前端服务...")
+def setup_and_run():
+    repo_url = "https://github.com/renkexuan369/Infinity-Parser2-pro.git"
+    repo_dir = "Infinity-Parser2-pro"
 
     # ==========================================
-    # 1. 动态注入环境变量 (覆盖代码中的硬编码)
+    # 1. 动态获取/拉取最新代码
     # ==========================================
-    # 在这里统一管理你们的 API 密钥和地址
-    os.environ['infinity_parser2_pro_base'] = "https://jgcochhge9k5cohjme5jgqeompbh998m.openapi-hw.infly.cn"
-    os.environ['infinity_parser2_pro_Authorization'] = "PpcXBN1L0ND81LdIgOBNYGC8ILFYno+wx7BMzapDaFc="
-    
-    # 禁用 Gradio 的 SSR 模式以提高兼容性
-    os.environ['GRADIO_SSR_MODE'] = "false"
+    if not os.path.exists(repo_dir):
+        logger.info(f"正在克隆仓库: {repo_url}")
+        subprocess.run(["git", "clone", repo_url], check=True)
+    else:
+        logger.info("仓库已存在，拉取最新代码...")
+        subprocess.run(["git", "pull"], cwd=repo_dir, check=True)
 
     # ==========================================
-    # 2. 依赖检查 (可选，确保 pdf2image 所需的 poppler 存在)
+    # 2. 安装 Python 依赖 (如果需要)
     # ==========================================
-    if sys.platform.startswith('linux'):
-        # 你的 inf_gradio.py 内部其实已经写了 apt-get install poppler-utils
-        # 这里可以直接跳过，或者做进一步的权限检查
-        pass
+    # 建议将依赖写在 HF Space 自带的 requirements.txt 中以加速启动
+    # 但如果一定要在此处动态安装，可以取消下方注释：
+    # req_file = os.path.join(repo_dir, "requirements.txt")
+    # if os.path.exists(req_file):
+    #     logger.info("安装仓库依赖...")
+    #     subprocess.run([sys.executable, "-m", "pip", "install", "-r", "requirements.txt"], cwd=repo_dir, check=True)
 
     # ==========================================
-    # 3. 启动 Gradio 应用
+    # 3. 注入安全环境变量
     # ==========================================
-    # 因为 inf_gradio.py 没有使用 argparse 接收参数，
-    # 我们直接通过 python 运行它即可，它内部的 demo.launch() 会接管一切。
+    # 强烈建议将真实密钥配置在 HF Space 的 Settings -> Variables and secrets 中
+    os.environ['infinity_parser2_pro_base'] = os.environ.get(
+        "INFINITY_API_BASE", 
+        "https://jgcochhge9k5cohjme5jgqeompbh998m.openapi-hw.infly.cn"
+    )
+    # 获取 HF Secret 中配置的 Authorization，如果没有则使用默认值（注意：直接写死在代码里有泄露风险）
+    os.environ['infinity_parser2_pro_Authorization'] = os.environ.get(
+        "INFINITY_API_AUTH", 
+        "PpcXBN1L0ND81LdIgOBNYGC8ILFYno+wx7BMzapDaFc="
+    )
+    os.environ['GRADIO_SSR_MODE'] = "false"
+
+    # HF Space 默认暴露 7860 端口
+    os.environ["GRADIO_SERVER_PORT"] = "7860"
+    os.environ["GRADIO_SERVER_NAME"] = "0.0.0.0"
+
+    # ==========================================
+    # 4. 启动 Gradio 服务
+    # ==========================================
+    logger.info("准备启动 inf_gradio.py ...")
+    
+    # 关键：必须设置 cwd=repo_dir，否则 Python 找不到 utils.py 和 prompts.py
     try:
-        logger.info("正在拉起 inf_gradio.py ...")
-        # 使用 os.system 或 subprocess 运行主脚本
-        exit_code = os.system(f"{sys.executable} inf_gradio.py")
-        
-        if exit_code != 0:
-            logger.error(f"服务异常退出，退出码: {exit_code}")
-            
-    except KeyboardInterrupt:
-        logger.info("接收到终止信号，服务已关闭。")
-    except Exception as e:
-        logger.exception(f"启动过程中发生错误: {e}")
+        subprocess.run([sys.executable, "inf_gradio.py"], cwd=repo_dir, check=True)
+    except subprocess.CalledProcessError as e:
+        logger.error(f"Gradio 服务异常退出: {e}")
+
+if __name__ == "__main__":
+    setup_and_run()