Update app.py

app.py

@@ -22,12 +22,9 @@ template = """Use the following pieces of context to answer the question at the
 
 QA_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], template = template)
 
-print(os.listdir("docs/chroma/"))
-print(os.listdir("docs/youtube/"))
-
 def invoke(openai_api_key, youtube_url, prompt):
     openai.api_key = openai_api_key
-    if (len(os.listdir("docs/chroma/")) == 0):
+    if (os.path.isdir("docs/chroma/")):
         youtube_dir = "docs/youtube/"
         loader = GenericLoader(YoutubeAudioLoader([youtube_url], youtube_dir), OpenAIWhisperParser())
         docs = loader.load()