Update app.py

app.py

@@ -17,11 +17,11 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "")
 OPENAI_TEMP  = 0
 
-PINECONE_KEY = os.environ["PINECONE_KEY"]
-PINECONE_ENV = "asia-northeast1-gcp"
-PINECONE_INDEX = "3gpp"
+PINECONE_KEY = os.environ.get("PINECONE_KEY", "")
+PINECONE_ENV = os.environ.get("PINECONE_ENV", "asia-northeast1-gcp")
+PINECONE_INDEX = os.environ.get("PINECONE_INDEX", "3gpp")
 
-EMBEDDING_MODEL = "sentence-transformers/all-mpnet-base-v2"
+EMBEDDING_MODEL = os.environ.get("PINECONE_INDEX", "sentence-transformers/all-mpnet-base-v2")
 
 # return top-k text chunks from vector store
 TOP_K_DEFAULT = 10
@@ -47,8 +47,8 @@ MODEL_WARNING = "Please paste your OpenAI API Key from openai.com and press 'Ent
 
 
 webui_title = """
-# 3GPP OpenAI Chatbot for Hackathon Demo
-
+# OpenAI Chatbot Based on Vector Database
+## Example of 3GPP
 """
 
 KEY_INIT   = "Initialize Model"
@@ -56,21 +56,22 @@ KEY_SUBMIT = "Submit"
 KEY_CLEAR  = "Clear"
 
 init_message = f"""Welcome to use 3GPP Chatbot, this demo toolkit is based on OpenAI with LangChain and Pinecone
-    1. Insert your OpenAI API key and click `{KEY_INIT}`
-    2. Insert your Question and click `{KEY_SUBMIT}`
+    1. Insert your OpenAI API key and click  `{KEY_INIT}`
+    2. Insert your Question and click  `{KEY_SUBMIT}`
 """
 
 #----------------------------------------------------------------------------------------------------------
 #----------------------------------------------------------------------------------------------------------
 
-def init_model(api_key):
+def init_model(api_key, emb_name, db_api_key, db_env, db_index):
     try:
-        if api_key and api_key.startswith("sk-") and len(api_key) > 50:
+        if (api_key and api_key.startswith("sk-") and len(api_key) > 50) and \
+        (emb_name and db_api_key and db_env and db_index):
             
-            embeddings = HuggingFaceEmbeddings(model_name=EMBEDDING_MODEL)
+            embeddings = HuggingFaceEmbeddings(model_name=emb_name)
 
-            pinecone.init(api_key     = PINECONE_KEY,
-                          environment = PINECONE_ENV)
+            pinecone.init(api_key     = db_api_key,
+                          environment = db_env)
 
             #llm = OpenAI(temperature=OPENAI_TEMP, model_name="gpt-3.5-turbo-0301")
 
@@ -79,7 +80,7 @@ def init_model(api_key):
 
             chain = load_qa_chain(llm, chain_type="stuff")
 
-            db = Pinecone.from_existing_index(index_name = PINECONE_INDEX,
+            db = Pinecone.from_existing_index(index_name = db_index,
                                               embedding  = embeddings)
 
             return api_key, MODEL_DONE, chain, db, None
@@ -173,7 +174,7 @@ with gr.Blocks(
     
     with gr.Row():
         with gr.Column(scale=10):
-            api_textbox = gr.Textbox(
+            llm_api_textbox = gr.Textbox(
                 label = "OpenAI API Key",
                 show_label = False,
                 value = OPENAI_API_KEY,
@@ -218,14 +219,49 @@ with gr.Blocks(
                           label="Vector similarity top_k",
                           interactive=True)
         detail_panel = gr.Chatbot(label="Related Docs")
-        
+    
+    with gr.Tab("Database"):
+        with gr.Row():
+            emb_textbox = gr.Textbox(
+                label = "Embedding Model",
+                # show_label = False,
+                value = EMBEDDING_MODEL,
+                placeholder = "Paste Your Embedding Model Repo on HuggingFace",
+                lines=1,
+                interactive=True,
+                type='email')
+        with gr.Row():
+            db_api_textbox = gr.Textbox(
+                label = "Pinecone API Key",
+                # show_label = False,
+                value = PINECONE_KEY,
+                placeholder = "Paste Your Pinecone API Key (xx-xx-xx-xx-xx) and Hit ENTER",
+                lines=1,
+                interactive=True,
+                type='password')
+        with gr.Row():
+            db_env_textbox = gr.Textbox(
+                label = "Pinecone Environment",
+                # show_label = False,
+                value = PINECONE_ENV,
+                placeholder = "Paste Your Pinecone Environment (xx-xx-xx) and Hit ENTER",
+                lines=1,
+                interactive=True,
+                type='email')
+            db_index_textbox = gr.Textbox(
+                label = "Pinecone Index",
+                # show_label = False,
+                value = PINECONE_INDEX,
+                placeholder = "Paste Your Pinecone Index (xxxx) and Hit ENTER",
+                lines=1,
+                interactive=True,
+                type='email')
+
+    init_input = [llm_api_textbox, emb_textbox, db_api_textbox, db_env_textbox, db_index_textbox]
+    init_output = [llm_api_textbox, model_statusbox, llm_chain, vector_db, chatbot]
                 
-    api_textbox.submit(init_model,
-                       api_textbox,
-                       [api_textbox, model_statusbox, llm_chain, vector_db, chatbot])
-    init.click(init_model,
-                       api_textbox,
-                       [api_textbox, model_statusbox, llm_chain, vector_db, chatbot])
+    llm_api_textbox.submit(init_model, init_input, init_output)
+    init.click(init_model, init_input, init_output)
     
     submit.click(user,
                  [query, chatbot],