Update app.py

app.py

@@ -24,6 +24,8 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], temp
 
 print(0)
 
+qa_chain = None 
+
 def invoke(openai_api_key, youtube_url, prompt):
     openai.api_key = openai_api_key
     if (os.path.isdir("docs/chroma/") == False):
@@ -36,9 +38,9 @@ def invoke(openai_api_key, youtube_url, prompt):
         chroma_dir = "docs/chroma/"
         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})
+        global qa_chain = RetrievalQA.from_chain_type(llm, retriever = vectordb.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
     print(2)
-    result = qa_chain({"query": prompt})
+    result = global qa_chain({"query": prompt})
     shutil.rmtree(youtube_dir)
     #shutil.rmtree(chroma_dir)
     return result["result"]