Added secrets

.env

@@ -1,2 +0,0 @@
-OPENAI_API_KEY=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
-HF_API_KEY=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

app.py

@@ -1,27 +1,17 @@
 import os
 import openai
-import sys
-import io
 import gradio as gr
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.vectorstores.pgvector import PGVector
 from langchain.chat_models import ChatOpenAI
 from langchain.chains import ConversationalRetrievalChain
 from langchain.memory import ConversationTokenBufferMemory
-from dotenv import load_dotenv, find_dotenv
-_ = load_dotenv(find_dotenv()) # read local .env file
 
-openai.api_key  = os.environ['OPENAI_API_KEY']
-hf_api_key = os.environ['HF_API_KEY']
+openai.api_key = OPENAI_API_KEY
+hf_api_key = HF_API_KEY
 
 embedding = OpenAIEmbeddings()
 
-host = "experian-ai-instance-1.cyqxijnzhxga.us-west-2.rds.amazonaws.com"
-port = "5432"
-database_name = "experian"
-user = "larryyin"
-passwd = "experianai"
-
 CONNECTION_STRING = PGVector.connection_string_from_db_params(
     driver=os.environ.get("PGVECTOR_DRIVER", "psycopg2"),
     host=os.environ.get("PGVECTOR_HOST", host),
@@ -31,22 +21,21 @@ CONNECTION_STRING = PGVector.connection_string_from_db_params(
     password=os.environ.get("PGVECTOR_PASSWORD", passwd),
 )
 
-COLLECTION_NAME = "experian230725"
-
 vectordb = PGVector(embedding_function=embedding,
                   collection_name=COLLECTION_NAME,
                   connection_string=CONNECTION_STRING,
                  )
 
 # llm_name = "gpt-3.5-turbo"
-llm_name = "gpt-3.5-turbo-16k"
+# llm_name = "gpt-3.5-turbo-16k"
 # llm_name = "gpt-4-32k"
+
 llm = ChatOpenAI(model_name=llm_name, temperature=0)
 
 retriever=vectordb.as_retriever()
 memory = ConversationTokenBufferMemory(
     llm = llm,
-    max_token_limit=8000,
+    max_token_limit=memory_token_limit,
     memory_key="chat_history",
     return_messages=True
 )
@@ -58,7 +47,7 @@ qa = ConversationalRetrievalChain.from_llm(
 )
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Experian Bot V0.2")
+    gr.Markdown(f"# {bot_name}")
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Type your message (Shift + Enter to submit)", lines=6)
     submit = gr.Button("Submit")