564 lines
15 KiB
TypeScript
564 lines
15 KiB
TypeScript
import Head from "next/head";
|
|
import Image from "next/image";
|
|
import { Inter } from "@next/font/google";
|
|
import styles from "@/styles/Home.module.css";
|
|
import {
|
|
QueryClientProvider,
|
|
QueryClient,
|
|
useQuery,
|
|
useMutation,
|
|
} from "@tanstack/react-query";
|
|
import { useCallback, useEffect, useState } from "react";
|
|
import ReactMarkdown from "react-markdown";
|
|
import { useKey } from "react-use";
|
|
import cn from "clsx";
|
|
// import rehypePrettyCode from "rehype-pretty-code";
|
|
// import * as shiki from "shiki";
|
|
import remarkGfm from "remark-gfm";
|
|
import styled from "styled-components";
|
|
// import rehypeHighlight from "rehype-highlight";
|
|
|
|
// @ts-ignore
|
|
import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
|
|
// @ts-ignore
|
|
import theme from "react-syntax-highlighter/dist/cjs/styles/prism/vs-dark";
|
|
// const CDN_BASE = "https://npm.elemecdn.com/";
|
|
// shiki.setCDN(`${CDN_BASE}/shiki@0.14.1/`);
|
|
|
|
const SMD = styled(ReactMarkdown)`
|
|
blockquote,
|
|
hr,
|
|
p {
|
|
margin-block: 1rem;
|
|
}
|
|
|
|
h1,
|
|
h2,
|
|
h3,
|
|
h4,
|
|
h5,
|
|
h6 {
|
|
font-weight: bold;
|
|
margin-block: 1rem;
|
|
}
|
|
h1 {
|
|
font-size: 3rem;
|
|
color: var(--h1-color);
|
|
}
|
|
h2 {
|
|
font-size: 2.5rem;
|
|
color: var(--h2-color);
|
|
}
|
|
h3 {
|
|
font-size: 2rem;
|
|
color: var(--h3-color);
|
|
}
|
|
h4 {
|
|
font-size: 1.5rem;
|
|
color: var(--h4-color);
|
|
}
|
|
h5 {
|
|
font-size: 1rem;
|
|
color: var(--h5-color);
|
|
}
|
|
h6 {
|
|
font-size: 0.9rem;
|
|
color: var(--h6-color);
|
|
}
|
|
|
|
img {
|
|
display: block;
|
|
margin-left: auto;
|
|
margin-right: auto;
|
|
}
|
|
|
|
.img-alt {
|
|
display: block;
|
|
margin: 0 0 1rem 0;
|
|
font-size: 16px;
|
|
text-align: center;
|
|
}
|
|
|
|
th {
|
|
font-weight: 600;
|
|
}
|
|
|
|
thead {
|
|
border-bottom: 2px solid var(--background-modifier-border);
|
|
}
|
|
|
|
tr {
|
|
line-height: normal;
|
|
padding: 0 4px;
|
|
background-color: var(--pre-code);
|
|
}
|
|
|
|
tr:nth-child(0) {
|
|
padding-top: 4px;
|
|
}
|
|
|
|
th,
|
|
td {
|
|
padding: 0.5em 1em;
|
|
}
|
|
|
|
td {
|
|
border-bottom: 1px solid var(--background-modifier-border);
|
|
}
|
|
td:not(:last-child) {
|
|
border-right: 1px solid var(--background-modifier-border);
|
|
}
|
|
|
|
strong {
|
|
font-weight: 600;
|
|
}
|
|
|
|
a {
|
|
color: var(--text-a);
|
|
text-decoration: none;
|
|
}
|
|
|
|
a:hover {
|
|
color: var(--text-a-hover);
|
|
text-decoration: none;
|
|
}
|
|
|
|
blockquote {
|
|
margin: 1rem 0;
|
|
padding-inline: 2ch;
|
|
padding-block: 0.5rem;
|
|
background-color: var(--pre-code);
|
|
border-left: 0.5ch solid var(--interactive-accent);
|
|
}
|
|
blockquote:has(> blockquote) {
|
|
padding-bottom: 0;
|
|
}
|
|
blockquote > blockquote {
|
|
margin-bottom: 0;
|
|
}
|
|
blockquote:not(:has(> blockquote)) {
|
|
margin-bottom: 1rem;
|
|
}
|
|
|
|
hr {
|
|
background-color: var(--background-modifier-border);
|
|
height: 1px;
|
|
border: 0;
|
|
}
|
|
|
|
ul {
|
|
list-style-type: revert;
|
|
}
|
|
ol {
|
|
list-style-type: decimal;
|
|
}
|
|
ul:not(.contains-task-list),
|
|
ol:not(.contains-task-list) {
|
|
padding-left: 2em;
|
|
}
|
|
ul.contains-task-list,
|
|
ol.contains-task-list {
|
|
margin-left: 0;
|
|
list-style-type: none;
|
|
}
|
|
`;
|
|
|
|
const TextTyper = ({
|
|
// now the phrase, interval and HTML element desired will come via props and we have some default values here
|
|
text = "",
|
|
skip = 0,
|
|
onFinish = () => {},
|
|
}) => {
|
|
const [typedText, setTypedText] = useState("");
|
|
const interval = 50;
|
|
const step = Math.ceil(text.length / 100);
|
|
|
|
// @ts-ignore
|
|
const typingRender = (text, updater, interval) => {
|
|
let localTypingIndex = skip;
|
|
let localTyping = text.slice(0, skip);
|
|
if (text) {
|
|
let printer = setInterval(() => {
|
|
if (localTypingIndex < text.length) {
|
|
updater(
|
|
(localTyping += text.slice(
|
|
localTypingIndex,
|
|
localTypingIndex + step
|
|
))
|
|
);
|
|
localTypingIndex += step;
|
|
document.querySelector("#anchor")?.scrollIntoView();
|
|
} else {
|
|
localTypingIndex = 0;
|
|
localTyping = "";
|
|
clearInterval(printer);
|
|
onFinish();
|
|
}
|
|
}, interval);
|
|
}
|
|
};
|
|
useEffect(() => {
|
|
typingRender(text, setTypedText, interval);
|
|
}, [text, interval]);
|
|
|
|
return <div className="whitespace-pre-wrap">{typedText}</div>;
|
|
};
|
|
|
|
const queryClient = new QueryClient({
|
|
defaultOptions: {
|
|
queries: {
|
|
refetchOnWindowFocus: false,
|
|
},
|
|
},
|
|
});
|
|
const p429 = "请求过多，请稍后再试";
|
|
|
|
const rehypePlugins: any[] = [
|
|
// [
|
|
// rehypePrettyCode,
|
|
// {
|
|
// // theme: {
|
|
// // dark: "github-dark",
|
|
// // light: "github-light",
|
|
// // },
|
|
// theme: "github-dark",
|
|
// keepBackground: true,
|
|
// },
|
|
// ],
|
|
// rehypeHighlight,
|
|
];
|
|
const remarkPlugins = [remarkGfm];
|
|
const components = {
|
|
// @ts-ignore
|
|
code({ node, inline, className, children, ...props }: any) {
|
|
const match = /language-(\w+)/.exec(className || "");
|
|
return !inline && match ? (
|
|
<SyntaxHighlighter
|
|
style={theme}
|
|
language={match[1] || "clike"}
|
|
PreTag="div"
|
|
{...props}
|
|
>
|
|
{String(children).replace(/\n$/, "")}
|
|
</SyntaxHighlighter>
|
|
) : (
|
|
<code className={className} {...props}>
|
|
{children}
|
|
</code>
|
|
);
|
|
},
|
|
};
|
|
const Markdown = ({ children }: { children: string }) => {
|
|
return (
|
|
<SMD
|
|
rehypePlugins={rehypePlugins}
|
|
remarkPlugins={remarkPlugins}
|
|
// components={components}
|
|
>
|
|
{children}
|
|
</SMD>
|
|
);
|
|
};
|
|
|
|
const ask = async (
|
|
query: string,
|
|
context: [string, string][],
|
|
model: string,
|
|
jwt?: string
|
|
) => {
|
|
const res = await fetch(`/ai/ask`, {
|
|
method: "POST",
|
|
headers: {
|
|
Authorization: jwt,
|
|
} as Record<string, string>,
|
|
body: JSON.stringify({
|
|
query: query,
|
|
context: context
|
|
.slice(1)
|
|
.filter((c) => c[1] !== p429)
|
|
.map((c) => ({ query: c[0], answer: c[1] })),
|
|
model,
|
|
}),
|
|
});
|
|
if (res.status === 429) {
|
|
return { code: 1, msg: p429 };
|
|
}
|
|
return (await res.json()) as { code: number; msg: string };
|
|
};
|
|
|
|
let contexts = [
|
|
[
|
|
"hello",
|
|
"您好!我是小Vaala。我可以回答您的问题、写文章、写作业、翻译，对于一些法律等领域的问题我也可以给你提供信息。",
|
|
],
|
|
];
|
|
|
|
const models = ["newbing", "chatvaala", "chatyuan", "chatglm"];
|
|
|
|
function Home() {
|
|
const [model, setModel] = useState(models[0]);
|
|
const [query, setQuery] = useState("");
|
|
const [question, setQuestion] = useState("");
|
|
const [answer, setAnswer] = useState("");
|
|
const [context, setContext] = useState<[string, string][]>(
|
|
contexts as [string, string][]
|
|
);
|
|
const { data: jwt } = useQuery(["jwt"], async () => {
|
|
const resp = await fetch(`/auth/jwt`);
|
|
const res = (await resp.json())["authorization"];
|
|
return res;
|
|
});
|
|
useKey("Enter", (e) => {
|
|
if (!e.shiftKey && !e.metaKey && !e.ctrlKey && !e.isComposing) {
|
|
e.preventDefault();
|
|
handleAsk.mutate();
|
|
}
|
|
});
|
|
const handleTypeFinish = useCallback(() => {
|
|
setQuestion("");
|
|
setContext((v) => [...v, [question, answer]]);
|
|
setAnswer("");
|
|
document.querySelector("#anchor")?.scrollIntoView();
|
|
}, [query, answer]);
|
|
|
|
const handleAsk = useMutation(async () => {
|
|
if (query.length === 0) return;
|
|
setQuestion(query);
|
|
setQuery("");
|
|
|
|
setTimeout(() => document.querySelector("#anchor")?.scrollIntoView(), 100);
|
|
(async () => {
|
|
const ans = await ask(query, context, model, jwt || "").finally(() => {
|
|
setTimeout(
|
|
() => document.querySelector("#anchor")?.scrollIntoView(),
|
|
300
|
|
);
|
|
});
|
|
if (ans.code) alert(ans.msg);
|
|
else setAnswer(ans.msg);
|
|
})();
|
|
});
|
|
return (
|
|
<>
|
|
<Head>
|
|
<title>Vaala Chat</title>
|
|
<meta name="description" content="Chat to small vaala" />
|
|
<meta name="viewport" content="width=device-width, initial-scale=1" />
|
|
<link rel="icon" href="/favicon.ico" />
|
|
<style
|
|
dangerouslySetInnerHTML={{
|
|
__html: `.lds-dual-ring {
|
|
display: inline-block;
|
|
width: 18px;
|
|
height: 18px;
|
|
}
|
|
.lds-dual-ring:after {
|
|
content: " ";
|
|
display: block;
|
|
width: 18px;
|
|
height: 18px;
|
|
margin: 0 0 0 8px;
|
|
border-radius: 50%;
|
|
border: 2px solid #fff;
|
|
border-color: #fff transparent #fff transparent;
|
|
animation: lds-dual-ring 1.2s linear infinite;
|
|
}
|
|
@keyframes lds-dual-ring {
|
|
0% {
|
|
transform: rotate(0deg);
|
|
}
|
|
100% {
|
|
transform: rotate(360deg);
|
|
}
|
|
}
|
|
pre > code {
|
|
white-space: pre-wrap;
|
|
}
|
|
`,
|
|
}}
|
|
/>
|
|
</Head>
|
|
<main
|
|
className={cn(styles.main, "px-2 md:px-16 xl:px-64")}
|
|
style={{ width: "100dvw", height: "100dvh" }}
|
|
>
|
|
<div className="flex justify-end items-center w-full h-12">
|
|
<label
|
|
htmlFor="model"
|
|
className="mr-2 block text-sm font-medium text-gray-900 dark:text-white"
|
|
>
|
|
Model:
|
|
</label>
|
|
<select
|
|
id="model"
|
|
style={{
|
|
// @ts-ignore
|
|
"-webkit-appearance": "none",
|
|
"-moz-appearance": "none",
|
|
appearance: "none",
|
|
backgroundImage: `url("data:image/svg+xml;utf8,<svg xmlns='http://www.w3.org/2000/svg' width='100' height='100' fill='%238C98F2'><polygon points='0,0 100,0 50,50'/></svg>")`,
|
|
backgroundSize: "14px",
|
|
backgroundPosition: "calc(100% - 10px) 9px",
|
|
backgroundRepeat: "no-repeat",
|
|
}}
|
|
className="w-48 bg-gray-50 border border-gray-300 text-gray-900 text-sm focus:ring-blue-500 focus:border-blue-500 block w-full py-1 pl-2 dark:bg-gray-700 dark:border-gray-600 dark:placeholder-gray-400 dark:text-white dark:focus:ring-blue-500 dark:focus:border-blue-500"
|
|
onChange={(e) => setModel(e.target.value)}
|
|
>
|
|
{models.map((m) => (
|
|
<option key={m} value={m}>
|
|
{m}
|
|
</option>
|
|
))}
|
|
</select>
|
|
</div>
|
|
<div
|
|
style={{
|
|
maxHeight: "calc(100dvh - 123px)",
|
|
overflowY: "auto",
|
|
}}
|
|
>
|
|
{context?.map((qa, i) => {
|
|
return (
|
|
<div
|
|
key={i}
|
|
className="flex w-full flex-col items-stretch justify-start"
|
|
>
|
|
<div className="p-4 w-full bg-white dark:bg-slate-800">
|
|
<Markdown>{`ME: ${qa[0]}`}</Markdown>
|
|
</div>
|
|
<div className="p-4 w-full bg-slate-50 dark:bg-slate-900">
|
|
{qa[1] === p429 ? (
|
|
<span className="text-red-500">{qa[1]}</span>
|
|
) : (
|
|
<Markdown
|
|
// remarkPlugins={[remarkGfm]}
|
|
>
|
|
{`🐳: ${qa[1]}`}
|
|
</Markdown>
|
|
)}
|
|
</div>
|
|
</div>
|
|
);
|
|
})}
|
|
{question || answer ? (
|
|
<>
|
|
<div className="p-4 w-full bg-white dark:bg-slate-800">
|
|
<Markdown>{`ME: ${question}`}</Markdown>
|
|
</div>
|
|
<div
|
|
className="p-4 w-full flex items-center bg-slate-50 dark:bg-slate-900"
|
|
style={{ lineHeight: "18px" }}
|
|
>
|
|
{answer ? (
|
|
<TextTyper
|
|
skip={3}
|
|
text={`🐳: ${answer}`}
|
|
onFinish={handleTypeFinish}
|
|
/>
|
|
) : (
|
|
<>
|
|
<div>{`🐳: 思考中`}</div>
|
|
<svg
|
|
// xmlns="http://www.w3.org/2000/svg"
|
|
// xmlns:xlink="http://www.w3.org/1999/xlink"
|
|
// style="margin: auto; background: rgb(241, 242, 243); display: block;"
|
|
width="18px"
|
|
height="18px"
|
|
viewBox="0 0 100 100"
|
|
preserveAspectRatio="xMidYMid"
|
|
>
|
|
<circle cx="27.5" cy="57.5" r="5" fill="#85a2b6">
|
|
<animate
|
|
attributeName="cy"
|
|
calcMode="spline"
|
|
keySplines="0 0.5 0.5 1;0.5 0 1 0.5;0.5 0.5 0.5 0.5"
|
|
repeatCount="indefinite"
|
|
values="57.5;42.5;57.5;57.5"
|
|
keyTimes="0;0.3;0.6;1"
|
|
dur="1s"
|
|
begin="-0.6s"
|
|
></animate>
|
|
</circle>{" "}
|
|
<circle cx="42.5" cy="57.5" r="5" fill="#bbcedd">
|
|
<animate
|
|
attributeName="cy"
|
|
calcMode="spline"
|
|
keySplines="0 0.5 0.5 1;0.5 0 1 0.5;0.5 0.5 0.5 0.5"
|
|
repeatCount="indefinite"
|
|
values="57.5;42.5;57.5;57.5"
|
|
keyTimes="0;0.3;0.6;1"
|
|
dur="1s"
|
|
begin="-0.44999999999999996s"
|
|
></animate>
|
|
</circle>{" "}
|
|
<circle cx="57.5" cy="57.5" r="5" fill="#dce4eb">
|
|
<animate
|
|
attributeName="cy"
|
|
calcMode="spline"
|
|
keySplines="0 0.5 0.5 1;0.5 0 1 0.5;0.5 0.5 0.5 0.5"
|
|
repeatCount="indefinite"
|
|
values="57.5;42.5;57.5;57.5"
|
|
keyTimes="0;0.3;0.6;1"
|
|
dur="1s"
|
|
begin="-0.3s"
|
|
></animate>
|
|
</circle>{" "}
|
|
<circle cx="72.5" cy="57.5" r="5" fill="#fdfdfd">
|
|
<animate
|
|
attributeName="cy"
|
|
calcMode="spline"
|
|
keySplines="0 0.5 0.5 1;0.5 0 1 0.5;0.5 0.5 0.5 0.5"
|
|
repeatCount="indefinite"
|
|
values="57.5;42.5;57.5;57.5"
|
|
keyTimes="0;0.3;0.6;1"
|
|
dur="1s"
|
|
begin="-0.15s"
|
|
></animate>
|
|
</circle>
|
|
</svg>
|
|
</>
|
|
)}
|
|
</div>
|
|
</>
|
|
) : null}
|
|
<div id="anchor" className="w-full h-1"></div>
|
|
</div>
|
|
<div
|
|
className="fixed bottom-0 p-4 w-full flex justify-center items-center bg-gray-50 dark:bg-gray-900 border-t border-solid border-sky-400"
|
|
style={{ width: "100vw" }}
|
|
>
|
|
<textarea
|
|
rows={1}
|
|
className="flex-1 border-sky-500 border border-solid focus:border-sky-300 focus:outline-none rounded p-2"
|
|
placeholder="..."
|
|
value={query}
|
|
onChange={(e) => setQuery(e.target.value)}
|
|
></textarea>
|
|
<button
|
|
type="submit"
|
|
className="text-white bg-sky-500 border-sky-500 border border-solid rounded h-12 p-2 ml-4 hover:border-sky-600 hover:bg-sky-600"
|
|
onClick={() => handleAsk.mutate()}
|
|
onSubmit={() => handleAsk.mutate()}
|
|
>
|
|
Send
|
|
</button>
|
|
<button
|
|
className="text-white bg-sky-500 border-sky-500 border border-solid rounded h-12 p-2 ml-4 hover:border-sky-600 hover:bg-sky-600"
|
|
onClick={() => setContext(contexts as any)}
|
|
>
|
|
Clear
|
|
</button>
|
|
</div>
|
|
</main>
|
|
</>
|
|
);
|
|
}
|
|
|
|
export default function Wrapper() {
|
|
return (
|
|
<QueryClientProvider client={queryClient}>
|
|
<Home />
|
|
</QueryClientProvider>
|
|
);
|
|
}
|