Update app.py

app.py

@@ -44,7 +44,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")
+            print("2 Load DB")
         else:
             loader = GenericLoader(YoutubeAudioLoader([YOUTUBE_URL], YOUTUBE_DIR), 
                                    OpenAIWhisperParser())
@@ -55,7 +55,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")
+            print("1 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}),