import requests
import os
url = "http://127.0.0.1:5000/api/v1/rag_upload"
# file_path1 = r"终版-（0119）沭阳县循环经济产业园污废水资源化项目方案简述(3).docx"
# file_path2 = r"（0119）沭阳县循环经济产业园污废水资源化项目方案简述(3).pdf"
file_path3 = r"D:\code\repository\污水处理工艺手册.pdf"
# file_list = [file_path1, file_path2]
file_list = [file_path3]


# 获取文件和路径
#root = r'D:\code\tem_1103\RAG资料库—开发\污水处理工艺知识'
#file_list = [os.path.join(root, i) for i in os.listdir(root) if i.split('.')[-1] == 'pdf' or i.split('.')[-1] == 'docx']
# tem_root = r"D:\code\rag_tools\chat_output"
# tem_list = [os.path.join(root, i).replace("md","docx") for i in os.listdir(tem_root) if i.split('.')[-1] == 'md']
#new_file_list = [i for i in file_list if i not in tem_list]
#以二进制模式打开文件并上传
for file in file_list:
    with open(file, "rb") as f:
        files = {"file": f}
        response = requests.post(url, files=files)
    print("状态码:", response.status_code)
    print("响应内容:", response.json().get("message"))
print()