Text Extraction Done, UI Fixes Done, Extracted Template Layout Saved in DB

2026-07-12 14:24:17 +05:30
parent 3163bb213e
commit 3d6614f408
79 changed files with 3139 additions and 81 deletions
--- a/backend/models/template_models.py
+++ b/backend/models/template_models.py
@@ -0,0 +1,105 @@
+from sqlalchemy import Column, Integer, String, Text, DateTime, Boolean, ForeignKey, Numeric
+from sqlalchemy.sql import func
+from sqlalchemy.orm import relationship
+from database import Base
+
+class TemplateDocument(Base):
+    __tablename__ = "documents"
+    __table_args__ = {"schema": "templates"}
+    
+    pk_document_id = Column(Integer, primary_key=True, index=True)
+    document_name = Column(String(500))
+    document_type = Column(String(100))
+    file_name = Column(String(500))
+    file_hash = Column(String(500), index=True)
+    page_count = Column(Integer)
+    status = Column(String(50))
+    created_at = Column(DateTime, server_default=func.now())
+    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())
+    
+    # Relationships
+    layouts = relationship("DocumentLayout", back_populates="document", cascade="all, delete")
+
+class DocumentLayout(Base):
+    __tablename__ = "document_layout"
+    __table_args__ = {"schema": "templates"}
+    
+    pk_document_data_id = Column(Integer, primary_key=True, index=True)
+    fk_document_id = Column(Integer, ForeignKey("templates.documents.pk_document_id", ondelete="CASCADE"), nullable=False, index=True)
+    page_no = Column(Integer, nullable=False, index=True)
+    text_value = Column(Text)
+    block_type = Column(String(100))
+    parent_block_id = Column(Integer, ForeignKey("templates.document_layout.pk_document_data_id", ondelete="SET NULL"))
+    x_coordinate = Column(Numeric)
+    y_coordinate = Column(Numeric)
+    width = Column(Numeric)
+    height = Column(Numeric)
+    confidence = Column(Numeric)
+    sequence_no = Column(Integer)
+    layout_path = Column(Text)
+    created_at = Column(DateTime, server_default=func.now())
+    
+    document = relationship("TemplateDocument", back_populates="layouts")
+    parent = relationship("DocumentLayout", remote_side=[pk_document_data_id])
+    mappings = relationship("TemplateFieldMapping", back_populates="document_layout", cascade="all, delete")
+
+class Template(Base):
+    __tablename__ = "templates"
+    __table_args__ = {"schema": "templates"}
+    
+    pk_template_id = Column(Integer, primary_key=True, index=True)
+    template_name = Column(String(255), nullable=False, index=True)
+    template_fingerprint = Column(Text)
+    active_flag = Column(Boolean, default=True)
+    created_at = Column(DateTime, server_default=func.now())
+    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())
+    
+    fields = relationship("TemplateField", back_populates="template", cascade="all, delete")
+    mappings = relationship("TemplateFieldMapping", back_populates="template", cascade="all, delete")
+
+class TemplateField(Base):
+    __tablename__ = "template_fields"
+    __table_args__ = {"schema": "templates"}
+    
+    pk_template_field_id = Column(Integer, primary_key=True, index=True)
+    fk_template_id = Column(Integer, ForeignKey("templates.templates.pk_template_id", ondelete="CASCADE"), nullable=False, index=True)
+    field_label = Column(String(255), nullable=False)
+    field_type = Column(String(100), nullable=False)
+    display_order = Column(Integer)
+    required_flag = Column(Boolean, default=False)
+    created_at = Column(DateTime, server_default=func.now())
+    
+    template = relationship("Template", back_populates="fields")
+    mappings = relationship("TemplateFieldMapping", back_populates="field", cascade="all, delete")
+
+class TemplateFieldMapping(Base):
+    __tablename__ = "template_fields_mapping"
+    __table_args__ = {"schema": "templates"}
+    
+    pk_mapping_id = Column(Integer, primary_key=True, index=True)
+    fk_template_id = Column(Integer, ForeignKey("templates.templates.pk_template_id", ondelete="CASCADE"), nullable=False, index=True)
+    fk_template_field_id = Column(Integer, ForeignKey("templates.template_fields.pk_template_field_id", ondelete="CASCADE"), nullable=False, index=True)
+    fk_document_data_id = Column(Integer, ForeignKey("templates.document_layout.pk_document_data_id", ondelete="SET NULL"))
+    page_no = Column(Integer)
+    x_coordinate = Column(Numeric)
+    y_coordinate = Column(Numeric)
+    width = Column(Numeric)
+    height = Column(Numeric)
+    mapping_confidence = Column(Numeric)
+    layout_path = Column(Text)
+    created_at = Column(DateTime, server_default=func.now())
+    
+    template = relationship("Template", back_populates="mappings")
+    field = relationship("TemplateField", back_populates="mappings")
+    document_layout = relationship("DocumentLayout", back_populates="mappings")
+
+class TemplateRecognitionHistory(Base):
+    __tablename__ = "template_recognition_history"
+    __table_args__ = {"schema": "templates"}
+    
+    pk_history_id = Column(Integer, primary_key=True, index=True)
+    fk_template_id = Column(Integer, ForeignKey("templates.templates.pk_template_id", ondelete="CASCADE"), nullable=False)
+    fk_document_id = Column(Integer, ForeignKey("templates.documents.pk_document_id", ondelete="CASCADE"), nullable=False, index=True)
+    recognition_score = Column(Numeric)
+    matched_flag = Column(Boolean)
+    created_at = Column(DateTime, server_default=func.now())