feat(api): Shifted gears and added logging 🪵 and CORS! What a ride!

- Added logging functionality to the `/ocr` and `/` endpoints in `app.py`. Now we can keep an eye on who's doing what with our awesome API! 👀
- Integrated CORS middleware in `app.py` to allow cross-origin requests. Let's break down those barriers and spread the OCR joy! 🌍 🎉

app.py

@@ -1,12 +1,28 @@
 import base64
+import logging
 import os
+from datetime import datetime
 
 import torch
-from fastapi import FastAPI, File, Form, UploadFile
+from fastapi import FastAPI, File, Form, Request, UploadFile
+from fastapi.middleware.cors import CORSMiddleware
 from transformers import AutoModel, AutoTokenizer
 
+# 配置日志
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
+
 app = FastAPI()
 
+# 添加CORS中间件
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # 初始化模型
 model_name = "ucaslcl/GOT-OCR2_0"
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -44,7 +60,23 @@ async def ocr_process(image_path, got_mode, ocr_color="", ocr_box=""):
 
 
 @app.post("/ocr")
-async def ocr_api(image: UploadFile = File(...), got_mode: str = Form(...), ocr_color: str = Form(""), ocr_box: str = Form("")):
+async def ocr_api(request: Request, image: UploadFile = File(...), got_mode: str = Form(...), ocr_color: str = Form(""), ocr_box: str = Form("")):
+    # 记录请求信息
+    client_host = request.client.host
+    user_agent = request.headers.get("user-agent", "Unknown")
+    current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+    log_message = f"""
+    时间: {current_time}
+    IP地址: {client_host}
+    User-Agent: {user_agent}
+    图片名称: {image.filename}
+    OCR模式: {got_mode}
+    OCR颜色: {ocr_color}
+    OCR边界框: {ocr_box}
+    """
+    logger.info(log_message)
+
     # 保存上传的图片
     image_path = f"temp_{image.filename}"
     with open(image_path, "wb") as buffer:
@@ -56,9 +88,24 @@ async def ocr_api(image: UploadFile = File(...), got_mode: str = Form(...), ocr_
     # 删除临时文件
     os.remove(image_path)
 
+    # 记录处理结果
+    logger.info(f"OCR处理结果: {result}")
+
     return result
 
 
 @app.get("/")
-def read_root():
+async def read_root(request: Request):
+    client_host = request.client.host
+    user_agent = request.headers.get("user-agent", "Unknown")
+    current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+    log_message = f"""
+    时间: {current_time}
+    IP地址: {client_host}
+    User-Agent: {user_agent}
+    访问: 根路径
+    """
+    logger.info(log_message)
+
     return {"message": "欢迎使用OCR API"}