Update logic.py

logic.py

@@ -22,8 +22,8 @@ import numpy as np
 import plotly.express as px
 import umap
 
-def get_index(links):
-    os.environ["OPENAI_API_KEY"] = userdata.get('oai')
+def get_index(links,token):
+    #os.environ["OPENAI_API_KEY"] = userdata.get('oai')
     logging.basicConfig(stream=sys.stdout, level=logging.INFO)
     
     db = kuzu.Database("kg1")
@@ -34,7 +34,7 @@ def get_index(links):
         links
     )
     
-    llm = OpenAI(temperature=0, model="gpt-3.5-turbo",api_key='')
+    llm = OpenAI(temperature=0, model="gpt-3.5-turbo",api_key=token)
     service_context = ServiceContext.from_defaults(llm=llm, chunk_size=512)
     
     storage_context = StorageContext.from_defaults(graph_store=graph_store)
@@ -92,55 +92,3 @@ def get_visualize_embeddings(embedding_series, n_neighbors=15, min_dist=0.1, n_c
 
 
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-