forked from ChatGPTNextWeb/ChatGPT-Next-Web
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request ChatGPTNextWeb#4983 from OpenAI-Next/dev-sd
[Feature] Stable Diffusion
- Loading branch information
Showing
35 changed files
with
1,714 additions
and
140 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -44,3 +44,5 @@ dev | |
|
||
*.key | ||
*.key.pub | ||
|
||
masks.json |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,104 @@ | ||
import { NextRequest, NextResponse } from "next/server"; | ||
import { getServerSideConfig } from "@/app/config/server"; | ||
import { ModelProvider, STABILITY_BASE_URL } from "@/app/constant"; | ||
import { auth } from "@/app/api/auth"; | ||
|
||
async function handle( | ||
req: NextRequest, | ||
{ params }: { params: { path: string[] } }, | ||
) { | ||
console.log("[Stability] params ", params); | ||
|
||
if (req.method === "OPTIONS") { | ||
return NextResponse.json({ body: "OK" }, { status: 200 }); | ||
} | ||
|
||
const controller = new AbortController(); | ||
|
||
const serverConfig = getServerSideConfig(); | ||
|
||
let baseUrl = serverConfig.stabilityUrl || STABILITY_BASE_URL; | ||
|
||
if (!baseUrl.startsWith("http")) { | ||
baseUrl = `https://${baseUrl}`; | ||
} | ||
|
||
if (baseUrl.endsWith("/")) { | ||
baseUrl = baseUrl.slice(0, -1); | ||
} | ||
|
||
let path = `${req.nextUrl.pathname}`.replaceAll("/api/stability/", ""); | ||
|
||
console.log("[Stability Proxy] ", path); | ||
console.log("[Stability Base Url]", baseUrl); | ||
|
||
const timeoutId = setTimeout( | ||
() => { | ||
controller.abort(); | ||
}, | ||
10 * 60 * 1000, | ||
); | ||
|
||
const authResult = auth(req, ModelProvider.Stability); | ||
|
||
if (authResult.error) { | ||
return NextResponse.json(authResult, { | ||
status: 401, | ||
}); | ||
} | ||
|
||
const bearToken = req.headers.get("Authorization") ?? ""; | ||
const token = bearToken.trim().replaceAll("Bearer ", "").trim(); | ||
|
||
const key = token ? token : serverConfig.stabilityApiKey; | ||
|
||
if (!key) { | ||
return NextResponse.json( | ||
{ | ||
error: true, | ||
message: `missing STABILITY_API_KEY in server env vars`, | ||
}, | ||
{ | ||
status: 401, | ||
}, | ||
); | ||
} | ||
|
||
const fetchUrl = `${baseUrl}/${path}`; | ||
console.log("[Stability Url] ", fetchUrl); | ||
const fetchOptions: RequestInit = { | ||
headers: { | ||
"Content-Type": req.headers.get("Content-Type") || "multipart/form-data", | ||
Accept: req.headers.get("Accept") || "application/json", | ||
Authorization: `Bearer ${key}`, | ||
}, | ||
method: req.method, | ||
body: req.body, | ||
// to fix #2485: https://stackoverflow.com/questions/55920957/cloudflare-worker-typeerror-one-time-use-body | ||
redirect: "manual", | ||
// @ts-ignore | ||
duplex: "half", | ||
signal: controller.signal, | ||
}; | ||
|
||
try { | ||
const res = await fetch(fetchUrl, fetchOptions); | ||
// to prevent browser prompt for credentials | ||
const newHeaders = new Headers(res.headers); | ||
newHeaders.delete("www-authenticate"); | ||
// to disable nginx buffering | ||
newHeaders.set("X-Accel-Buffering", "no"); | ||
return new Response(res.body, { | ||
status: res.status, | ||
statusText: res.statusText, | ||
headers: newHeaders, | ||
}); | ||
} finally { | ||
clearTimeout(timeoutId); | ||
} | ||
} | ||
|
||
export const GET = handle; | ||
export const POST = handle; | ||
|
||
export const runtime = "edge"; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
export * from "./sd"; | ||
export * from "./sd-panel"; |
Oops, something went wrong.