From 0742dd791db33121472d269f9a691a53cd5ac8a4 Mon Sep 17 00:00:00 2001
From: chris <kokosias@yahoo.gr>
Date: Thu, 15 Feb 2024 19:50:33 +0200
Subject: [PATCH] fix: reset active state

---
 .../src/AiService/components/ToggleAiComponent.js         | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js b/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js
index 9e6ca37ed..6575a79f9 100644
--- a/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js
+++ b/wax-prosemirror-services/src/AiService/components/ToggleAiComponent.js
@@ -1,4 +1,4 @@
-import React, { useContext, useMemo, useState } from 'react';
+import React, { useContext, useEffect, useMemo, useState } from 'react';
 import { WaxContext, MenuButton } from 'wax-prosemirror-core';
 import PropTypes from 'prop-types';
 
@@ -26,6 +26,12 @@ const ToggleAiComponent = ({ item }) => {
     main.focus();
   };
 
+  useEffect(() => {
+    setChecked(false);
+    context.setOption({ AiOn: false });
+    main.dispatch(main.state.tr.setMeta('addToHistory', false));
+  }, [checked && main.state.selection.from === main.state.selection.to]);
+
   return useMemo(
     () =>
       enableService.AiOn ? (
-- 
GitLab