Function useRunAiWithStore

app/src/lib/usePromptStore.ts:130–235

()

 * Shows paywall if the user has exceeded their limit.
 */
export function useRunAiWithStore() {
  const hasProAccess = useHasProAccess();
  const customer = useContext(AppContext).customer;
  const sid = customer?.subscription?.id;
  const [abortController, setAbortController] =
    useState<AbortController | null>(null);

  const handleError = useCallback(
    (error: Error) => {
      if (error.name === "AbortError") {
        setError(t`Operation canceled`);
      } else if (!hasProAccess && error.message === RATE_LIMIT_EXCEEDED) {
        showPaywall({
          title: t`Get Unlimited AI Requests`,
          content: t`You've used all your free AI conversions. Upgrade to Pro for unlimited AI use, custom themes, private sharing, and more. Keep creating amazing flowcharts effortlessly!`,
          movieUrl: "/images/ai-convert.mp4",
          toPricingCode: "ConvertToFlowchart",
          buttonText: t`Unlock Unlimited AI Flowcharts`,
        });
      } else {
        if (error.message === RATE_LIMIT_EXCEEDED) {
          setError(t`Rate limit exceeded. Please try again later.`);
        } else {
          setError(error.message);
        }
      }
    },
    [hasProAccess]
  );

  const runAiCallback = useCallback(() => {
    const store = usePromptStore.getState();
    if (store.isRunning) return;

    // Snapshot BEFORE any state mutations (unfreezeDoc, loadTemplate, etc.)
    const { text: snapshotText, meta: snapshotMeta } = useDoc.getState();
    const metaCopy = JSON.parse(JSON.stringify(snapshotMeta));

    setIsOpen(false);
    startConvert();

    if (store.mode === "convert" || store.mode === "prompt") {
      unfreezeDoc();
    }

    const newAbortController = new AbortController();
    setAbortController(newAbortController);

    runAi({
      endpoint: store.mode,
      prompt: store.currentText,
      sid,
      signal: newAbortController.signal,
    })
      .catch(handleError)
      .then((result) => {
        if (result) {
          const text = repairText(result);