Spaces:
Sleeping
Sleeping
File size: 3,430 Bytes
52a0fe9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | """
DOCX text extraction using python-docx.
Extracts text preserving paragraph structure, tables, and document properties.
"""
import time
import os
from docx import Document
from models.schemas import ExtractionResult, DocumentMetadata
def extract_docx(file_path: str) -> ExtractionResult:
"""Extract text and metadata from a DOCX file."""
start_time = time.time()
try:
doc = Document(file_path)
# Extract paragraphs
paragraphs = []
for para in doc.paragraphs:
text = para.text.strip()
if text:
# Preserve heading structure
if para.style and para.style.name.startswith("Heading"):
level = para.style.name.replace("Heading ", "").strip()
prefix = "#" * int(level) if level.isdigit() else "##"
paragraphs.append(f"{prefix} {text}")
else:
paragraphs.append(text)
# Extract tables
tables_text = []
for table_idx, table in enumerate(doc.tables):
table_data = []
for row in table.rows:
row_data = [cell.text.strip() for cell in row.cells]
table_data.append(" | ".join(row_data))
if table_data:
tables_text.append(f"\n[Table {table_idx + 1}]\n" + "\n".join(table_data))
# Combine all text
full_text = "\n\n".join(paragraphs)
if tables_text:
full_text += "\n\n" + "\n".join(tables_text)
# Extract metadata from core properties
props = doc.core_properties
metadata = DocumentMetadata(
title=props.title or os.path.basename(file_path),
author=props.author or "Unknown",
creation_date=str(props.created) if props.created else "",
modification_date=str(props.modified) if props.modified else "",
page_count=None, # DOCX doesn't expose page count easily
word_count=len(full_text.split()) if full_text else 0,
character_count=len(full_text),
file_type="DOCX",
extra={
"category": props.category or "",
"comments": props.comments or "",
"last_modified_by": props.last_modified_by or "",
"revision": props.revision,
"subject": props.subject or "",
"keywords": props.keywords or "",
"paragraph_count": len(doc.paragraphs),
"table_count": len(doc.tables),
}
)
elapsed = (time.time() - start_time) * 1000
if not full_text.strip():
return ExtractionResult(
raw_text="",
metadata=metadata,
success=False,
error_message="No text content found in the DOCX file.",
extraction_time_ms=elapsed,
)
return ExtractionResult(
raw_text=full_text,
metadata=metadata,
success=True,
extraction_time_ms=elapsed,
)
except Exception as e:
elapsed = (time.time() - start_time) * 1000
return ExtractionResult(
raw_text="",
metadata=DocumentMetadata(file_type="DOCX"),
success=False,
error_message=f"DOCX extraction failed: {str(e)}",
extraction_time_ms=elapsed,
)
|