import { NextResponse, NextRequest } from "next/server" import queryString from "query-string" import { predict } from "../../providers/huggingface/predictWithHuggingFace" import { systemPrompt } from "./systemPrompt" export async function GET(req: NextRequest) { const qs = queryString.parseUrl(req.url || "") const query = (qs || {}).query let prompt = "" try { prompt = decodeURIComponent(query?.p?.toString() || "").trim() } catch (err) {} if (!prompt) { return NextResponse.json({ error: 'no prompt provided' }, { status: 400 }); } const userPrompt = `HTML snippet to generate: ${prompt}` const html = await predict({ systemPrompt, userPrompt, nbMaxNewTokens: 400, prefix: "

` return new NextResponse(html, { status: 200, headers: new Headers({ "content-type": "text/html" }), }) }