Update app.py

app.py

@@ -36,6 +36,7 @@ def invoke(openai_api_key, youtube_url, process_video, prompt):
         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())