Spaces:
Runtime error
Runtime error
File size: 1,709 Bytes
c05c725 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
import requests
import pickle
import browser_cookie3
import selenium.webdriver
import os
cookie_name = "openAIKey"
cookie_fname = "cookies.pcl"
def saveOpenAIKey(value):
global cookie_name, cookie_fname
print(f"Saving the value in cookie...")
s = requests.session()
s.cookies.set(cookie_name, value)
#print(f"Session cookies before save: {s.cookies}")
# Save the cookies to file:
#with open(cookie_fname, 'wb') as f:
# pickle.dump(s.cookies, f)
# Chrome browser
try:
driver = selenium.webdriver.Chrome()
driver.get("https://huggingface.co")
driver.add_cookie({cookie_name: value})
except Exception as e:
print(f"Exception: {e}")
def loadOpenAIKey():
global cookie_name, cookie_fname
openAIkey = None
print(f"Loading the value from cookie...")
s = requests.session()
#try:
# if os.path.exists(cookie_fname):
# with open(cookie_fname, 'rb') as f:
# s.cookies.update(pickle.load(f))
#except Exception as e:
# print(f"Exception: {f}")
print(f"Saved cokies: {s.cookies}")
openAIkey = s.cookies.get(cookie_name)
print(f"Server cookie: {openAIkey!=None}")
if openAIkey == None:
try:
driver = selenium.webdriver.Chrome()
driver.get("https://huggingface.co")
print("Cookies from Chrome:")
for cookie in driver.get_cookies():
print(cookie)
if cookie_name in cookie:
print("Found open ai key!")
openAIkey = cookie[cookie_name]
except Exception as e:
print(f"Exception: {e}")
return openAIkey |