JivddlmZddlmZmZmZmZddlZddlm Z ddl m Z GddZ de fd Z e Zy) ) LlamaParse)OptionalListDictAnyN)settings)loggerceZdZdZddeedefdZdedeeee ffdZ d edeeee ffd Z d eedeeee ffd Z d eeee fdefdZ y) PDFParserz A class to parse PDF documents using LlamaParse API. Supports both file paths and URLs for extracting structured text from PDFs. Napi_key result_typec|xstj|_|js tdt |j|d|_t jd|y)z Initialize the PDF parser. Args: api_key: LlamaParse API key. If None, uses settings.LLAMA_CLOUD_API_KEY result_type: Output format - "markdown" or "text" z?LLAMA_CLOUD_API_KEY must be configured in environment variablesT)r r verbosez'PDFParser initialized with result_type=N)rLLAMA_CLOUD_API_KEYr ValueErrorrparserr info)selfr r s D/var/www/html/BankruptcyAI-uat/bankruptcy-ai/src/utils/pdf_parser.py__init__zPDFParser.__init__sX>(">"> ||^_ _ LL#   =k]KLurlreturnc tjd||jj|}tjdt |d|S#t $r=}tj dt|tdt|d}~wt$r=}tj dt|tdt|d}~wt$rZ}t|j}d |vsd |vr8tj d t|td t|d |vsd|vr8tj dt|tdt|d|vsd|vsd|vr8tj dt|tdt|d|vsd|vr8tj dt|tdt|tj dt|tdt|d}~wwxYw)a Parse a PDF from a URL. Args: url: URL of the PDF document Returns: List of parsed documents with text content Raises: Exception: If PDF parsing fails zParsing PDF from URL: Successfully parsed z documents from URLz6LLAMAPARSE FAILURE REASON: Network connection error - zLlamaParse network error: Nz-LLAMAPARSE FAILURE REASON: Request timeout - zLlamaParse timeout: ztcptransport closedzhandler is closedzCLLAMAPARSE FAILURE REASON: HTTP connection pool exhausted/closed - z"LlamaParse connection pool issue: z rate limit429z5LLAMAPARSE FAILURE REASON: API rate limit exceeded - zLlamaParse rate limit: authentication401403z;LLAMAPARSE FAILURE REASON: API key authentication failed - zLlamaParse auth error: zservice unavailable503z>QR S  C LLQRUVWRXQYZ [8QAB B = LLHQQ R23q6(;< < ?A I$ 15HI5U bcfghcibjkl"DSVH MNN*ey.@ TUXYZU[T\]^"9#a& BCC!Y.%92DQZHZ Z[^_`[aZbcd"9#a& BCC&)3u 7I [\_`a\b[cde";CF8 DEE J3q6(ST"4SVH =>>! ?s1AA I"8B I&8C I+EII file_pathc J tjd||jj|}tjdt |d|S#t $r@}tj d|dt|t dt|d}~wwxYw)a  Parse a local PDF file. Args: file_path: Path to the PDF file Returns: List of parsed documents with text content Raises: Exception: If PDF parsing fails zParsing PDF from file: rz documents from filezError parsing PDF file : zError parsing PDF file: N)r rrr!r"r&r$r%)rr-r)r*s r parse_filezPDFParser.parse_fileKs A KK1)= > --i8I KK.s9~.>>RS T  A LL29+RAxH I6s1vh?@ @ AsAA B"";BB"sourcesc g}tjdt|d|D]G} |jj |}|j |tjd|Itjdt||S#t $r.}tjd|dt|Yd}~d}~wwxYw)z Parse multiple PDFs from URLs or file paths. Args: sources: List of URLs or file paths Returns: List of parsed documents from all sources zParsing z PDF sourcesrzError parsing r/NzTotal documents parsed: ) r rr"rr!extendr&r$r%)rr1 all_documentssourcedocsr*s rparse_multiplezPDFParser.parse_multipleas  hs7|nL9: BF B{{,,V4$$T* 26(;<  B  .s=/A.BCD  B ~fXRAx@AA BsAB C $CC r)cd|sydj|Dcgc]}|jc}Scc}w)z Extract plain text from parsed documents. Args: documents: List of parsed documents Returns: Combined text from all documents z )jointext)rr)docs rget_textzPDFParser.get_textys+{{ :CHH:;;:s-)Nmarkdown)__name__ __module__ __qualname____doc__rr%rrrrr,r0r7r=rrr r s M M3M&(?S(?T$sCx.%9(?TACADc3h,@A,d3iDc3h4H0 <$tCH~"6 <3 rLs9",, $#}<}<@ [ r