Update app.py

app.py

@@ -60,19 +60,20 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
     if type(cookie) == str:
         header_array.update({"cookie":unquote(cookie)})
 
-    if 'cookie' in header_array:
-        cookie_array = header_array['cookie']
-        del header_array['cookie']
-        cookie_domain = f'.{get_root_domain(target_url)}'
-        for key, value in cookie_array.items():
-            driver.execute_script(f'document.cookie = "{key}={value}; path=/; domain={cookie_domain}; expires=Thu, 01 Jan 2050 00:00:00 GMT;');
-        print(driver.execute_script(f'return document.cookie'))
+    
 
     options = Options()
     options.add_argument('--headless')
     
     driver = webdriver.Chrome(options=options)
 
+    if 'cookie' in header_array:
+        cookie_array = header_array['cookie']
+        del header_array['cookie']
+        cookie_domain = f'.{get_root_domain(target_url)}'
+        for key, value in cookie_array.items():
+            driver.add_cookie({"name": key, "value": value, "domain": cookie_domain, "path": "/"})
+            
     driver.header_overrides = header_array
     
     driver.get(target_url)