Update app.py

app.py

@@ -31,6 +31,7 @@ CHROMA_DIR  = "docs/chroma"
 YOUTUBE_DIR = "docs/youtube"
 
 YOUTUBE_URL = "https://www.youtube.com/watch?v=--khbXchTeE"
+#YOUTUBE_URL = "https://www.youtube.com/watch?v=RfvL_423a-I&list=PL2yQDdvlhXf_hIzmfHCdbcXj2hS52oP9r&index=2"
 
 MODEL_NAME  = "gpt-4"
 
@@ -42,6 +43,7 @@ def invoke(openai_api_key, use_rag, prompt):
         if (os.path.isdir(CHROMA_DIR)):
             vector_db = Chroma(embedding_function = OpenAIEmbeddings(),
                                persist_directory = CHROMA_DIR)
+            print("Load DB")
         else:
             loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
                                    OpenAIWhisperParser())
@@ -52,6 +54,7 @@ def invoke(openai_api_key, use_rag, prompt):
             vector_db = Chroma.from_documents(documents = splits, 
                                               embedding = OpenAIEmbeddings(), 
                                               persist_directory = CHROMA_DIR)
+            print("Create DB")
         rag_chain = RetrievalQA.from_chain_type(llm, 
                                                 chain_type_kwargs = {"prompt": RAG_CHAIN_PROMPT}, 
                                                 retriever = vector_db.as_retriever(search_kwargs = {"k": 3}),