Update app.py

app.py

@@ -24,6 +24,13 @@ def get_root_domain(url):
         return '.'.join(parts[-2:])
     else:
         return domain
+
+
+def try_json_decode(headers):
+    try:
+        return json.loads(str(headers))
+    except Exception as e:
+        return headers
         
 @app.get("/")
 def main():
@@ -112,12 +119,12 @@ def chrome(url:str=None,wait:int=5,header:str=None,cookie:str=None):
     # 遍历输出过程中的har
     for request in driver.requests:
         if request.response:
-            network.append({"method":request.method, "status":request.response.status_code ,"url":request.url, "requestheaders":request.headers ,"responseheaders":request.response.headers})
+            network.append({"method":request.method, "status":request.response.status_code ,"url":request.url, "responseheaders":try_json_decode(request.response.headers), "requestheaders":try_json_decode(request.headers)})
         
     data = {
         "url": current_url,
         "page_source": page_source,
-        "cookies": cookies,
+        "end_cookies": cookies,
         "is_jump": is_jump,
         "network": network
     }