Update app.py

app.py

@@ -60,6 +60,11 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
     if type(cookie) == str:
         header_array.update({"cookie":unquote(cookie)})
 
+    options = Options()
+    options.add_argument('--headless')
+    
+    driver = webdriver.Chrome(options=options)
+
     if 'cookie' in header_array:
         cookie_array = convert_cookies_to_dict(header_array['cookie'])
         del header_array['cookie']
@@ -67,12 +72,7 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
         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)
-
+        
     driver.header_overrides = header_array
     
     driver.get(target_url)