Update rag.py

rag.py

@@ -49,9 +49,9 @@ def document_loading():
     print("### Load Web")
     
     # YouTube
-    #loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
-    #                       OpenAIWhisperParser())
-    #docs.extend(loader.load())
+    loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
+                           OpenAIWhisperParser())
+    docs.extend(loader.load())
     print("### Load YouTube")
     
     return docs
@@ -122,7 +122,7 @@ def rag_chain(config, openai_api_key, rag_option, prompt):
                                             chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT}, 
                                             retriever = db.as_retriever(search_kwargs = {"k": config["k"]}), 
                                             return_source_documents = True,
-                                            verbose = True)
+                                            verbose = False)
     
     completion = rag_chain({"query": prompt})
     print(completion)