Photoshot

一款开源的AI头像生成Web应用

快来 photoshot.app 体验吧！

技术栈

• ▲ Next.js 用于构建Web应用
• 🖼 Chakra UI 用于UI组件
• 📦 Prisma 用于数据库
• 🧠 Replicate，一个在云端运行机器学习模型的平台
• 💰 Stripe 用于支付
• 👩‍🎨 Flux 一个开源的文字到图像生成模型

快速上手

安装依赖：

yarn install

你可以使用Docker来运行本地PostgreSQL数据库和Maildev服务器（可通过http://localhost:1080访问）：

docker-compose up -d

创建.env.local文件：

cp .env.example .env.local

更新环境变量值：

环境变量	说明
DATABASE_URL	你的PostgreSQL数据库连接字符串。如果你使用的是提供的Docker配置，则为 postgresql://photoshot:photoshot@localhost:5432/photoshot。
NEXTAUTH_URL	你的Next.js应用的URL，用于与NextAuth.js进行身份验证。
S3_UPLOAD_KEY	用于存储图片的AWS S3存储桶的访问密钥。
S3_UPLOAD_SECRET	用于存储图片的AWS S3存储桶的秘密访问密钥。
S3_UPLOAD_BUCKET	用于存储图片的AWS S3存储桶名称。
S3_UPLOAD_REGION	你的S3存储桶所在的AWS区域。
REPLICATE_API_TOKEN	Replicate的API令牌。
REPLICATE_USERNAME	与你的Replicate账户关联的用户名。
REPLICATE_MAX_TRAIN_STEPS	Dreambooth AI模型的最大训练步数。默认值为 3000。
REPLICATE_NEGATIVE_PROMPT	用于Replicate AI模型中负面训练示例的提示词。默认值为 cropped face, cover face, cover visage, mutated hands
REPLICATE_HD_VERSION_MODEL_ID	用于放大生成图像的模型版本。你可以在这里浏览此类模型
NEXT_PUBLIC_REPLICATE_INSTANCE_TOKEN	训练数据的唯一标识符。可以是任意字符串。为了获得最佳效果，建议使用包含三个Unicode字符且不含空格的标识符，例如 cjw
SECRET	用于NextAuth.js身份验证的随机字符串。
EMAIL_FROM	发送邮件时使用的发件人邮箱地址。
EMAIL_SERVER	用于发送邮件的SMTP服务器URL。如果你使用的是提供的Docker配置，则为 http://localhost:25，
STRIPE_SECRET_KEY	你的Stripe账户的API密钥。
NEXT_PUBLIC_STRIPE_STUDIO_PRICE	工作室的价格，单位为美分（例如，1000 = $10）。
NEXT_PUBLIC_STUDIO_SHOT_AMOUNT	每个工作室允许的最大拍摄次数。
OPENAI_API_KEY	OpenAI API的密钥，用于提示向导功能。
OPENAI_API_SEED_PROMPT	用于通过OpenAI API生成风格提示的种子提示。

请注意，如果你想使用提供的docker-compose配置，你需要在.env.local中禁用TLS，方法是在文件中添加：

NODE_TLS_REJECT_UNAUTHORIZED = "0"

执行数据库迁移：

yarn prisma:migrate:dev

启动开发服务器：

npm run dev
# 或者
yarn dev

打开浏览器访问 http://localhost:3000，即可看到效果。

Photoshot 快速上手指南

Photoshot 是一个开源的 AI 头像生成 Web 应用，基于 Next.js、Replicate (Flux 模型) 和 Stripe 构建。

环境准备

在开始之前，请确保您的开发环境满足以下要求：

• 操作系统: macOS, Linux 或 Windows (推荐 WSL2)
• Node.js: v18+ (推荐使用 nvm 管理版本)
• 包管理器: Yarn (项目首选) 或 npm
• Docker & Docker Compose: 用于本地运行 PostgreSQL 数据库和邮件服务
• 云服务账号:
  • Replicate: 用于运行 AI 模型训练和推理
  • AWS S3: 用于存储生成的图片
  • Stripe: 用于支付功能（本地测试可配置测试密钥）
  • OpenAI (可选): 用于提示词向导功能

安装步骤

1. 克隆项目并安装依赖

git clone <repository-url>
cd photoshot
yarn install

提示: 如果下载依赖较慢，可临时配置淘宝镜像源： yarn config set registry https://registry.npmmirror.com

2. 启动本地基础设施 (数据库与邮件)

使用 Docker Compose 启动 PostgreSQL 和 Maildev 服务：

docker-compose up -d

• 数据库将运行在 localhost:5432
• 邮件测试界面可访问 http://localhost:1080

3. 配置环境变量

复制示例配置文件：

cp .env.example .env.local

编辑 .env.local 文件，填入您的配置信息。关键变量说明如下：

变量名	说明	默认/示例值 (Docker 环境)
DATABASE_URL	PostgreSQL 连接字符串	postgresql://photoshot:photoshot@localhost:5432/photoshot
NEXTAUTH_URL	应用访问地址	http://localhost:3000
S3_UPLOAD_*	AWS S3 存储配置 (Key, Secret, Bucket, Region)	需填写真实 AWS 信息
REPLICATE_API_TOKEN	Replicate API 令牌	从 Replicate 账户 获取
REPLICATE_USERNAME	Replicate 用户名	您的 Replicate 用户名
SECRET	NextAuth 加密密钥	任意随机字符串 (如 openssl rand -base64 32)
EMAIL_SERVER	SMTP 服务器地址	http://localhost:25 (配合 Docker)
NODE_TLS_REJECT_UNAUTHORIZED	重要: 若使用 Docker 本地邮件服务，必须设为 "0" 以禁用 TLS 验证	"0"

其他变量如 STRIPE_SECRET_KEY 和 OPENAI_API_KEY 可根据需求填写测试密钥或留空（部分功能可能受限）。

4. 执行数据库迁移

初始化数据库表结构：

yarn prisma:migrate:dev

5. 启动开发服务器

yarn dev

基本使用

1. 访问应用: 打开浏览器访问 http://localhost:3000。
2. 创建工作室: 登录后，创建一个新的“工作室”(Studio)，上传几张您的人像照片作为训练数据。
3. 生成头像:
   • 系统会自动调用 Replicate 上的 Flux 模型进行训练（耗时取决于队列和设置）。
   • 训练完成后，输入文本提示词（Prompt），即可生成不同风格的 AI 头像。
4. 查看结果: 生成的图片将保存在配置的 S3 存储桶中，并显示在前端界面上。

现在您已经成功在本地运行了 Photoshot，可以开始探索 AI 头像生成的功能了。