Ng3ddlZddlZddlmZmZmZmZmZddlZddl m Z m Z m Z m Z dedee e ffdZdeeeefdee e ffd Zdded ede fd Zdd ejd ede fd ZdS)N)ListUnionDictrAny)BaseLayoutElement TextBlockLayoutBASECOORD_ELEMENT_NAMEMAPfilenamereturnct|d5}tj|}dddn #1swxYwYt|S)aTLoad a JSON file and save it as a layout object with appropriate data types. Args: filename (str): The name of the JSON file. Returns: Union[BaseLayoutElement, Layout]: Based on the JSON file format, it will automatically parse the type of the data and load it accordingly. rN)openjsonload load_dict)r fpress Q/var/www/html/ai-engine/env/lib/python3.11/site-packages/layoutparser/io/basic.py load_jsonrs h  imm S>>s 266datacttrdvr/ttdjdSdt vrt ddtfdtj D}|rtj St d SttrtdDSt d) aLoad a dict of list of dict representations of some layout data, automatically parse its type, and save it as any of BaseLayoutElement or Layout datatype. Args: data (Union[Dict, List]): A dict of list of dict representations of the layout data Raises: ValueError: If the data format is incompatible with the layout-data-JSON format, raise a `ValueError`. ValueError: If any `block_type` name is not in the available list of layout element names defined in `BASECOORD_ELEMENT_NAMEMAP`, raise a `ValueError`. Returns: Union[BaseLayoutElement, Layout]: Based on the dict format, it will automatically parse the type of the data and load it accordingly. page_datablocks)r block_typezInvalid block_type c3 K|]}|vV dSN).0elers r zload_dict..Os'JJssd{JJJJJJc,g|]}t|Sr)r)r r!s r zload_dict..Vs666#y~~666r#zInvalid input JSON structure.) isinstancedictr r_blocksr ValueErroranyr _features from_dictlist)r is_textblocks` rrr/s,$; $  )DN33;tKGXYYY YL!)BBB !KtL7I!K!KLLLJJJJi6IJJJJJL U *40000l1CDNNtTTT D$  ;666667779:::r#rcHttj||S)aLoad the Layout object from the given CSV file. Args: filename (str): The name of the CSV file. A row of the table represents an individual layout element. block_type (str): If there's no block_type column in the CSV file, you must pass in a block_type variable such that layout parser can appropriately detect the type of the layout elements. Returns: Layout: The parsed Layout object from the CSV file. )r)load_dataframepdread_csv)r rs rload_csvr3\s!$ "+h//J G G GGr#dfc|}d|jvr5|djtkr|dd|d<|d|jvrt dn||d<t d|jDrd|jvr |j|d<t| dd  S) aLoad the Layout object from the given dataframe. Args: df (pd.DataFrame): block_type (str): If there's no block_type column in the CSV file, you must pass in a block_type variable such that layout parser can appropriately detect the type of the layout elements. Returns: Layout: The parsed Layout object from the CSV file. pointscVtj|stj|n|Sr)r1isnaast literal_evalxs rz load_dataframe..s#RWQZZF#*1---Qr#Nrz:`block_type` not specified both in dataframe and argumentsc32K|]}|tjvVdSr)r r+)r cols rr"z!load_dataframe..s* < <#3)% % < < < < < .s (:(:(<(<r#)axis) copycolumnsdtypeobjectmapr)r*indexrapplyto_list)r4rs rr0r0qs B2: h<  ' 'h<++FFBxL rz ) )L  * &< < < < < <<< rz ! !xBtH RXX<<1XEEMMOO P PPr#r)r9rtypingrrrrpandasr1elementsrr r r strrrr3 DataFramer0rr#rrSsn //////////////&7&? @$*;E$T *+*;6G6O0P*;*;*;*;ZHHsHHvHHHH*#Q#Qr|#Q#Q#Q#Q#Q#Q#Q#Qr#