Update app.py

app.py

@@ -135,15 +135,16 @@ def upload_attachment(cookie ,organization_id,file_path):
 
 
 
-def request_to_v2(message, cookie, user_id,attachment=None,context=[]):
+def request_to_v2(message, cookie, user_id,attachment=None,delst=True,context=[]):
     context = [message]
     
     
     ## 是否加入文件
     # Upload attachment if provided
-    attachment = attachment.name
+    
     attachments = []
     if attachment:
+        attachment = attachment.name
         attachment_response = upload_attachment(cookie,user_id,attachment)
         if attachment_response:
             attachments = [attachment_response]
@@ -241,12 +242,13 @@ def request_to_v2(message, cookie, user_id,attachment=None,context=[]):
         return responses, context
 
     finally:
-        time.sleep(random.uniform(1,2))
-        ds = delete_conversation(user_id,channel_id,cookie)
-        if ds:
-            print(f"成功删除")
-        else:
-            print(f"删除失败")
+        if delst:
+            time.sleep(random.uniform(1,2))
+            ds = delete_conversation(user_id,channel_id,cookie)
+            if ds:
+                print(f"成功删除")
+            else:
+                print(f"删除失败")
 
         print("*-"*100)
 
@@ -278,6 +280,7 @@ with gr.Blocks() as dialog_app:
                 cookies = gr.Textbox(lines=2, label="输入cookies")
                 user_id = gr.Textbox(lines=2, label="输入user_id/organization_uuid")
                 file = gr.File(label='请上传知识库文件, 目前支持txt、csv、pdf格式',file_types=['.txt', '.csv', '.pdf'])
+                delst = gr.Checkbox(label="是", info="在该对话回复后，是否要删除该对话？",value = True)
                 # channel_id = gr.Textbox(lines=2, label="输入channel_id/conversation_uuid")
                 # chatbot = gr.Chatbot()
             with gr.Column(scale=0.6):
@@ -290,9 +293,9 @@ with gr.Blocks() as dialog_app:
                 label="输入问题",
                 placeholder="请输入你的文本，确保已经正确填入cookies、user_id"
             )
-        inputs.submit(request_to_v2, [inputs, cookies, user_id,file,state], [chatbot, state])
+        inputs.submit(request_to_v2, [inputs, cookies, user_id,file,delst,state], [chatbot, state])
         send = gr.Button("发送请求.....")
-        send.click(request_to_v2, [inputs, cookies, user_id,file,state], [chatbot, state], api_name="xiaolvgpt", show_progress=True)
+        send.click(request_to_v2, [inputs, cookies, user_id,file,delst,state], [chatbot, state], api_name="xiaolvgpt", show_progress=True)
 
 
         gr.Markdown("""