Update app.py

app.py

@@ -60,11 +60,11 @@ def document_loading_splitting():
     loader = WebBaseLoader(WEB_URL_1)
     docs.extend(loader.load())
     # Load YouTube
-    loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL_1,
-                                               YOUTUBE_URL_2,
-                                               YOUTUBE_URL_3], YOUTUBE_DIR), 
-                           OpenAIWhisperParser())
-    docs.extend(loader.load())
+    #loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL_1,
+    #                                           YOUTUBE_URL_2,
+    #                                           YOUTUBE_URL_3], YOUTUBE_DIR), 
+    #                       OpenAIWhisperParser())
+    #docs.extend(loader.load())
     # Document splitting
     text_splitter = RecursiveCharacterTextSplitter(chunk_overlap = 150,
                                                    chunk_size = 1500)
@@ -124,8 +124,8 @@ def invoke(openai_api_key, rag_option, prompt):
             db = document_retrieval_chroma(llm, prompt)
             result = rag_chain(llm, prompt, db)
         elif (rag_option == "MongoDB"):
-            #splits = document_loading_splitting()
-            #document_storage_mongodb(splits)
+            splits = document_loading_splitting()
+            document_storage_mongodb(splits)
             db = document_retrieval_mongodb(llm, prompt)
             result = rag_chain(llm, prompt, db)
         else: