correct bug

app.py

@@ -766,7 +766,7 @@ agent_ZEROSHOT_REACT_2 = initialize_agent(tools_remote, GPTfake,
 llm_chain = LLMChain(llm=llm, prompt=prompt)
 llm_chain_2 = LLMChain(llm=GPTfake, prompt=prompt)
 
-print("Test LLM Chain", llm_chain_2({'agent_scratchpad':"", 'input':"what is PDP?"}))
+# print("Test LLM Chain", llm_chain_2({'agent_scratchpad':"", 'input':"what is PDP?"}))
 
 # llm_chain_openai = LLMChain(llm=llm, prompt=prompt_openai, verbose=True)
 
@@ -1414,11 +1414,13 @@ def CreatDb_P():
 def QAQuery_p(question: str):
     global vectordb_p
     global agent
+    global Choice
     # vectordb = Chroma(persist_directory='db', embedding_function=embeddings)
     retriever = vectordb_p.as_retriever()
     retriever.search_kwargs['k'] = int(os.environ["search_kwargs_k"])
     # retriever.search_kwargs['fetch_k'] = 100
-    if agent == agent_ZEROSHOT_REACT_2 or agent == agent_ZEROSHOT_AGENT_2:
+    # if agent == agent_ZEROSHOT_REACT_2 or agent == agent_ZEROSHOT_AGENT_2:
+    if Choice == 'Zero Short React 2' or Choice == "Zero Short Agent 2":
         print("--------------- QA with Remote --------------")
         qa = RetrievalQA.from_chain_type(llm=GPTfake, chain_type="stuff", 
                                         retriever=retriever, return_source_documents = True,