Update app.py

app.py

@@ -22,21 +22,22 @@ template = """Use the following pieces of context to answer the question at the
 
 QA_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], template = template)
 
+CHROMA_DIR  = "docs/chroma/"
 YOUTUBE_DIR = "docs/youtube/" 
-CHROMA_DIR = "docs/chroma/"
 
 MODEL_NAME = "gpt-4"
 
 def invoke(openai_api_key, youtube_url, process_video, prompt):
     openai.api_key = openai_api_key
-    print(process_video)
     if (process_video):
+        if (os.path.isdir(CHROMA_DIR)):
+            shutil.rmtree(CHROMA_DIR)
+        if (os.path.isdir(YOUTUBE_DIR)):
+            shutil.rmtree(YOUTUBE_DIR)
         loader = GenericLoader(YoutubeAudioLoader([youtube_url], YOUTUBE_DIR), OpenAIWhisperParser())
         docs = loader.load()
-        shutil.rmtree(YOUTUBE_DIR)
         text_splitter = RecursiveCharacterTextSplitter(chunk_size = 1500, chunk_overlap = 150)
         splits = text_splitter.split_documents(docs)
-        shutil.rmtree(CHROMA_DIR)
         vector_db = Chroma.from_documents(documents = splits, embedding = OpenAIEmbeddings(), persist_directory = CHROMA_DIR)
     else:
         vector_db = Chroma(persist_directory = CHROMA_DIR, embedding_function = OpenAIEmbeddings())