summaryrefslogtreecommitdiff
path: root/src/pages/api/nlp.ts
blob: 27c330ddcddaee4388d5bf20678503435b282ab8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// import db from "../../lib/db";
import { z } from "zod";
import { NLP } from "sortug-ai";

const schema = z.object({
  app: z.enum(["stanza", "spacy"]),
  text: z.string().min(3, "minimum 3 characters"),
  lang: z
    .custom<string>((val) => {
      const check = NLP.ISO.BCP47.parse(val);
      if (!check.language) return false;
      const twochars = Object.values(NLP.ISO.iso6393To1);
      return twochars.includes(check.language);
    })
    .optional(),
});

export const POST = async (request: Request): Promise<Response> => {
  const bod = await request.json();
  const { app, text, lang } = await schema.parseAsync(bod);

  try {
    const res =
      app === "stanza"
        ? NLP.Stanza.segmenter(text, lang)
        : NLP.Spacy.run(text, lang);
    const r = await res;
    return Response.json(r, { status: 200 });
  } catch (error) {
    return Response.json({ message: "Failure" }, { status: 500 });
  }
};

type AnalyzeRes = {
  word: string;
  syllables: string[];
  ipa: string;
  pos: string;
};

export async function thaiData(word: string): Promise<AnalyzeRes[]> {
  const [head, tail] = await Promise.all([
    analyzeTHWord(word),
    segmentateThai(word),
  ]);
  return [head, ...tail];
}

export async function analyzeTHWord(word: string): Promise<AnalyzeRes> {
  const opts = {
    method: "POST",
    headers: { "Content-type": "application/json" },
    body: JSON.stringify({ word }),
  };
  const r1 = await fetch("http://localhost:8001" + "/analyze", opts);
  // const r2 = await fetch(`http://192.168.1.110:8000/analyze`, opts);
  const jj = await r1.json();
  return jj;
}
export async function segmentateThai(sentence: string): Promise<AnalyzeRes[]> {
  const opts = {
    method: "POST",
    headers: { "Content-type": "application/json" },
    body: JSON.stringify({ word: sentence }),
  };
  // const r1 = await fetch(`http://localhost:8000/segmentate`, opts);
  const r2 = await fetch("http://localhost:8001" + `/segmentate`, opts);
  const jj = await r2.json();
  return jj;
}