Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import openai, os, time
+import time
 
 from dotenv import load_dotenv, find_dotenv
 
@@ -12,11 +12,6 @@ RAG_OFF     = "Off"
 RAG_CHROMA  = "Chroma"
 RAG_MONGODB = "MongoDB"
 
-config = {
-    "model_name": "gpt-4-0613",
-    "temperature": 0,
-}
-
 def invoke(openai_api_key, rag_option, prompt):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
@@ -35,26 +30,22 @@ def invoke(openai_api_key, rag_option, prompt):
     try:
         start_time_ms = round(time.time() * 1000)
 
-        llm = ChatOpenAI(model_name = config["model_name"], 
-                         openai_api_key = openai_api_key, 
-                         temperature = config["temperature"])
-        
         if (rag_option == RAG_CHROMA):
             #splits = document_loading_splitting()
             #document_storage_chroma(splits)
             
-            db = document_retrieval_chroma(llm, prompt)
-            completion, chain = rag_chain(llm, prompt, db)
+            db = document_retrieval_chroma(openai_api_key, prompt)
+            completion, chain = rag_chain(openai_api_key, prompt, db)
             result = completion["result"]
         elif (rag_option == RAG_MONGODB):
             #splits = document_loading_splitting()
             #document_storage_mongodb(splits)
             
-            db = document_retrieval_mongodb(llm, prompt)
-            completion, chain = rag_chain(llm, prompt, db)
+            db = document_retrieval_mongodb(openai_api_key, prompt)
+            completion, chain = rag_chain(openai_api_key, prompt, db)
             result = completion["result"]
         else:
-            completion, chain = llm_chain(llm, prompt)
+            completion, chain = llm_chain(openai_api_key, prompt)
             
             if (completion.generations[0] != None and completion.generations[0][0] != None):
                 result = completion.generations[0][0].text