diff --git a/app/routes/documents.py b/app/routes/documents.py
index e3b83d3..efa60ed 100644
--- a/app/routes/documents.py
+++ b/app/routes/documents.py
@@ -1,7 +1,7 @@
from pathlib import Path
from uuid import uuid4
-from fastapi import APIRouter, Depends, Request
+from fastapi import APIRouter, Depends, Form, Request
from fastapi.responses import HTMLResponse, RedirectResponse
from fastapi.templating import Jinja2Templates
from sqlalchemy.orm import Session, selectinload
@@ -80,6 +80,44 @@ def test_ingest(db: Session = Depends(get_db)):
return RedirectResponse(url=f"/documents/{document.document_id}", status_code=303)
+@router.post("/{document_id}/review-text", response_class=RedirectResponse)
+def save_reviewed_text(
+ document_id: str,
+ reviewed_text: str = Form(...),
+ db: Session = Depends(get_db),
+):
+ document = (
+ db.query(Document)
+ .options(selectinload(Document.text_versions))
+ .filter(Document.document_id == document_id)
+ .first()
+ )
+
+ if document is None:
+ return RedirectResponse(url="/documents/", status_code=303)
+
+ existing_reviewed = [
+ tv for tv in document.text_versions if tv.version_type == "reviewed" and tv.is_current
+ ]
+ for tv in existing_reviewed:
+ tv.is_current = False
+
+ reviewed_version = TextVersion(
+ document_id=document.id,
+ version_type="reviewed",
+ text_content=reviewed_text,
+ created_by="mcelwain",
+ is_current=True,
+ )
+ db.add(reviewed_version)
+
+ document.review_status = "reviewed"
+
+ db.commit()
+
+ return RedirectResponse(url=f"/documents/{document.document_id}", status_code=303)
+
+
@router.get("/{document_id}", response_class=HTMLResponse)
def document_detail(document_id: str, request: Request, db: Session = Depends(get_db)):
document = (
@@ -97,15 +135,28 @@ def document_detail(document_id: str, request: Request, db: Session = Depends(ge
if document is None:
return HTMLResponse(content="Document not found", status_code=404)
+ sorted_text_versions = sorted(
+ document.text_versions,
+ key=lambda x: x.created_at,
+ reverse=True,
+ )
+
raw_ocr = next(
- (
- tv
- for tv in sorted(document.text_versions, key=lambda x: x.created_at, reverse=True)
- if tv.version_type == "raw_ocr"
- ),
+ (tv for tv in sorted_text_versions if tv.version_type == "raw_ocr"),
None,
)
+ reviewed_ocr = next(
+ (tv for tv in sorted_text_versions if tv.version_type == "reviewed" and tv.is_current),
+ None,
+ )
+
+ review_text_value = (
+ reviewed_ocr.text_content
+ if reviewed_ocr is not None
+ else raw_ocr.text_content if raw_ocr is not None else ""
+ )
+
return templates.TemplateResponse(
request=request,
name="documents/detail.html",
@@ -113,5 +164,7 @@ def document_detail(document_id: str, request: Request, db: Session = Depends(ge
"request": request,
"document": document,
"raw_ocr": raw_ocr,
+ "reviewed_ocr": reviewed_ocr,
+ "review_text_value": review_text_value,
},
)
diff --git a/app/templates/documents/detail.html b/app/templates/documents/detail.html
index ec56947..0f66f7d 100644
--- a/app/templates/documents/detail.html
+++ b/app/templates/documents/detail.html
@@ -5,7 +5,7 @@
{{ document.document_id }}
- Back to documents
+ Back to documents
{{ document.document_id }}
@@ -48,5 +48,24 @@
{% else %}
No raw OCR text found.
{% endif %}
+
+ Reviewed OCR
+ {% if reviewed_ocr %}
+ Current reviewed version saved at {{ reviewed_ocr.created_at }}
+ {% else %}
+ No reviewed OCR saved yet.
+ {% endif %}
+
+