| 304 | } |
| 305 | |
| 306 | class FileReader: |
| 307 | def set_reader(self, suffix) -> None: |
| 308 | self.reader = READER_MAP[suffix] |
| 309 | logger.info(f"Setting Reader to {type(self.reader).__name__}") |
| 310 | |
| 311 | def read_file(self, file_path: Path, task="describe the file") -> str: |
| 312 | suffix = '.' + file_path.split(".")[-1] |
| 313 | self.set_reader(suffix) |
| 314 | if isinstance(self.reader, IMGReader) or isinstance(self.reader, VideoReader): |
| 315 | file_content = self.reader.parse(file_path, task) |
| 316 | else: |
| 317 | file_content = self.reader.parse(file_path) |
| 318 | logger.info(f"Reading file {file_path} using {type(self.reader).__name__}") |
| 319 | return file_content |
| 320 | |
| 321 | |
| 322 | class GeneralReader: |