Update app.py

app.py

@@ -18,34 +18,27 @@ from wandb.sdk.data_types.trace_tree import Trace
 
 _ = load_dotenv(find_dotenv())
 
-WANDB_API_KEY = os.environ["WANDB_API_KEY"]
-
-DESCRIPTION = os.environ["DESCRIPTION"]
-
-MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
-MONGODB_DB_NAME = "langchain_db"
-MONGODB_COLLECTION_NAME = "gpt-4"
-MONGODB_INDEX_NAME = "default"
-
-template = """If you don't know the answer, just say that you don't know, don't try to make up an answer. Keep the answer as concise as possible. Always say "Thanks for using the 🧠 app - Bernd" at the end of the answer. """
-
-llm_template = "Answer the question at the end. " + template + "Question: {question} Helpful Answer: "
-rag_template = "Use the following pieces of context to answer the question at the end. " + template + "{context}. Question: {question} Helpful Answer: "
-
-LLM_CHAIN_PROMPT = PromptTemplate(input_variables = ["question"], template = llm_template)
-RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], template = rag_template)
-
-CHROMA_DIR  = "/data/chroma"
-YOUTUBE_DIR = "/data/youtube"
-
 PDF_URL       = "https://arxiv.org/pdf/2303.08774.pdf"
 WEB_URL       = "https://openai.com/research/gpt-4"
 YOUTUBE_URL_1 = "https://www.youtube.com/watch?v=--khbXchTeE"
 YOUTUBE_URL_2 = "https://www.youtube.com/watch?v=hdhZwyf24mE"
 YOUTUBE_URL_3 = "https://www.youtube.com/watch?v=vw-KWfKwvTQ"
 
-RAG_OFF = "Off"
-RAG_CHROMA = "Chroma"
+YOUTUBE_DIR = "/data/youtube"
+CHROMA_DIR  = "/data/chroma"
+
+MONGODB_ATLAS_CLUSTER_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
+MONGODB_DB_NAME           = "langchain_db"
+MONGODB_COLLECTION_NAME   = "gpt-4"
+MONGODB_INDEX_NAME        = "default"
+
+LLM_CHAIN_PROMPT = PromptTemplate(input_variables = ["question"], template = os.environ["LLM_TEMPLATE"])
+RAG_CHAIN_PROMPT = PromptTemplate(input_variables = ["context", "question"], template = os.environ["RAG_TEMPLATE"])
+
+WANDB_API_KEY = os.environ["WANDB_API_KEY"]
+
+RAG_OFF     = "Off"
+RAG_CHROMA  = "Chroma"
 RAG_MONGODB = "MongoDB"
 
 client = MongoClient(MONGODB_ATLAS_CLUSTER_URI)
@@ -155,7 +148,7 @@ def wandb_trace(rag_option, prompt, completion, result, generation_info, llm_out
         end_time_ms = end_time_ms
     )
     
-    trace.log("test")
+    trace.log("evaluation")
     wandb.finish()
 
 def invoke(openai_api_key, rag_option, prompt):
@@ -219,5 +212,5 @@ demo = gr.Interface(fn=invoke,
                              ],
                     outputs = [gr.Textbox(label = "Completion", lines = 1)],
                     title = "Generative AI - LLM & RAG",
-                    description = DESCRIPTION)
+                    description = os.environ["DESCRIPTION"])
 demo.launch()