Update logic.py

logic.py

@@ -31,7 +31,7 @@ def save_uploadedfile(uploadedfile):
     with open(os.path.join("data",uploadedfile.name),"wb") as f:
         f.write(uploadedfile.getbuffer())
 
-def load_index(token,name):
+def load_index(token, name, base_url):
     os.environ["OPENAI_API_KEY"] = token
     os.environ["OPENAI_API_BASE"] = base_url
     logging.basicConfig(stream=sys.stdout, level=logging.INFO)
@@ -46,7 +46,7 @@ def load_index(token,name):
     return index
 
 
-def get_index_pdf(token,name):
+def get_index_pdf(token, name, base_url):
     documents = SimpleDirectoryReader("./data").load_data()
     print(documents)
     print(documents)
@@ -72,7 +72,7 @@ def get_index_pdf(token,name):
 
     return index
 
-def get_index(links,token,name):
+def get_index(links, token, name, base_url):
     os.mkdir(name) 
     os.environ["OPENAI_API_KEY"] = token
     os.environ["OPENAI_API_BASE"] = base_url