Update app.py

app.py

@@ -8,6 +8,34 @@ def generate_uuid():
     formatted_uuid = f"{random_uuid_str[0:8]}-{random_uuid_str[9:13]}-{random_uuid_str[14:18]}-{random_uuid_str[19:23]}-{random_uuid_str[24:]}"
     return formatted_uuid
 
+
+# Deletes the conversation
+def delete_conversation(user_id,channel_id,cookie):
+    url = f"https://claude.ai/api/organizations/{user_id}/chat_conversations/{channel_id}"
+
+    payload = json.dumps(f"{channel_id}")
+    headers = {
+        'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36 Edg/114.0.1823.82',
+        'Accept-Language': 'en-US,en;q=0.5',
+        'Content-Type': 'application/json',
+        'Content-Length': '38',
+        'Referer': 'https://claude.ai/chats',
+        'Origin': 'https://claude.ai',
+        'Sec-Fetch-Dest': 'empty',
+        'Sec-Fetch-Mode': 'cors',
+        'Sec-Fetch-Site': 'same-origin',
+        'Connection': 'keep-alive',
+        'Cookie': f'{cookie}',
+        'TE': 'trailers'
+    }
+
+    response = requests.delete(url, headers=headers, data=payload)
+    # Returns True if deleted or False if any error in deleting
+    if response.status_code == 204:
+        return True
+    else:
+        return False
+
 def create_new_chat(user_id,cookie):
     url = f"https://claude.ai/api/organizations/{user_id}/chat_conversations"
     uuid = generate_uuid()
@@ -35,11 +63,11 @@ def create_new_chat(user_id,cookie):
     # Returns JSON of the newly created conversation information
     return response.json()
 
-def request_to_v2(message, cookie, user_id, channel_id,context=[]):
+def request_to_v2(message, cookie, user_id,context=[]):
+
+    channel_id = create_new_chat(user_id,cookie)
+    print(channel_id)
 
-    datas = create_new_chat(user_id,cookie)
-    print(datas)
-    exit(0)
     
     timeout = 5*60 #5分钟不回复，显示超时
     context = [message]
@@ -107,6 +135,13 @@ def request_to_v2(message, cookie, user_id, channel_id,context=[]):
 
         return responses, context
 
+    finally:
+        ds = delete_conversation(user_id,channel_id,cookie)
+        if ds:
+            print(f"成功删除")
+        else:
+            print(f"删除失败")
+
 
 
 with gr.Blocks() as dialog_app:
@@ -123,7 +158,7 @@ with gr.Blocks() as dialog_app:
             with gr.Column(scale=0.4):
                 cookies = gr.Textbox(lines=2, label="输入cookies")
                 user_id = gr.Textbox(lines=2, label="输入user_id/organization_uuid")
-                channel_id = gr.Textbox(lines=2, label="输入channel_id/conversation_uuid")
+                # channel_id = gr.Textbox(lines=2, label="输入channel_id/conversation_uuid")
                 # chatbot = gr.Chatbot()
             with gr.Column(scale=0.6):
                 chatbot = gr.Chatbot([])
@@ -135,9 +170,9 @@ with gr.Blocks() as dialog_app:
                 label="输入问题",
                 placeholder="请输入你的文本，确保已经正确填入cookies、user_id、channel_id"
             )
-        inputs.submit(request_to_v2, [inputs, cookies, user_id,channel_id,state], [chatbot, state])
+        inputs.submit(request_to_v2, [inputs, cookies, user_id,state], [chatbot, state])
         send = gr.Button("发送请求.....")
-        send.click(request_to_v2, [inputs, cookies, user_id,channel_id,state], [chatbot, state], api_name="xiaolvgpt", show_progress=True)
+        send.click(request_to_v2, [inputs, cookies, user_id,state], [chatbot, state], api_name="xiaolvgpt", show_progress=True)
 
 
         gr.Markdown("""