@lobehub/chat 1.120.6 → 1.121.0

package/src/hooks/useGeminiChineseWarning.tsx

@@ -0,0 +1,91 @@
+import { containsChinese } from '@lobechat/utils';
+import { App, Checkbox } from 'antd';
+import React, { useCallback } from 'react';
+import { useTranslation } from 'react-i18next';
+
+import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
+
+const shouldShowChineseWarning = (
+  model: string,
+  prompt: string,
+  hasWarningBeenDismissed: boolean,
+): boolean => {
+  return (
+    model.includes('gemini-2.5-flash-image-preview') &&
+    !hasWarningBeenDismissed &&
+    Boolean(prompt) &&
+    containsChinese(prompt)
+  );
+};
+
+interface UseGeminiChineseWarningOptions {
+  model: string;
+  prompt: string;
+  scenario?: 'chat' | 'image';
+}
+
+export const useGeminiChineseWarning = () => {
+  const { t } = useTranslation('common');
+  const { modal } = App.useApp();
+
+  const [hideGeminiChineseWarning, updateSystemStatus] = useGlobalStore((s) => [
+    systemStatusSelectors.systemStatus(s).hideGemini2_5FlashImagePreviewChineseWarning ?? false,
+    s.updateSystemStatus,
+  ]);
+
+  const checkWarning = useCallback(
+    async ({
+      model,
+      prompt,
+      scenario = 'chat',
+    }: UseGeminiChineseWarningOptions): Promise<boolean> => {
+      if (!shouldShowChineseWarning(model, prompt, hideGeminiChineseWarning)) {
+        return true;
+      }
+
+      return new Promise<boolean>((resolve) => {
+        let doNotShowAgain = false;
+
+        // 根据场景选择不同的按钮文案
+        const continueText =
+          scenario === 'image'
+            ? t('geminiImageChineseWarning.continueGenerate')
+            : t('geminiImageChineseWarning.continueSend');
+
+        modal.confirm({
+          cancelText: t('cancel', { ns: 'common' }),
+          centered: true,
+          content: (
+            <div>
+              <p>{t('geminiImageChineseWarning.content')}</p>
+              <div style={{ marginTop: 16 }}>
+                <Checkbox
+                  onChange={(e) => {
+                    doNotShowAgain = e.target.checked;
+                  }}
+                >
+                  {t('geminiImageChineseWarning.doNotShowAgain')}
+                </Checkbox>
+              </div>
+            </div>
+          ),
+          okText: continueText,
+          onCancel: () => {
+            resolve(false);
+          },
+          onOk: () => {
+            if (doNotShowAgain) {
+              updateSystemStatus({ hideGemini2_5FlashImagePreviewChineseWarning: true });
+            }
+            resolve(true);
+          },
+          title: t('geminiImageChineseWarning.title'),
+        });
+      });
+    },
+    [modal, t, hideGeminiChineseWarning, updateSystemStatus],
+  );
+
+  return checkWarning;
+};

package/src/locales/default/common.ts

@@ -185,6 +185,13 @@ export default {
     title: '喜欢我们的产品？',
   },
   fullscreen: '全屏模式',
+  geminiImageChineseWarning: {
+    content: 'Nano Banana 使用中文有概率性生成图片失败。建议使用英文以获得更好的效果。',
+    continueGenerate: '继续生成',
+    continueSend: '继续发送',
+    doNotShowAgain: '不再提示',
+    title: '中文输入提示',
+  },
   historyRange: '历史范围',
   import: '导入',
   importData: '导入数据',

package/src/store/global/initialState.ts

@@ -50,6 +50,7 @@ export interface SystemStatus {
   // which sessionGroup should expand
   expandSessionGroupKeys: string[];
   filePanelWidth: number;
+  hideGemini2_5FlashImagePreviewChineseWarning?: boolean;
   hidePWAInstaller?: boolean;
   hideThreadLimitAlert?: boolean;
   imagePanelWidth: number;
@@ -108,6 +109,7 @@ export interface GlobalState {
 export const INITIAL_STATUS = {
   expandSessionGroupKeys: [SessionDefaultGroup.Pinned, SessionDefaultGroup.Default],
   filePanelWidth: 320,
+  hideGemini2_5FlashImagePreviewChineseWarning: false,
   hidePWAInstaller: false,
   hideThreadLimitAlert: false,
   imagePanelWidth: 320,