Update app.py

app.py

@@ -53,17 +53,17 @@ def invoke(openai_api_key, prompt, rag_option):
         start_time_ms = round(time.time() * 1000)
 
         if (rag_option == RAG_LANGCHAIN):
-            prompt_template = ""
+            prompt_template = os.environ["LANGCHAIN_TEMPLATE"]
             rag = LangChainRAG()
             completion, chain, callback = rag.rag_chain(config, prompt)
 
             result = completion["result"]
         elif (rag_option == RAG_LLAMAINDEX):
-            prompt_template = ""
+            prompt_template = os.environ["LLAMAINDEX_TEMPLATE"]
             rag = LlamaIndexRAG()
             result = rag.retrieval(config, prompt)
         else:
-            prompt_template = ""
+            prompt_template = os.environ["TEMPLATE"]
             rag = LangChainRAG()
             completion, chain, callback = rag.llm_chain(config, prompt)