renamed AI_Model to AIModel for consistency, added it as a functioning drop-down select on settings page, added face_distance to db and code, put face_distance model_used into all classes ready for use
This commit is contained in:
4
face.py
4
face.py
@@ -2,7 +2,6 @@ from main import db, app, ma
|
||||
from sqlalchemy import Sequence
|
||||
from sqlalchemy.exc import SQLAlchemyError
|
||||
|
||||
|
||||
class Face(db.Model):
|
||||
__tablename__ = "face"
|
||||
id = db.Column(db.Integer, db.Sequence('face_id_seq'), primary_key=True )
|
||||
@@ -17,6 +16,7 @@ class FaceFileLink(db.Model):
|
||||
__tablename__ = "face_file_link"
|
||||
face_id = db.Column(db.Integer, db.ForeignKey("face.id"), primary_key=True )
|
||||
file_eid = db.Column(db.Integer, db.ForeignKey("file.eid"), primary_key=True )
|
||||
model_used = db.Column(db.Integer, db.ForeignKey("ai_model.id"), primary_key=True )
|
||||
|
||||
def __repr__(self):
|
||||
return f"<face_id: {self.face_id}, file_eid={self.file_eid}"
|
||||
@@ -25,6 +25,8 @@ class FaceRefimgLink(db.Model):
|
||||
__tablename__ = "face_refimg_link"
|
||||
face_id = db.Column(db.Integer, db.ForeignKey("face.id"), primary_key=True )
|
||||
refimg_id = db.Column(db.Integer, db.ForeignKey("refimg.id"), primary_key=True )
|
||||
model_used = db.Column(db.Integer, db.ForeignKey("ai_model.id"), primary_key=True )
|
||||
face_distance = db.Column(db.Integer)
|
||||
|
||||
def __repr__(self):
|
||||
return f"<face_id: {self.face_id}, refimg_id={self.refimg_id}"
|
||||
|
||||
Reference in New Issue
Block a user