Spaces:
Sleeping
Sleeping
File size: 1,061 Bytes
46db43c |
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 |
import os
from typing import List, Dict, Any
from fastapi import HTTPException
from utils.ingestion.parser import parse_pdf_bytes, parse_docx_bytes
# ────────────────────────────── Helpers ──────────────────────────────
def _infer_mime(filename: str) -> str:
lower = filename.lower()
if lower.endswith(".pdf"):
return "application/pdf"
if lower.endswith(".docx"):
return "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
return "application/octet-stream"
def _extract_pages(filename: str, file_bytes: bytes) -> List[Dict[str, Any]]:
mime = _infer_mime(filename)
if mime == "application/pdf":
return parse_pdf_bytes(file_bytes)
elif mime == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
return parse_docx_bytes(file_bytes)
else:
raise HTTPException(status_code=400, detail=f"Unsupported file type: {filename}")
|