Update functions.py

functions.py

@@ -37,6 +37,7 @@ nltk.download('punkt')
 
 from nltk import sent_tokenize
 
+OPEN_AI_KEY = os.environ.get('OPEN_AI_KEY')
 time_str = time.strftime("%d%m%Y-%H%M%S")
 HTML_WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; 
 margin-bottom: 2.5rem">{}</div> """
@@ -125,7 +126,7 @@ def process_corpus(corpus, tok, title, embeddings, chunk_size=200, overlap=50):
 
     '''Process text for Semantic Search'''
     
-    pinecone.init(api_key="2d1e8029-2d84-4724-9f7c-a4f0f5ae908a", environment="us-west1-gcp")
+    pinecone.init(api_key=OPEN_AI_KEY, environment="us-west1-gcp")
 
     tokenizer = tok
     text_splitter = CharacterTextSplitter.from_huggingface_tokenizer(tokenizer,chunk_size=chunk_size,chunk_overlap=overlap,separator='. ')
@@ -741,5 +742,4 @@ def save_network_html(kb, filename="network.html"):
 
 nlp = get_spacy()    
 
-with st.spinner('Loading all models...'):
-    sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer, emb_tokenizer  = load_models()
+sent_pipe, sum_pipe, ner_pipe, cross_encoder, kg_model, kg_tokenizer, emb_tokenizer  = load_models()