ChatGPT/src/hooks/useChatModel.ts

import { useState } from 'react';
import { clone } from 'lodash';

import { CHAT_MODEL_JSON, readJSON, writeJSON } from '@/utils';
import useInit from '@/hooks/useInit';

export default function useChatModel() {
  const [modelJson, setModelJson] = useState<Record<string, any>>({});

  useInit(async () => {
    const data = await readJSON(CHAT_MODEL_JSON, { name: 'ChatGPT Model', data: [] });
    setModelJson(data);
  });

  const modelSet = async (data: Record<string, any>[]) => {
    const oData = clone(modelJson);
    oData.data = data;
    await writeJSON(CHAT_MODEL_JSON, oData);
    setModelJson(oData);
  }

  return { modelJson, modelSet, modelData: modelJson?.data || [] }
}