Update main.py

main.py

@@ -5,7 +5,16 @@ import json
 import base64
 from io import BytesIO
 from datetime import datetime
+import logging
 
+# 配置日志记录器
+logging.basicConfig(
+    level=logging.INFO,  # 设置日志级别为INFO
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",  # 定义日志格式
+)
+
+# 获取 FastAPI 应用日志记录器
+logger = logging.getLogger("fastapi")
 app = FastAPI()
 
 
@@ -25,13 +34,13 @@ def read_root():
 
 @app.post("/upload")
 async def upload_file(file: UploadFile = File(...)):
-    print(f'收到请求 {file.filename}')
+    logger.info(f'收到请求 {file.filename}')
     # 定义文件存储路径和名称
     file_location = os.path.join(UPLOAD_DIRECTORY, file.filename)
 
     # 将上传的文件写入本地文件系统
     with open(file_location, "wb") as buffer:
-        print(f'写入本地文件 {file.filename}')
+        logger.info(f'写入本地文件 {file.filename}')
         buffer.write(await file.read())
 
     return {"info": f"file '{file.filename}' saved at '{file_location}'"}