aboutsummaryrefslogtreecommitdiff
path: root/.venv/lib/python3.12/site-packages/core/parsers/media/pptx_parser.py
diff options
context:
space:
mode:
authorS. Solomon Darnell2025-03-28 21:52:21 -0500
committerS. Solomon Darnell2025-03-28 21:52:21 -0500
commit4a52a71956a8d46fcb7294ac71734504bb09bcc2 (patch)
treeee3dc5af3b6313e921cd920906356f5d4febc4ed /.venv/lib/python3.12/site-packages/core/parsers/media/pptx_parser.py
parentcc961e04ba734dd72309fb548a2f97d67d578813 (diff)
downloadgn-ai-master.tar.gz
two version of R2R are hereHEADmaster
Diffstat (limited to '.venv/lib/python3.12/site-packages/core/parsers/media/pptx_parser.py')
-rw-r--r--.venv/lib/python3.12/site-packages/core/parsers/media/pptx_parser.py40
1 files changed, 40 insertions, 0 deletions
diff --git a/.venv/lib/python3.12/site-packages/core/parsers/media/pptx_parser.py b/.venv/lib/python3.12/site-packages/core/parsers/media/pptx_parser.py
new file mode 100644
index 00000000..8685c8fb
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/core/parsers/media/pptx_parser.py
@@ -0,0 +1,40 @@
+# type: ignore
+from io import BytesIO
+from typing import AsyncGenerator
+
+from pptx import Presentation
+
+from core.base.parsers.base_parser import AsyncParser
+from core.base.providers import (
+ CompletionProvider,
+ DatabaseProvider,
+ IngestionConfig,
+)
+
+
+class PPTXParser(AsyncParser[str | bytes]):
+ """A parser for PPT data."""
+
+ def __init__(
+ self,
+ config: IngestionConfig,
+ database_provider: DatabaseProvider,
+ llm_provider: CompletionProvider,
+ ):
+ self.database_provider = database_provider
+ self.llm_provider = llm_provider
+ self.config = config
+ self.Presentation = Presentation
+
+ async def ingest(
+ self, data: str | bytes, **kwargs
+ ) -> AsyncGenerator[str, None]: # type: ignore
+ """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