Function Home

pages/index.tsx:21–409

()

import { Template } from "../utils/templateStorage";

const Home: NextPage = () => {
  const t = useTranslations('Index')

  const [loading, setLoading] = useState(false);
  const [chat, setChat] = useState("");
  const [form, setForm] = useState<FormType>("paragraphForm");
  const [api_key, setAPIKey] = useState("")
  const [generatedChat, setGeneratedChat] = useState<String>("");
  const [isFromHistory, setIsFromHistory] = useState(false);
  const [showMobileHistory, setShowMobileHistory] = useState(false);
  const [historyCount, setHistoryCount] = useState(0);
  const [historyRefreshTrigger, setHistoryRefreshTrigger] = useState(0);
  const [showTemplateSelect, setShowTemplateSelect] = useState(false);
  const [showTemplateManager, setShowTemplateManager] = useState(false);

  // 更新历史记录数量和触发刷新
  const updateHistoryCount = () => {
    setHistoryCount(HistoryStorage.getHistory().length);
  };

  const triggerHistoryRefresh = () => {
    setHistoryRefreshTrigger(prev => prev + 1);
    updateHistoryCount();
  };

  // 组件挂载时加载历史记录数量
  useEffect(() => {
    updateHistoryCount();
  }, []);

  console.log("Streamed response: ", generatedChat);

  const prompt =
    form === 'paragraphForm'?
      `${chat}`
      : `${chat}`;

  const useUserKey = process.env.NEXT_PUBLIC_USE_USER_KEY === "true" ? true : false;

  const generateChat = async (e: any) => {
    e.preventDefault();
    setGeneratedChat("");
    setLoading(true);
    if (useUserKey && api_key == ""){
      toast.error(t("API_KEY_NULL_ERROR"))
      setLoading(false)
      return
    }
    if (chat == ""){
      toast.error(t("CONTENT_NULL_ERROR"))
      setLoading(false)
      return
    }
    const response = useUserKey ?
      await fetch("/api/generate", {
        method: "POST",
        headers: {
          "Content-Type": "application/json",