diff --git a/app.py b/app.py index af99e98..dd40853 100644 --- a/app.py +++ b/app.py @@ -1682,13 +1682,17 @@ async def hairline_grow_v2_final_v2( @app.get("/api/v1/debug/hairline_log", include_in_schema=False) async def download_hairline_log(rid: Optional[str] = None, tail: int = 500): - """返回 /home/xsl/hair/log/hairline_grow.log 的内容。 + """返回 <仓库根>/log/hairline_grow.log 的内容。 rid 非空时只返回该 request id 相关的行;tail 限制返回最后 N 行(默认 500)。 供调试页"下载日志"按钮调用。 """ from fastapi.responses import PlainTextResponse - log_path = "/home/xsl/hair/log/hairline_grow.log" + log_path = os.getenv( + "HAIR_LOG_DIR", + os.path.join(os.path.dirname(os.path.abspath(__file__)), "log"), + ) + log_path = os.path.join(log_path, "hairline_grow.log") try: with open(log_path, encoding="utf-8") as fh: lines = fh.readlines() diff --git a/face_analysis/hairline_grow.py b/face_analysis/hairline_grow.py index 4b39c29..5cfc5f1 100644 --- a/face_analysis/hairline_grow.py +++ b/face_analysis/hairline_grow.py @@ -47,8 +47,11 @@ from face_analysis.head_mask import ( _draw_baseline, ) -# 调试日志:写 /home/xsl/hair/log/hairline_grow.log,每个步骤详细记录 -_LOG_DIR = "/home/xsl/hair/log" +# 调试日志:写 <仓库根>/log/hairline_grow.log,每个步骤详细记录(可用 HAIR_LOG_DIR 覆盖) +_LOG_DIR = os.getenv( + "HAIR_LOG_DIR", + os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "log"), +) os.makedirs(_LOG_DIR, exist_ok=True) logger = logging.getLogger("hairline_grow") _log_fh = logging.FileHandler(os.path.join(_LOG_DIR, "hairline_grow.log"), encoding="utf-8")