Update app.py

app.py

@@ -1,8 +1,11 @@
-import json, requests
+import json
 import gradio as gr
 import uuid
 import time
 import random
+import os
+from curl_cffi import requests
+import re
 
 
 # 创建新的聊天频道
@@ -33,7 +36,7 @@ def delete_conversation(user_id,channel_id,cookie):
         'TE': 'trailers'
     }
 
-    response = requests.delete(url, headers=headers, data=payload)
+    response = requests.delete(url, headers=headers, data=payload,impersonate="chrome110")
     # Returns True if deleted or False if any error in deleting
     if response.status_code == 204:
         return True
@@ -58,10 +61,10 @@ def create_new_chat(user_id,cookie):
         'Sec-Fetch-Site': 'same-origin',
     }
 
-    # response = requests.request("POST", url, headers=headers, data=payload)
+    # response = requests.request("POST", url, headers=headers, data=payload,impersonate="chrome110")
     # 使用代理，解决国内网络不能访问问题
     # response = request("POST", url, headers=headers, data=payload, proxies=proxies)
-    response = requests.post(url, headers=headers, data=payload)
+    response = requests.post(url, headers=headers, data=payload,impersonate="chrome110")
     print(f"新的对话页面：{response.json()}")
 
     # Returns JSON of the newly created conversation information
@@ -120,21 +123,21 @@ def request_to_v2(message, cookie, user_id,context=[]):
 
     try:
         print(f"输入的问题：{message}")
-        response = requests.post(post_msg_url, headers=headers, data=post_msg_data, verify=False, stream=False,timeout = timeout)
-        bots = ""
-        print("a"*100)
-        for data in response.iter_lines():
-            if data:
-                # 将字节字符串转换为字符串
-                data_str = data.decode('utf-8')
-                # 删除开头的 "data: " 部分
-                data_str = data_str.replace("data: ", "")
-                data_json = json.loads(data_str)
-                # print(data_json)
-                bots +=  data_json["completion"]
+        response = requests.post(post_msg_url, headers=headers, data=post_msg_data, verify=False,timeout = timeout,impersonate="chrome110")
+        decoded_data = response.content.decode("utf-8")
+        decoded_data = re.sub('\n+', '\n', decoded_data).strip()
+        data_strings = decoded_data.split('\n')
+        completions = []
+        for data_string in data_strings:
+            json_str = data_string[6:].strip()
+            data = json.loads(json_str)
+            if 'completion' in data:
+                completions.append(data['completion'])
+        bots = ''.join(completions)
         print(bots)
         context += [bots]
         responses = [(u, b) for u, b in zip(context[::2], context[1::2])]
+        print("a"*100)
         return responses, context
 
     except Exception as e: