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

backend/repositories/template_repository.py

@@ -0,0 +1,75 @@
+from sqlalchemy.orm import Session
+from sqlalchemy import desc
+from models.template_models import Template, TemplateField, TemplateFieldMapping, TemplateRecognitionHistory
+from schemas.template_schemas import TemplateCreate, TemplateFieldCreate, TemplateFieldMappingCreate
+
+class TemplateRepository:
+    def __init__(self, db: Session):
+        self.db = db
+
+    def get_template(self, template_id: int):
+        return self.db.query(Template).filter(Template.pk_template_id == template_id).first()
+
+    def get_templates(self, skip: int = 0, limit: int = 100):
+        return self.db.query(Template).offset(skip).limit(limit).all()
+
+    def create_template(self, template_data: TemplateCreate):
+        db_template = Template(template_name=template_data.template_name)
+        self.db.add(db_template)
+        self.db.commit()
+        self.db.refresh(db_template)
+        
+        # Create fields if provided
+        for field in template_data.fields:
+            db_field = TemplateField(
+                fk_template_id=db_template.pk_template_id,
+                field_label=field.field_label,
+                field_type=field.field_type,
+                display_order=field.display_order,
+                required_flag=field.required_flag
+            )
+            self.db.add(db_field)
+            
+        self.db.commit()
+        self.db.refresh(db_template)
+        return db_template
+
+    def add_template_field(self, template_id: int, field_data: TemplateFieldCreate):
+        db_field = TemplateField(
+            fk_template_id=template_id,
+            field_label=field_data.field_label,
+            field_type=field_data.field_type,
+            display_order=field_data.display_order,
+            required_flag=field_data.required_flag
+        )
+        self.db.add(db_field)
+        self.db.commit()
+        self.db.refresh(db_field)
+        return db_field
+
+    def delete_template(self, template_id: int):
+        template = self.get_template(template_id)
+        if template:
+            self.db.delete(template)
+            self.db.commit()
+            return True
+        return False
+
+    def save_mapping(self, template_id: int, mappings: list[TemplateFieldMappingCreate]):
+        saved_mappings = []
+        for mapping in mappings:
+            db_mapping = TemplateFieldMapping(
+                fk_template_id=template_id,
+                fk_template_field_id=mapping.fk_template_field_id,
+                fk_document_data_id=mapping.fk_document_data_id,
+                page_no=mapping.page_no,
+                x_coordinate=mapping.x_coordinate,
+                y_coordinate=mapping.y_coordinate,
+                width=mapping.width,
+                height=mapping.height,
+                mapping_confidence=mapping.mapping_confidence
+            )
+            self.db.add(db_mapping)
+            saved_mappings.append(db_mapping)
+        self.db.commit()
+        return saved_mappings