blob: 8f1928400a534ad198b179697ab62b44f54cb709 (
about) (
plain)
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
28
29
30
|
from io import BytesIO
from typing import AsyncGenerator
from r2r.base.abstractions.document import DataType
from r2r.base.parsers.base_parser import AsyncParser
class PPTParser(AsyncParser[DataType]):
"""A parser for PPT data."""
def __init__(self):
try:
from pptx import Presentation
self.Presentation = Presentation
except ImportError:
raise ValueError(
"Error, `python-pptx` is required to run `PPTParser`. Please install it using `pip install python-pptx`."
)
async def ingest(self, data: DataType) -> AsyncGenerator[str, None]:
"""Ingest PPT data and yield text from each slide."""
if isinstance(data, str):
raise ValueError("PPT data must be in bytes format.")
prs = self.Presentation(BytesIO(data))
for slide in prs.slides:
for shape in slide.shapes:
if hasattr(shape, "text"):
yield shape.text
|
