Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import uuid
 import time
 import random
-
+import os
 
 # 创建新的聊天频道
 def generate_uuid():
@@ -67,7 +67,75 @@ def create_new_chat(user_id,cookie):
     # Returns JSON of the newly created conversation information
     return response.json()
 
+def get_content_type(file_path):
+    # Function to determine content type based on file extension
+    extension = os.path.splitext(file_path)[-1].lower()
+    if extension == '.pdf':
+        return 'application/pdf'
+    elif extension == '.txt':
+        return 'text/plain'
+    elif extension == '.csv':
+        return 'text/csv'
+    # Add more content types as needed for other file types
+    else:
+        return 'application/octet-stream'
+
+def upload_attachment(cookie ,organization_id,file_path):
+    url = 'https://claude.ai/api/convert_document'
+    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',
+        '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'
+    }
+
+    file_name = os.path.basename(file_path)
+    print(file_name)
+    content_type = get_content_type(file_path)
+    print(content_type)
+    files = {
+        'file': (file_name, open(file_path, 'rb'), content_type),
+        'orgUuid': (None, organization_id)
+    }
+    print(files)
+    response = requests.post(url, headers=headers, files=files)
+    # 使用代理，解决国内网络不能访问问题
+    #response = request(url, headers=headers, files=files,proxies=proxies)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        return False
+
+
+
+
+
+
+
+
 def request_to_v2(message, cookie, user_id,context=[]):
+    ## 是否加入文件
+    # Upload attachment if provided
+    attachment = attachment.name
+    attachments = []
+    if attachment:
+        attachment_response = upload_attachment(cookie,user_id,attachment)
+        if attachment_response:
+            attachments = [attachment_response]
+        else:
+            return {"Error: Invalid file format. Please try again."}
+
+    # Ensure attachments is an empty list when no attachment is provided
+    if not attachment:
+        attachments = []        
+
+
 
     response_json = create_new_chat(user_id,cookie)
     channel_id = response_json.get('uuid',"")
@@ -108,7 +176,7 @@ def request_to_v2(message, cookie, user_id,context=[]):
         "organization_uuid": user_id,
         "conversation_uuid": channel_id,
         "text": message,
-        "attachments": []
+        "attachments": attachments
     }
     post_msg_data = json.dumps(post_msg_data)
     # print(post_msg_data)
@@ -182,6 +250,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")
+                file = gr.File(label='请上传知识库文件, 目前支持txt、docx、md格式',file_types=['.txt', '.csv', '.pdf'])
                 # channel_id = gr.Textbox(lines=2, label="输入channel_id/conversation_uuid")
                 # chatbot = gr.Chatbot()
             with gr.Column(scale=0.6):
@@ -194,9 +263,9 @@ with gr.Blocks() as dialog_app:
                 label="输入问题",
                 placeholder="请输入你的文本，确保已经正确填入cookies、user_id"
             )
-        inputs.submit(request_to_v2, [inputs, cookies, user_id,state], [chatbot, state])
+        inputs.submit(request_to_v2, [inputs, cookies, user_id,file,state], [chatbot, state])
         send = gr.Button("发送请求.....")
-        send.click(request_to_v2, [inputs, cookies, user_id,state], [chatbot, state], api_name="xiaolvgpt", show_progress=True)
+        send.click(request_to_v2, [inputs, cookies, user_id,file,state], [chatbot, state], api_name="xiaolvgpt", show_progress=True)
 
 
         gr.Markdown("""