真人素材库本地完成

video-gen-api/app/models/user.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from sqlalchemy import Boolean, DateTime, Float, String, JSON
+from sqlalchemy import Boolean, DateTime, Float, Integer, String, JSON
 from sqlalchemy.orm import Mapped, mapped_column
 
 from app.enums.user import FrontendUserKind
@@ -39,6 +39,11 @@ class User(Base, TimestampMixin):
     )
     allowed_menus: Mapped[list | None] = mapped_column(JSON, nullable=True)
 
+    # 真人素材库图片总量限制。0 表示关闭真人素材模块；>0 表示启用并限制用户所有真人素材图片总量。
+    private_portrait_image_limit: Mapped[int] = mapped_column(
+        Integer, default=5, server_default="5", nullable=False
+    )
+
     @property
     def must_set_password(self) -> bool:
         return self.user_type == "frontend" and not self.hashed_password