mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-08-03 12:30:24 +08:00
60 lines
1.9 KiB
TypeScript
60 lines
1.9 KiB
TypeScript
import { Flex } from "@chakra-ui/react";
|
|
import Head from "next/head";
|
|
import { useEffect, useMemo, useState } from "react";
|
|
import { useTranslation } from "react-i18next";
|
|
import { LeaderboardTable, TaskOption, WelcomeCard } from "src/components/Dashboard";
|
|
import { getDashboardLayout } from "src/components/Layout";
|
|
import { TaskCategory } from "src/components/Tasks/TaskTypes";
|
|
import { get } from "src/lib/api";
|
|
import { AvailableTasks, TaskType } from "src/types/Task";
|
|
export { getDefaultStaticProps as getStaticProps } from "src/lib/default_static_props";
|
|
import useSWR from "swr";
|
|
|
|
const Dashboard = () => {
|
|
const {
|
|
i18n: { language },
|
|
} = useTranslation();
|
|
const [activeLang, setLang] = useState<string>(null);
|
|
const { data, mutate: fetchTasks } = useSWR<AvailableTasks>("/api/available_tasks", get, {
|
|
refreshInterval: 2 * 60 * 1000, //2 minutes
|
|
revalidateOnMount: false, // triggered in the hook below
|
|
});
|
|
|
|
useEffect(() => {
|
|
// re-fetch tasks if the language has changed
|
|
if (activeLang !== language) {
|
|
setLang(language);
|
|
fetchTasks();
|
|
}
|
|
}, [activeLang, setLang, language, fetchTasks]);
|
|
|
|
const availableTaskTypes = useMemo(() => {
|
|
const taskTypes = filterAvailableTasks(data ?? {});
|
|
return { [TaskCategory.Random]: taskTypes };
|
|
}, [data]);
|
|
|
|
return (
|
|
<>
|
|
<Head>
|
|
<title>Dashboard - Open Assistant</title>
|
|
<meta name="description" content="Chat with Open Assistant and provide feedback." />
|
|
</Head>
|
|
<Flex direction="column" gap="10">
|
|
<WelcomeCard />
|
|
<TaskOption content={availableTaskTypes} />
|
|
<LeaderboardTable />
|
|
</Flex>
|
|
</>
|
|
);
|
|
};
|
|
|
|
Dashboard.getLayout = getDashboardLayout;
|
|
|
|
export default Dashboard;
|
|
|
|
const filterAvailableTasks = (availableTasks: Partial<AvailableTasks>) =>
|
|
Object.entries(availableTasks)
|
|
.filter(([_, count]) => count > 0)
|
|
.sort((a, b) => b[1] - a[1])
|
|
.map(([taskType]) => taskType) as TaskType[];
|