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:
@@ -227,6 +227,14 @@ class Refimg(Base):
|
||||
def __repr__(self):
|
||||
return f"<id: {self.id}, fname: {self.fname}, created_on: {self.created_on}>"
|
||||
|
||||
class AIModel(Base):
|
||||
__tablename__ = "ai_model"
|
||||
id = Column(Integer, primary_key=True )
|
||||
name = Column(String)
|
||||
|
||||
def __repr__(self):
|
||||
return f"<id: {self.id}, name: {self.name}>"
|
||||
|
||||
class Face(Base):
|
||||
__tablename__ = "face"
|
||||
id = Column(Integer, Sequence('face_id_seq'), primary_key=True )
|
||||
@@ -240,6 +248,7 @@ class FaceFileLink(Base):
|
||||
__tablename__ = "face_file_link"
|
||||
face_id = Column(Integer, ForeignKey("face.id"), primary_key=True )
|
||||
file_eid = Column(Integer, ForeignKey("file.eid"), primary_key=True )
|
||||
model_used = Column(Integer, ForeignKey("ai_model.id") )
|
||||
|
||||
def __repr__(self):
|
||||
return f"<face_id: {self.face_id}, file_eid={self.file_eid}"
|
||||
@@ -248,6 +257,8 @@ class FaceRefimgLink(Base):
|
||||
__tablename__ = "face_refimg_link"
|
||||
face_id = Column(Integer, ForeignKey("face.id"), primary_key=True )
|
||||
refimg_id = Column(Integer, ForeignKey("refimg.id"), primary_key=True )
|
||||
model_used = Column(Integer, ForeignKey("ai_model.id") )
|
||||
face_distance = Column(Integer)
|
||||
|
||||
def __repr__(self):
|
||||
return f"<face_id: {self.face_id}, refimg_id={self.refimg_id}"
|
||||
|
||||
Reference in New Issue
Block a user