Update rag.py

rag.py

@@ -41,18 +41,15 @@ def document_loading():
     # PDF
     loader = PyPDFLoader(PDF_URL)
     docs.extend(loader.load())
-    print("### Load PDF")
     
     # Web
     loader = WebBaseLoader(WEB_URL)
     docs.extend(loader.load())
-    print("### Load Web")
     
     # YouTube
     loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
                            OpenAIWhisperParser())
     docs.extend(loader.load())
-    print("### Load YouTube")
     
     return docs
 
@@ -60,17 +57,14 @@ def document_splitting(config, docs):
     text_splitter = RecursiveCharacterTextSplitter(chunk_overlap = config["chunk_overlap"],
                                                    chunk_size = config["chunk_size"])
     
-    print("### Split")
     return text_splitter.split_documents(docs)
     
 def document_storage_chroma(chunks):
-    print("### Store Chroma")
     Chroma.from_documents(documents = chunks, 
                           embedding = OpenAIEmbeddings(disallowed_special = ()), 
                           persist_directory = CHROMA_DIR)
 
 def document_storage_mongodb(chunks):
-    print("### Store MongoDB")
     MongoDBAtlasVectorSearch.from_documents(documents = chunks,
                                             embedding = OpenAIEmbeddings(disallowed_special = ()),
                                             collection = collection,