🚀 feat: Allow pptx Upload (#51)

danny-avila · Jun 26, 2024 · c289c20 · c289c20
1 parent 1d58df5
commit c289c20
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -34,6 +34,7 @@
     UnstructuredXMLLoader,
     UnstructuredRSTLoader,
     UnstructuredExcelLoader,
+    UnstructuredPowerPointLoader,
 )
 
 from models import (
@@ -285,6 +286,8 @@ def get_loader(filename: str, file_content_type: str, filepath: str):
         loader = UnstructuredRSTLoader(filepath, mode="elements")
     elif file_ext == "xml":
         loader = UnstructuredXMLLoader(filepath)
+    elif file_ext == "pptx":
+        loader = UnstructuredPowerPointLoader(filepath)
     elif file_ext == "md":
         loader = UnstructuredMarkdownLoader(filepath)
     elif file_content_type == "application/epub+zip":

diff --git a/requirements.txt b/requirements.txt
@@ -26,3 +26,5 @@ opencv-python-headless==4.9.0.80
 pymongo==4.6.3
 langchain-mongodb==0.1.3
 cryptography==42.0.7
+python-magic==0.4.27
+python-pptx==0.6.23