Update src/app.py

src/app.py

@@ -90,7 +90,14 @@ async def startup_event():
     os.makedirs(os.path.join(Config.STATIC_DIR, "admin"), exist_ok=True)
     os.makedirs(os.path.join(Config.STATIC_DIR, "admin/js"), exist_ok=True)
     os.makedirs(os.path.join(Config.STATIC_DIR, "admin/css"), exist_ok=True)
+    
+    # 确保图片保存目录存在
     os.makedirs(Config.IMAGE_SAVE_DIR, exist_ok=True)
+    logger.info(f"已确保图片保存目录存在: {Config.IMAGE_SAVE_DIR}")
+    
+    # 确保/tmp目录的api_keys.json父目录存在
+    os.makedirs(os.path.dirname(Config.KEYS_STORAGE_FILE), exist_ok=True)
+    logger.info(f"已确保API密钥存储目录存在: {os.path.dirname(Config.KEYS_STORAGE_FILE)}")
     
     # 输出图片保存目录的信息 
     logger.info(f"图片保存目录: {Config.IMAGE_SAVE_DIR}")
@@ -116,6 +123,13 @@ async def shutdown_event():
 # 挂载静态文件目录
 app.mount("/static", StaticFiles(directory=Config.STATIC_DIR), name="static")
 
+# 挂载图片目录，确保能访问/tmp/images中的图片
+try:
+    app.mount("/images", StaticFiles(directory=Config.IMAGE_SAVE_DIR), name="images")
+    logger.info(f"已挂载图片目录: {Config.IMAGE_SAVE_DIR} -> /images")
+except Exception as e:
+    logger.error(f"挂载图片目录失败: {str(e)}")
+
 # 通用图片访问路由 - 支持多种路径格式
 @app.get("/images/{filename}")
 @app.get("/static/images/{filename}")
@@ -126,8 +140,13 @@ async def get_image(filename: str):
     if os.path.exists(file_path):
         return FileResponse(file_path)
     else:
-        logger.warning(f"请求的图片不存在: {file_path}")
-        raise HTTPException(status_code=404, detail="图片不存在")
+        # 尝试从静态目录获取（向后兼容）
+        alt_path = os.path.join(Config.STATIC_DIR, "images", filename)
+        if os.path.exists(alt_path):
+            return FileResponse(alt_path)
+        else:
+            logger.warning(f"请求的图片不存在: {file_path} 或 {alt_path}")
+            raise HTTPException(status_code=404, detail="图片不存在")
 
 # 添加静态文件路径前缀的兼容路由
 if Config.STATIC_PATH_PREFIX: