Coverage for apps/inners/models/dtos/element_category.py: 89%

1from typing import List, Union 

2 

3from unstructured.documents.elements import Text, Table, Image, NarrativeText 

4 

5from apps.inners.models.base_model import BaseModel 

6 

7 

8class ElementCategory(BaseModel): 

9 texts: List[Union[Text, NarrativeText]] 

10 tables: List[Union[Table]] 

11 images: List[Image] 

12 

13 def to_dict(self): 

14 return { 

15 "texts": [text.to_dict() for text in self.texts], 

16 "tables": [table.to_dict() for table in self.tables], 

17 "images": [image.to_dict() for image in self.images] 

18 }