lisichatbot 2.0.1 → 2.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "lisichatbot",
-  "version": "2.0.1",
+  "version": "2.0.2",
   "type": "module",
   "main": "./src/index.js",
   "exports": {

package/src/index.js

@@ -213,16 +213,21 @@ function updateEditIcons() {
                 return;
             }
             
-            if (chatState.completed) {
+            // ✅ FIX: In edit mode, keep edit icons visible even after completion
+            if (chatState.completed && chatState.chatMode !== 'edit') {
                 editIcon.style.setProperty('display', 'none', 'important');
-                console.log(`  🏁 Step ${stepNumber}: Icon HIDDEN (flow completed)`);
+                console.log(`  🏁 Step ${stepNumber}: Icon HIDDEN (flow completed in create mode)`);
                 return;
             }
             
             // ✅ NEW: Check if edit icon should be forced to show (even without input)
             const forceShowEdit = stepData && stepData.showEditIcon === true;
             
-            if ((hasInput || forceShowEdit) && stepNumber < chatState.step && isLatest) {
+            // ✅ FIX: In edit mode, show edit icons for all completed steps (not just previous)
+            const shouldShowInEditMode = chatState.chatMode === 'edit' && chatState.completed && isLatest;
+            const shouldShowNormally = stepNumber < chatState.step && isLatest;
+            
+            if ((hasInput || forceShowEdit) && (shouldShowInEditMode || shouldShowNormally)) {
                 editIcon.onclick = (e) => {
                     e.stopPropagation();
                     e.preventDefault();