Edit Captions JSON Format

编辑captions-json-format

此Python脚本从CSV文件中读取图像标题，使用OpenAI的API缩短它们，然后将它们保存在JSON文件中。如果您需要图像标题等于或小于某些角色计数以进行可访问性，社交媒体或训练图像模型，则这可能特别有用。

特征

环境变量管理

从.env文件安全地加载API键和其他敏感设置。

OpenAI API集成

使用OpenAI的GPT-3.5-Turbo模型有效缩短字幕。

CSV输入

从CSV文件中读取字幕，该字幕允许对多个字幕进行批处理处理。

JSON输出

以JSON格式输出处理的字幕，使其易于与Web应用程序或其他媒体集成。设置和用法

先决条件

• Python安装在您的系统上。
• OpenAI API键。
• 具有至少两个列的CSV文件：文件名和标题。

安装

1. 建立Python环境（推荐）：

 python -m venv venv
source venv/bin/activate  # On Windows use `venvScriptsactivate`

2. 安装所需的软件包：

 pip install openai python-dotenv

3. 准备您的环境：

在与脚本相同的目录中创建一个.ENV文件，并添加您的OpenAI API密钥：

 OPENAI_API_KEY='your_api_key_here'

用法

准备您的CSV文件

确保使用文件名和标题标题正确格式化CSV文件。

设置您的路

修改脚本以在process_captions（'您的路径此处'）调用中包含CSV文件的正确路径。

运行脚本

输出将在captions.json中。