init: 完成了图片加水印

backend-bridge/.gitignore

@@ -0,0 +1,2 @@
+.idea/
+.venv/

backend-bridge/README.md

@@ -0,0 +1,3 @@
+# 简介
+这个后端用来直接桥接TrustMark，不做过多的请求参数校验，仅仅监听localhost，端口为8081
+前端不能直接请求这个服务，应当请求 ../backend 项目下的后端服务

backend-bridge/main.py

@@ -0,0 +1,64 @@
+from fastapi import FastAPI, UploadFile, File, Form, HTTPException, APIRouter
+from fastapi.responses import Response
+from starlette.responses import JSONResponse
+from trustmark import TrustMark
+from PIL import Image
+import io
+
+tm = TrustMark(model_type="Q", verbose=False)
+appTmp = FastAPI(title="TrustMark Encode API", version="1.0")
+app = APIRouter(prefix="/watermark")
+
+@app.post("/encode")
+async def encode_watermark(
+    image: UploadFile = File(...),
+    payload: str = Form(...),
+    outputType: str = Form(...),
+    strength: float = Form(1.2)
+):
+    """
+    给图片嵌入TrustMark盲水印
+    :param image: 原图文件上传
+    :param payload: 最多8个ASCII字符
+    :param outputType: 输出类型，必须是png，jpg，jpeg
+    :param strength: 取值一般0.0~1.5
+    :return: 带水印图片二进制
+    """
+    img_bytes = await image.read()
+    img = Image.open(io.BytesIO(img_bytes)).convert("RGB")
+    watermarked_img = tm.encode(img, payload, WM_STRENGTH=strength)
+    buf = io.BytesIO()
+    watermarked_img.save(buf, format=outputType, quality=90)
+    buf.seek(0)
+    return Response(
+        content=buf.read(),
+        media_type=f"image/{outputType}"
+    )
+
+@app.post("/decode", summary="上传图片提取TrustMark盲水印")
+async def extract_watermark(image: UploadFile = File(...)):
+    MAX_FILE_SIZE = 10 * 1024 * 1024
+    if image.size and image.size > MAX_FILE_SIZE:
+        raise HTTPException(status_code=413, detail="文件过大，最大支持10MB")
+    allowed_types = ["image/png", "image/jpeg", "image/jpg"]
+    if image.content_type not in allowed_types:
+        raise HTTPException(status_code=400, detail="仅支持 png / jpg 图片")
+
+    try:
+        img_bytes = await image.read()
+        img = Image.open(io.BytesIO(img_bytes)).convert("RGB")
+        payload, has_watermark, schema = tm.decode(img)
+
+        return JSONResponse({
+            "success": True,
+            "has_watermark": has_watermark,
+            "payload": payload if has_watermark else None,
+            "schema": schema if has_watermark else None
+        })
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"解析失败: {str(e)}")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run("main:app", host="localhost", port=8081, workers=1)

backend-bridge/requirements.txt

@@ -0,0 +1,8 @@
+fastapi==0.112.2
+uvicorn[standard]==0.30.1
+python-multipart==0.0.20
+pillow==10.4.0
+trustmark==0.9.2
+
+starlette
+starlette