Update app.py

app.py

@@ -22,11 +22,11 @@ template = """Use the following pieces of context to answer the question at the
 
 QA_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], template = template)
 
-vectordb = None
+initial_load = True
 
 def invoke(openai_api_key, youtube_url, prompt):
     openai.api_key = openai_api_key
-    if vectordb == None:
+    if initial_load == True:
         youtube_dir = "docs/youtube/"
         loader = GenericLoader(YoutubeAudioLoader([youtube_url], youtube_dir), OpenAIWhisperParser())
         docs = loader.load()
@@ -36,6 +36,7 @@ def invoke(openai_api_key, youtube_url, prompt):
         vectordb = Chroma.from_documents(documents = splits, embedding = OpenAIEmbeddings(), persist_directory = chroma_dir)
         llm = ChatOpenAI(model_name = "gpt-4", temperature = 0)
         qa_chain = RetrievalQA.from_chain_type(llm, retriever = vectordb.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
+        initial_load = False
     result = qa_chain({"query": prompt})
     #shutil.rmtree(youtube_dir)
     #shutil.rmtree(chroma_dir)