Update initate.py
Browse files- initate.py +10 -10
initate.py
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
-
from utils.ingest_image import extract_and_store_images
|
2 |
-
from utils.ingest_text import create_vector_database
|
3 |
-
from utils.ingest_video import intiate_video
|
4 |
-
|
5 |
-
def process_pdf(pdf_path):
|
6 |
-
image_collection = extract_and_store_images(pdf_path)
|
7 |
-
text_collection = create_vector_database(pdf_path)
|
8 |
-
video_collection = intiate_video()
|
9 |
-
return image_collection, text_collection, video_collection
|
10 |
-
|
|
|
1 |
+
from utils.ingest_image import extract_and_store_images
|
2 |
+
from utils.ingest_text import create_vector_database
|
3 |
+
from utils.ingest_video import intiate_video
|
4 |
+
|
5 |
+
def process_pdf(pdf_path,temp_folder):
|
6 |
+
image_collection = extract_and_store_images(pdf_path)
|
7 |
+
text_collection = create_vector_database(pdf_path)
|
8 |
+
video_collection = intiate_video(temp_folder)
|
9 |
+
return image_collection, text_collection, video_collection
|
10 |
+
|