58 lines
1.5 KiB
TypeScript
58 lines
1.5 KiB
TypeScript
export const runtime = "nodejs";
|
|
|
|
import { type NextRequest, NextResponse } from "next/server";
|
|
import { transcriptAweme } from ".";
|
|
|
|
const runningJobs = new Map<string, Promise<void>>();
|
|
|
|
function startBackgroundAsr(awemeId: string) {
|
|
if (runningJobs.has(awemeId)) return false;
|
|
|
|
const job = transcriptAweme(awemeId, {
|
|
log: (message) => console.log(`[ASR ${awemeId}] ${message}`),
|
|
})
|
|
.then((result) => {
|
|
console.log(
|
|
`[ASR ${awemeId}] done tags=${result.content_tags.join("、")}`,
|
|
);
|
|
})
|
|
.catch((error) => {
|
|
console.error(`[ASR ${awemeId}] failed:`, error);
|
|
})
|
|
.finally(() => {
|
|
runningJobs.delete(awemeId);
|
|
});
|
|
|
|
runningJobs.set(awemeId, job);
|
|
return true;
|
|
}
|
|
|
|
export async function GET(req: NextRequest) {
|
|
const { searchParams } = new URL(req.url);
|
|
const awemeId = searchParams.get("awemeId");
|
|
if (!awemeId) {
|
|
return NextResponse.json(
|
|
{ error: "Missing awemeId parameter" },
|
|
{ status: 400 },
|
|
);
|
|
}
|
|
const asyncMode =
|
|
searchParams.get("async") === "1" || searchParams.get("mode") === "async";
|
|
|
|
if (asyncMode) {
|
|
const started = startBackgroundAsr(awemeId);
|
|
return NextResponse.json(
|
|
{ awemeId, status: started ? "started" : "running" },
|
|
{ status: 202 },
|
|
);
|
|
}
|
|
|
|
try {
|
|
const result = await transcriptAweme(awemeId);
|
|
return NextResponse.json(result);
|
|
} catch (error) {
|
|
console.error("ASR failed:", error);
|
|
return NextResponse.json({ error: "ASR failed" }, { status: 500 });
|
|
}
|
|
}
|