# flask_app/general/清除file_id.py
import os
import openai
import json

# 初始化 OpenAI 客户端
openai.api_key = os.getenv("DASHSCOPE_API_KEY")
openai.api_base = "https://dashscope.aliyuncs.com/compatible-mode/v1"
def delete_all_files():
    """
    查询所有文件并删除。
    """
    try:
        # 获取文件列表
        file_stk = openai.File.list()

        # 将文件信息解析为字典格式
        file_data = file_stk.to_dict()

        # 提取所有文件的 id
        file_ids = [file["id"] for file in file_data["data"]]

        # 循环删除每个文件
        for file_id in file_ids:
            file_object = openai.File.delete(file_id)
            print(f"Deleted file with id: {file_id} - {file_object}")

    except Exception as e:
        print(f"An error occurred while deleting files: {e}")


def read_file_ids(output_folder):
    file_ids = []
    file_ids_path = os.path.join(output_folder, 'file_ids.txt')
    # 检查文件是否存在
    if os.path.exists(file_ids_path):
        try:
            with open(file_ids_path, 'r', encoding='utf-8') as file:
                # 按行读取文件内容
                file_ids = [line.strip() for line in file.readlines()]
            print(f"读取到的文件 ID 列表：{file_ids}")
        except Exception as e:
            print(f"读取 file_ids.txt 文件时发生错误: {e}")
    else:
        print(f"文件 {file_ids_path} 不存在。")

    return file_ids


def delete_file_by_ids(file_ids):
    """
    根据传入的 file_id 列表删除指定的文件。

    :param file_ids: 一个包含文件 ID 的字符串列表
    """
    if not isinstance(file_ids, list):
        print("Error: file_ids should be a list.")
        return

    if not all(isinstance(file_id, str) for file_id in file_ids):
        print("Error: Each file_id should be a string.")
        return

    try:
        # 删除指定文件
        for file_id in file_ids:
            try:
                # 假设 openai.File.delete 会返回一个文件对象
                file_object = openai.File.delete(file_id)
                # print(f"Deleted file with id: {file_id} - {file_object}")
            except Exception as e:
                # 处理删除单个文件时的异常
                print(f"Failed to delete file with id {file_id}: {e}")

    except Exception as e:
        print(f"An error occurred while processing the file_ids: {e}")


if __name__ == '__main__':
    delete_all_files()