@sapui5/sap.suite.ui.generic.template 1.147.2 → 1.148.1

package/src/sap/suite/ui/generic/template/lib/ai/EasyFill/EasyFillHandler.js

@@ -1,482 +1,408 @@
 sap.ui.define([
     "sap/ui/base/Object",
     "sap/base/util/extend",
-    "sap/ui/comp/smartform/Group",
-    "sap/ui/comp/smartform/GroupElement",
-    "sap/ui/comp/smartfield/SmartField",
-    "sap/ui/comp/smartfield/SmartLabel",
-    "sap/m/MessageToast",
-    "sap/ui/model/json/JSONModel",
-	"sap/ui/model/odata/AnnotationHelper"
-], function(BaseObject, extend, Group, GroupElement, SmartField, SmartLabel,MessageToast,JSONModel,AnnotationHelperModel) {
-    'use strict';
-    /*
-     * This is a handler class for EasyFill.
-     * It gets initialized when the Fiori AI EasyFill  is enabled
-     * Below things are handled in this class:
-     * 1. Prepare Field metadata
-     * 2. Prepare EasyFill Dialog
+    "sap/suite/ui/generic/template/library",
+    "sap/suite/ui/generic/template/genericUtilities/FeError",
+    "sap/suite/ui/generic/template/lib/ai/EasyFill/EasyFillFieldCollector",
+    "sap/suite/ui/generic/template/lib/ai/EasyFill/EasyFillAIOrchestrator",
+    "sap/suite/ui/generic/template/lib/ai/EasyFill/EasyFillFormRenderer",
+    "sap/suite/ui/generic/template/lib/ai/EasyFill/EasyFillTableRenderer",
+    "sap/suite/ui/generic/template/lib/ai/EasyFill/EasyFillApplyHandler",
+    "sap/suite/ui/generic/template/lib/ai/EasyFill/EasyFillDialogController",
+    "sap/suite/ui/generic/template/js/AnnotationHelper"
+], function(
+    BaseObject,
+    extend,
+    Library,
+    FeError,
+    EasyFillFieldCollector,
+    EasyFillAIOrchestrator,
+    EasyFillFormRenderer,
+    EasyFillTableRenderer,
+    EasyFillApplyHandler,
+    EasyFillDialogController,
+    GenericAnnotationHelper
+) {
+    "use strict";
+
+    /**
+     * EasyFillHandler
      *
+     * Thin orchestrator for the EasyFill AI feature on the ObjectPage.
+     *
+     * This class wires all EasyFill sub-modules together and exposes a single public
+     * entry point (onEasyFillButtonClick) to the ObjectPage ControllerImplementation.
+     *
+     * Sub-module responsibilities:
+     *   EasyFillFieldCollector   — OData metadata field/section extraction
+     *   EasyFillAIOrchestrator   — AI service call and response post-processing
+     *   EasyFillFormRenderer     — SmartForm/GroupElement/SmartField population
+     *   EasyFillTableRenderer    — SmartTable collection preview rendering
+     *   EasyFillApplyHandler     — Confirm/save, validation, reset logic
+     *   EasyFillDialogController — Dialog lifecycle, formatters, UI event handlers
+     *
+     * Shared mutable state lives here and is passed by reference into sub-modules.
+     * Sub-modules are stateless — they never store state themselves.
      */
     function getMethods(oState, oController, oTemplateUtils, oObjectPage) {
-        let oTransientContextForEasyFill;
-        let oObjectPageContext;
-        let oObjectPageModel;
-        let mUpdatableLookupMap;
-        let oRb;
-        let oAIPopover;
-        let aEditableFields = [];
-
-        async function openEasyFillDialog(oEvent,oContext) {
-            let oEasyFillDialog;
+        var EasyFillLayoutMode = Library.EasyFillLayoutMode;
+
+        // ─── Shared Mutable State ────────────────────────────────────────────────
+        // All state is declared here and passed into sub-modules as needed.
+        // Sub-modules must not cache these values — always use the latest reference.
+
+        var oTransientContextForEasyFill = null; // Transient OData context for staged AI scalar values
+        var oObjectPageContext = null;            // Live ObjectPage binding context
+        var oObjectPageModel = null;              // OData model of the ObjectPage
+        var mUpdatableLookupMap = {};             // { fieldName: value } for updatable fields only
+        var oRb = null;                           // i18n resource bundle
+        var oAIPopoverRef = { popover: null };    // Mutable ref for lazy-loaded AINotice popover
+
+        // Collection state — tracks OData row contexts and property metadata for SmartTable collections
+        var mCollectionState = {
+            mCollectionRowContexts: Object.create(null),
+            mCollectionRowContextsByPath: Object.create(null),
+            mCollectionProperties: Object.create(null),
+            mCollectionEntityTypes: Object.create(null)
+        };
+
+        var mCollectionUpdates = Object.create(null); // { navProperty: [ rowUpdate ] } from AI response
+        var oTableStagingState = {
+            mStagedRows: Object.create(null)
+        };
+
+        // ─── Internal Helpers ────────────────────────────────────────────────────
+
+        /**
+         * Opens the EasyFill dialog for the given ObjectPage context.
+         * Initializes the dialog model on first open, then loads and opens the dialog fragment
+         * with all event handlers and formatters bound.
+         *
+         * @param {sap.ui.base.Event} oEvent - the button click event
+         * @param {sap.ui.model.Context} [oContext] - optional context override (used when OP just entered edit mode)
+         */
+        async function openEasyFillDialog(oEvent, oContext) {
             oRb = oController.getView().getModel("i18n").getResourceBundle();
-            const bIsModelCreated = !!oController.getView().getModel("easyFillDialogModel");
             oObjectPageModel = oObjectPage.getModel();
-            if (!bIsModelCreated) {
-                const aDeferredGroups = oObjectPageModel.getDeferredGroups();
-                //Adding a defferred group will make sure that no batch call would get triggered to backend until submitChanges() is invoked
-                aDeferredGroups.push("EasyFillFETransientChanges");
-                oObjectPageModel.setDeferredGroups(aDeferredGroups);
-
-                //Creating a JSON model for the dialog
-
-                const oData = {
-                    isBusy: false,
-                    isUpdatableVisible: false,
-                    isNonUpdatableVisible: false,
-                    isIllustrationVisible: true,
-                    isSaveEnabled: false,
-                    stateType: "Initial",
-                    feedbackEnabled: true,
-                    feedbackPressedUp: false,
-                    feedbackPressedDown: false,
-                    isEasyFillButtonEnabled: false
-                  };
-
-                  let oModel = new JSONModel(oData);
-                  oController.getView().setModel(oModel, "easyFillDialogModel");
-            }
 
-            oObjectPageContext = (oContext) ? oContext : oObjectPage.getBindingContext();
+            // Initialize dialog model and deferred group once per view lifetime
+            EasyFillDialogController.initDialogModel(oController, oObjectPageModel, EasyFillLayoutMode);
 
-            // cannot use OP path to create a list binding as it is a individual entry
-            var sFragmentname = "sap.suite.ui.generic.template.lib.ai.EasyFill.fragments.EasyFillDialog";
-			oTemplateUtils.oCommonUtils.getDialogFragmentAsync(sFragmentname, {
-                onFeedbackPress: function() {
-                    const easyFillDialogModel = oController.getView().getModel("easyFillDialogModel");
-                    easyFillDialogModel.setProperty("/feedbackEnabled",false);
-                    MessageToast.show(oRb.getText("EASYFILL_RESULT_FEEDBACK"));
-                },
-                formatIllustrationType: function (sState) {
-                    switch (sState) {
-                        case "Initial":
-                            return "sapIllus-NoSearchResults";
-                        case "NoEntries":
-                            return "sapIllus-NoEntries";
-                        case "Error":
-                            return "sapIllus-UnableToLoad";
-                        default:
-                            return "sapIllus-NoSearchResults";
-                    }
-                },
-                formatIllustrationTitle: function (sState) {
-                    switch (sState) {
-                        case "Initial":
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_TITLE_INITIAL");
-                        case "NoEntries":
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_TITLE_NO_ENTRIES");
-                        case "Error":
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_TITLE_ERROR");
-                        default:
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_TITLE_INITIAL");
-                    }
-                },
-                formatIllustrationDescription: function (sState) {
-                    switch (sState) {
-                        case "Initial":
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_DESCRIPTION_INITIAL");
-                        case "NoEntries":
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_DESCRIPTION_NO_ENTRIES");
-                        case "Error":
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_DESCRIPTION_ERROR");
-                        default:
-                            return oRb.getText("EASY_FILL_ILLUSTRATION_DESCRIPTION_INITIAL");
-                    }
+            // Use the provided context override or fall back to the current OP binding context
+            oObjectPageContext = oContext || oObjectPage.getBindingContext();
+
+            // Build the handler/formatter map that the dialog fragment binds to
+            var oHandlers = _buildDialogHandlers();
+
+            await EasyFillDialogController.openDialog(oController, oTemplateUtils, oHandlers);
+        }
+
+        /**
+         * Builds the event handler and formatter map passed to the EasyFillDialog fragment.
+         * Each entry corresponds to a function called from the fragment via event binding or formatter.
+         *
+         * @returns {object} handler/formatter map
+         */
+        function _buildDialogHandlers() {
+            return {
+                // ── Layout mode toggle (Condensed ↔ ObjectPage view) ──────────────
+                onSelectionChange: function(oEvent) {
+                    EasyFillDialogController.onSelectionChange(
+                        oEvent, oController, oTemplateUtils, oRb,
+                        EasyFillFieldCollector, EasyFillLayoutMode
+                    );
                 },
-                onSaveFromEasyFillDialog : async function (oEvent) {
-                    var oTransientObject = oTransientContextForEasyFill.getObject();
-                    for (const oProp in oTransientObject) {
-                        if (oProp !== "__metadata" && mUpdatableLookupMap[oProp]) {
-                            oObjectPageContext.setProperty(oProp, oTransientObject[oProp]);
-                        }
-                    }
-                    activateSideEffects();
-                    resetEasyFill(true);
-                    oEasyFillDialog?.close();
+
+                // ── Feedback thumbs-up/down ───────────────────────────────────────
+                onFeedbackPress: function() {
+                    EasyFillDialogController.onFeedbackPress(oController, oRb);
                 },
-                onLiveChange: function(oEvent) {
-                    const easyFillDialogModel = oController.getView().getModel("easyFillDialogModel");
-                    easyFillDialogModel.setProperty("/isEasyFillButtonEnabled",!!oEvent.getSource().getValue());
+
+                // ── Section title formatter (resolves i18n patterns) ──────────────
+                formatSectionTitle: function(sTitle) {
+                    return EasyFillDialogController.formatSectionTitle(sTitle, oController);
                 },
-                onCancelEasyFillDialog : function () {
-                    resetEasyFill(true);
-                    oEasyFillDialog.close();
+
+                // ── IllustratedMessage state formatters ───────────────────────────
+                formatIllustrationType: function(sState) {
+                    return EasyFillDialogController.formatIllustrationType(sState);
                 },
-                onEasyFillSubmitInputToAI : async function (oEvent) {  
-                    const easyFillDialogModel = oController.getView().getModel("easyFillDialogModel");
-                    easyFillDialogModel.setProperty("/isBusy",true);
-                    let mFieldMapping = getEntitySetMapForLLM();
-                    let aiCallResult = await  oTemplateUtils.oServices.oFioriAIHandler.fioriaiLib.EasyFill.extractFieldValuesFromText(oController.byId("EasyFillInputTextArea").getValue(), mFieldMapping);
-                    resetEasyFill();
-                    aEditableFields = [];
-                    easyFillDialogModel.setProperty("/isBusy",false);
-                    formatDateValues(aiCallResult,mFieldMapping);       
-                    if (aiCallResult.success) {
-                        const {updatableFields,nonUpdatableFields} = getUpdatableAndNonUpdatableFields(aiCallResult.data);
-                        mUpdatableLookupMap = updatableFields;
-                        if (Object.keys(updatableFields).length + Object.keys(nonUpdatableFields).length === 0) {
-                            easyFillDialogModel.setProperty("/stateType","NoEntries");
-                            return;
-                        }
-                        easyFillDialogModel.setProperty("/isIllustrationVisible",false);
-                        //Fillable Fields
-                        makeTransientContext();
-                        easyFillDialogModel.setProperty("/isUpdatableVisible",Object.keys(updatableFields).length > 0);
-                        if (Object.keys(updatableFields).length > 0) {
-                            const oAIResponseReviewRegionSmartFormForUpdatableFields = oController.byId("EasyFillUpdatebleForm");
-
-                            const oOldGroup = getGroupForSmartForm(updatableFields,false,oObjectPage.getBindingContext(),false);
-                            const oNewGroup = getGroupForSmartForm(updatableFields,true,oTransientContextForEasyFill,true);
-                            oAIResponseReviewRegionSmartFormForUpdatableFields.addGroup(oOldGroup);
-                            oAIResponseReviewRegionSmartFormForUpdatableFields.addGroup(oNewGroup);
-                            validateSmartForm();
-                        }
-                        //Non-Fillable Fields
-                        easyFillDialogModel.setProperty("/isNonUpdatableVisible",Object.keys(nonUpdatableFields).length > 0);
-                        if (Object.keys(nonUpdatableFields).length > 0) {
-                            const oAIResponseReviewRegionSmartFormForNonUpdatableFields = oController.byId("EasyFillNonUpdatebleForm");
-                            const oOldGroup = getGroupForSmartForm(nonUpdatableFields,false,oObjectPage.getBindingContext(),false);
-                            const oNewGroup = getGroupForSmartForm(nonUpdatableFields,false,oTransientContextForEasyFill,true);
-                            oAIResponseReviewRegionSmartFormForNonUpdatableFields.addGroup(oOldGroup);
-                            oAIResponseReviewRegionSmartFormForNonUpdatableFields.addGroup(oNewGroup);
-                        }
-                    } else {
-                        easyFillDialogModel.setProperty("/stateType","Error");
-                    }
+                formatIllustrationTitle: function(sState) {
+                    return EasyFillDialogController.formatIllustrationTitle(sState, oRb);
                 },
-                onEasyFillClearAll: function() {
-                    resetEasyFill(true);
+                formatIllustrationDescription: function(sState) {
+                    return EasyFillDialogController.formatIllustrationDescription(sState, oRb);
                 },
-                onLinkPress: async function(oEvent) {
-                    if (!oAIPopover) {
-                        oAIPopover = await getAIPopover();
-                    }
-                    oAIPopover.openBy(oEvent.getSource());
-                }
-			}, "easyFillPopup",undefined,true,true).then(function (oPopup) {
-                oEasyFillDialog = oPopup;
-                oPopup?.open();
-			});
-        }
-
-        function activateSideEffects() {
-            //Making a set so that , duplicate field group ids are not passed
-            const aFieldGroupIdSet = new Set();
-            aEditableFields.forEach((oSmartField)=>{
-                //SideEffect would be invoked only for the editable controls inside the SmartField
-                const aEditableInnerFields = oSmartField.getInnerControls().filter((oControl)=>oControl.getEditable());
-                aEditableInnerFields.forEach((oControl)=>{
-                    const aControlFieldGroupId = oControl.getFieldGroupIds();
-                    if (aControlFieldGroupId.length > 0) {
-                        aControlFieldGroupId.forEach((sFieldGroupId)=>{
-                            aFieldGroupIdSet.add(sFieldGroupId);
-                        });
-                    }
-                });
 
-            });
+                // ── Confirm: apply AI values to the live OP context ───────────────
+                onSaveFromEasyFillDialog: async function() {
+                    EasyFillApplyHandler.applyAndSave(
+                        oTransientContextForEasyFill,
+                        oObjectPageContext,
+                        mUpdatableLookupMap,
+                        oTableStagingState,
+                        mCollectionState
+                    );
+                    _resetEasyFill(true);
+                    EasyFillDialogController.closeDialog(oController);
+                },
 
-            const oView = oController.getView();
+                // ── Text area live change: enable/disable submit button ───────────
+                onLiveChange: function(oEvent) {
+                    EasyFillDialogController.onLiveChange(oEvent, oController);
+                },
 
-            aFieldGroupIdSet.forEach((sFieldGroupId)=>{
-                const oID = oView.data(sFieldGroupId);
+                // ── Cancel: revert staged values and close dialog ─────────────────
+                onCancelEasyFillDialog: function() {
+                    _resetEasyFill(true);
+                    EasyFillDialogController.closeDialog(oController);
+                },
 
-                /**
-                 * Currently, the SmartFields are created using the TransientContext, which means both
-                 * "context" and "contextObject" refer to the TransientContext instance.
-                 *
-                 * However, we need to use the original OP context so that side effects are triggered
-                 * on the original OP context when the user saves data from the EasyFill dialog back to OP.
-                 *
-                 * Therefore, we update the "context" and "contextObject" to reference the original OP context.
-                 */
+                // ── Field value formatter (binding expression builder) ────────────
+                formatFieldValue: function(sFieldName) {
+                    return "{" + sFieldName + "}";
+                },
 
-                if (oID) {
-                    oID["context"] = oObjectPageContext.getPath();
-                    oID["contextObject"] = oObjectPageContext;
-                    oView.data(sFieldGroupId,oID);
-                }
-                //To make sure that all the fields sideeffect would be triggered
-                const sSideEffectQualifier = oTemplateUtils.oServices.oApplicationController._getSideEffectsQualifier(oID.name);
-                oTemplateUtils.oServices.oApplicationController.registerGroupChange(sSideEffectQualifier);
-            });
+                // ── Submit user text to AI ────────────────────────────────────────
+                onEasyFillSubmitInputToAI: async function() {
+                    await _handleAISubmit();
+                },
 
+                // ── Clear all (reset without closing) ────────────────────────────
+                onEasyFillClearAll: function() {
+                    _resetEasyFill(true);
+                },
 
-            oView.triggerValidateFieldGroup([...aFieldGroupIdSet]);
-        }
+                // ── SmartForm group population formatter ──────────────────────────
+                populateSmartFields: function(sId, aFields, sEasyFillMode) {
+                    return EasyFillFormRenderer.populateSmartFields(
+                        sId, aFields, sEasyFillMode,
+                        oController,
+                        oTransientContextForEasyFill,
+                        oObjectPageContext,
+                        mUpdatableLookupMap,
+                        oDialogModel().getProperty("/llmData")
+                    );
+                },
 
-        function getAIPopover() {
-            return new Promise(async(resolve,reject) => {
-                const sFragmentname = "sap.suite.ui.generic.template.fragments.AINotice";
-                const oPopover = await oTemplateUtils.oCommonUtils.getDialogFragmentAsync(sFragmentname,{
-                    onCloseAiPopover: function() {
-                        oAIPopover.close();
-                    }
-                },"aiPopover",undefined,true);
-                resolve(oPopover);
-            });
-        }
+                // ── Fullscreen toggle ─────────────────────────────────────────────
+                onToggleFullScreen: function() {
+                    EasyFillDialogController.onToggleFullScreen(oController);
+                },
 
-        function getEntitySetMapForLLM() {
-            const mPropertyMap = {};
-            const oEntityType = oTemplateUtils.oCommonUtils.getMetaModelEntityType(oController.getOwnerComponent().getEntitySet());
-            oEntityType.property.forEach((oProperty)=>{
-                if (isVisible(oProperty)) {
-                    mPropertyMap[oProperty.name] = {
-                        description: getLabel(oProperty),
-                        dataType: oProperty.type
-                    };
+                // ── AI Notice popover ─────────────────────────────────────────────
+                onLinkPress: async function(oEvent) {
+                    await EasyFillDialogController.onLinkPress(oEvent, oTemplateUtils, oAIPopoverRef);
                 }
-            });
-            return mPropertyMap;
+            };
         }
 
-        function getLabel(oProperty) {
-            return oProperty["com.sap.vocabularies.Common.v1.Label"] ? AnnotationHelperModel.format(oObjectPageContext,oProperty["com.sap.vocabularies.Common.v1.Label"]) : oProperty["sap:label"];
-        }
+        /**
+         * Handles the full AI submission flow:
+         * 1. Build field mapping
+         * 2. Call AI service
+         * 3. Post-process response (dates, split)
+         * 4. Update dialog model
+         * 5. Populate SmartForms and table previews
+         */
+        async function _handleAISubmit() {
+            var oModel = oDialogModel();
+            oModel.setProperty("/isBusy", true);
+
+            // Build complete LLM field mapping (entity fields + collection fields)
+            var mFieldMapping = EasyFillAIOrchestrator.buildFieldMapping(
+                oTemplateUtils, oController, oObjectPage, oObjectPageModel,
+                oObjectPageContext, EasyFillFieldCollector, mCollectionState
+            );
+
+            // Submit to AI
+            var aiCallResult = await EasyFillAIOrchestrator.submitToAI(
+                oController.byId("EasyFillInputTextArea").getValue(),
+                mFieldMapping,
+                oTemplateUtils
+            );
+
+            // Reset editable fields tracker for fresh population
+            oModel.setProperty("/isBusy", false);
+
+            // Convert date strings in AI response to JS Date objects
+            EasyFillAIOrchestrator.formatDateValues(aiCallResult, mFieldMapping);
+
+            var sSelectedKey = oModel.getProperty("/easyFillMode") || EasyFillLayoutMode.CondensedMode;
+            oModel.setProperty("/easyFillMode", sSelectedKey);
 
-        function formatDateValues(aiCallResult,mFieldMapping) {
             if (aiCallResult.success) {
-                for (const key of Object.keys(aiCallResult.data)) {
-                    const sDataType = mFieldMapping[key].dataType;
-                    if (sDataType === "Edm.DateTimeOffset" || sDataType === "Edm.DateTime") {
-                        aiCallResult.data[key] = new Date(aiCallResult.data[key]);
+                // Split AI response into simple fields (SmartForm) and collection rows (SmartTable)
+                var oSplitResult = EasyFillAIOrchestrator.splitAIResultByCollection(
+                    aiCallResult.data, mFieldMapping, EasyFillFieldCollector, oObjectPage
+                );
+                var simpleFields = oSplitResult.simpleFields;
+                var collectionFields = oSplitResult.collectionFields;
+
+                // Separate updatable from non-updatable fields
+                var oEntityType = oTemplateUtils.oCommonUtils.getMetaModelEntityType(
+                    oController.getOwnerComponent().getEntitySet()
+                );
+                var oSplit = EasyFillFieldCollector.getUpdatableAndNonUpdatableFields(
+                    simpleFields, oEntityType, oObjectPageContext
+                );
+                mUpdatableLookupMap = oSplit.updatableFields;
+
+                // Build section tree for selected view mode
+                var aData = (sSelectedKey === EasyFillLayoutMode.CondensedMode)
+                    ? EasyFillFieldCollector.getSectionsForCondensedView(simpleFields)
+                    : EasyFillFieldCollector.getSectionsForObjectPageView(oEntityType, simpleFields, oRb);
+
+                // Reset form (destroy previous SmartForm content + revert table staging)
+                // then create a fresh transient context for the new AI result
+                _resetEasyFill();
+                oTransientContextForEasyFill = EasyFillFormRenderer.makeTransientContext(
+                    null, oObjectPageModel, oController
+                );
+
+                // Update dialog model with section data and AI result
+                oModel.setProperty("/sections", aData);
+                oModel.setProperty("/llmData", simpleFields);
+
+                // Process collection/table updates
+                mCollectionUpdates = collectionFields;
+                var bHasTableUpdates = Object.keys(mCollectionUpdates).length > 0;
+                oModel.setProperty("/hasTableUpdates", bHasTableUpdates);
+
+                // Render collection previews into the table preview container
+                EasyFillTableRenderer.renderCollectionPreviews(
+                    mCollectionUpdates,
+                    mFieldMapping,
+                    [],
+                    oController,
+                    oRb,
+                    mCollectionState,
+                    oTableStagingState,
+                    oObjectPageModel,
+                    function() {
+                        _refreshTableValidationAndSaveState().catch(Function.prototype);
                     }
-                }
-            }
-        }
+                );
+
+                await _refreshTableValidationAndSaveState();
 
-        function validateSmartForm() {
-            setTimeout(async() => {
-                const easyFillDialogModel = oController.getView().getModel("easyFillDialogModel");
-                try {
-                    const oAIResponseReviewRegionSmartFormForUpdatableFields = oController.byId("EasyFillUpdatebleForm");
-                    const aIds = await oAIResponseReviewRegionSmartFormForUpdatableFields.check();
-                    easyFillDialogModel.setProperty("/isSaveEnabled",aIds.length === 0);
-                } catch (error) {
-                    easyFillDialogModel.setProperty("/isSaveEnabled",false);
+                var iTotalFields = Object.keys(oSplit.updatableFields).length + Object.keys(oSplit.nonUpdatableFields).length;
+                if (iTotalFields === 0 && !bHasTableUpdates) {
+                    oModel.setProperty("/stateType", "NoEntries");
+                    return;
                 }
-            }, 0);
-        }
 
-        function resetEasyFill(bRemoveTextAreaValue) {
-            const easyFillDialogModel =  oController.getView().getModel("easyFillDialogModel");
-            const oAIResponseReviewRegionSmartFormForNonUpdatableFields = oController.byId("EasyFillNonUpdatebleForm");
-            const oAIResponseReviewRegionSmartFormForUpdatableFields = oController.byId("EasyFillUpdatebleForm");
-            oAIResponseReviewRegionSmartFormForUpdatableFields.destroyGroups();
-            oAIResponseReviewRegionSmartFormForNonUpdatableFields.destroyGroups();
-            resetTransientContext();
-            easyFillDialogModel.setProperty("/isUpdatableVisible",false);
-            easyFillDialogModel.setProperty("/isNonUpdatableVisible",false);
-            easyFillDialogModel.setProperty("/isIllustrationVisible",true);
-            easyFillDialogModel.setProperty("/isSaveEnabled",false);
-            easyFillDialogModel.setProperty("/stateType","Initial");
-            easyFillDialogModel.setProperty("/feedbackEnabled",true);
-            easyFillDialogModel.setProperty("/feedbackPressedUp",false);
-            easyFillDialogModel.setProperty("/feedbackPressedDown",false);
-            if (bRemoveTextAreaValue) {
-                oController.byId("EasyFillInputTextArea").setValue("");
-                easyFillDialogModel.setProperty("/isEasyFillButtonEnabled",false);
+                oModel.setProperty("/isIllustrationVisible", false);
+                oModel.setProperty("/isFormVisible", iTotalFields > 0);
+            } else {
+                // AI call failed — show error illustration
+                EasyFillApplyHandler.applyTableValidationState([], oModel, oRb);
+                oModel.setProperty("/stateType", "Error");
             }
         }
 
-        function getUpdatableAndNonUpdatableFields(oAiCallResult) {
-            const oEntityType = oTemplateUtils.oCommonUtils.getMetaModelEntityType(oController.getOwnerComponent().getEntitySet());
-            const updatableFields = {};
-            const nonUpdatableFields = {};
-
-            Object.keys(oAiCallResult).forEach((sKey)=>{
-                const oProperty = oEntityType.property.find((oProperty)=>oProperty.name === sKey);
-                    if (isEditable(oProperty)) {
-                        updatableFields[sKey] = oAiCallResult[sKey];
-                    } else {
-                        nonUpdatableFields[sKey] = oAiCallResult[sKey];
-                    }
-
-            });
-            return {updatableFields,nonUpdatableFields};
+        async function _refreshTableValidationAndSaveState() {
+            var oModel = oDialogModel();
+            var mStagedCollectionUpdates = EasyFillApplyHandler.buildStagedCollectionUpdates(
+                oTableStagingState,
+                mCollectionState
+            );
+            var aTableValidationErrors = await EasyFillApplyHandler.validateCollectionUpdatesAgainstValueHelp(
+                mStagedCollectionUpdates,
+                mCollectionState,
+                oObjectPageModel,
+                oRb,
+                EasyFillFieldCollector,
+                GenericAnnotationHelper
+            );
+            EasyFillApplyHandler.applyTableValidationState(aTableValidationErrors, oModel, oRb);
+
+            var bHasScalarChanges = Object.keys(mUpdatableLookupMap || {}).length > 0;
+            var bHasTableChanges = EasyFillApplyHandler.hasStagedCollectionChanges(
+                oTableStagingState,
+                mCollectionState
+            );
+
+            oModel.setProperty(
+                "/isSaveEnabled",
+                bHasScalarChanges || (bHasTableChanges && aTableValidationErrors.length === 0)
+            );
+
+            return aTableValidationErrors;
         }
 
-        //Checks if the field is editable or not
-
-        // It can be set in 2 ways using field-control, it would be present when sap:updatable attribute is not present
-        // 1. By attaching to an existing metadata property, sap:field-control-> path for the property
-        //Below are the values by which the editable state would be defined
-
         /**
-         * Value	Meaning
-         -----------------------------------------------
-            0	    Hidden (field not shown at all)
-            1	    Read-only (display only)
-            3	    Editable
-            7	    Mandatory (Still Editable)
-            */
-
-        // 2. By Adding to an annotation "com.sap.vocabularies.Common.v1.FieldControl"
-        
-        //Since "com.sap.vocabularies.Common.v1.FieldControl" is an V4 annotation, it would be given the first priority
-
-        function isEditable(oProperty) {
-
-            //Check for sap:updatable
-            const sUpdatable = oProperty["sap:updatable"];
-            if (sUpdatable) {
-               return sUpdatable === "false" ? false : true;
-            }
-
-            //Check for "com.sap.vocabularies.Common.v1.FieldControl"
-            const oAnnotation = oProperty["com.sap.vocabularies.Common.v1.FieldControl"];
-
-            if (oAnnotation && oAnnotation["EnumMember"]) {
-                const enumNumberType = oAnnotation["EnumMember"];
-                return enumNumberType === 'com.sap.vocabularies.Common.v1.FieldControlType/ReadOnly' ? false : true;
-            }
-
-
-            //Check for "sap:field-control"
-            const sFieldControl = oProperty["sap:field-control"];
-
-            if (sFieldControl) {
-                const iFieldControlValue = oObjectPageContext.getObject()[sFieldControl];
-                return iFieldControlValue === 1 ? false : true;
-            }
-
-            //If no restriction is present then by default all fields are editable
-            return true;
-        }
-
-        function isVisible(oProperty) {
-
-            //Check for "com.sap.vocabularies.Common.v1.FieldControl"
-            const oAnnotation = oProperty["com.sap.vocabularies.Common.v1.FieldControl"];
-
-            if (oAnnotation && oAnnotation["EnumMember"]) {
-                const enumNumberType = oAnnotation["EnumMember"];
-                return enumNumberType === 'com.sap.vocabularies.Common.v1.FieldControlType/Hidden' ? false : true;
-            }
-
-            //Check for Hidden annotation i.e com.sap.vocabularies.UI.v1.Hidden
-            const oHiddenAnnotation = oProperty["com.sap.vocabularies.UI.v1.Hidden"];
-            if (oHiddenAnnotation) {
-                let bIsVisible = false;    // <Annotation Term="UI.Hidden"/>
-                if (oHiddenAnnotation["Bool"]) {
-                    bIsVisible = (oHiddenAnnotation["Bool"] === "true") ? false : true;
-                } else if (oHiddenAnnotation["Path"]) {
-                    bIsVisible = !oObjectPageContext.getObject()[oHiddenAnnotation["Path"]];
+         * Resets the EasyFill dialog to its initial state.
+         * Delegates to EasyFillApplyHandler.resetEasyFill with the current state references.
+         *
+         * @param {boolean} [bRemoveTextAreaValue=false] - if true, also clears the text area input
+         */
+        function _resetEasyFill(bRemoveTextAreaValue) {
+            EasyFillApplyHandler.resetEasyFill(
+                !!bRemoveTextAreaValue,
+                oDialogModel(),
+                oController,
+                mCollectionUpdates,
+                EasyFillLayoutMode,
+                function() {
+                    EasyFillTableRenderer.resetStagedCollectionContexts(oTableStagingState, oObjectPageModel);
+                },
+                function() {
+                    EasyFillFormRenderer.resetTransientContext(oTransientContextForEasyFill, oObjectPageModel);
+                    oTransientContextForEasyFill = null;
                 }
-                return bIsVisible;
-            }
-
-            //Check for "sap:field-control"
-            const sFieldControl = oProperty["sap:field-control"];
-
-            if (sFieldControl) {
-                const iFieldControlValue = oObjectPageContext.getObject()[sFieldControl];
-                return iFieldControlValue === 0 ? false : true;
-            }
-
-            //If no restriction is present then by default all fields are visible
-            return true;
-        }
-
-        function resetTransientContext() {
-            if (oTransientContextForEasyFill) {
-                oObjectPageModel.resetChanges([oTransientContextForEasyFill.getPath()],false,true);
-                oTransientContextForEasyFill = null;
-            }
-        }
-
-        function makeTransientContext() {
-            if (oTransientContextForEasyFill) {
-                resetTransientContext();
-            }
-            oTransientContextForEasyFill = oObjectPageModel.createEntry("/" + oController.getOwnerComponent().getEntitySet(), {
-                "groupId" : "EasyFillFETransientChanges",
-                "changeSetId":"EasyFillFETransientChanges"
-            });
+            );
         }
 
-         function onSmartFieldChange() {
-            validateSmartForm();
+        /**
+         * Convenience accessor for the easyFillDialogModel JSONModel.
+         * @returns {sap.ui.model.json.JSONModel}
+         */
+        function oDialogModel() {
+            return oController.getView().getModel("easyFillDialogModel");
         }
 
-        function getGroupForSmartForm(oDataFromLLM, isEditable,oContext,isTransient) {
-            var oGroup = new Group({
-                title: (isTransient) ? oRb.getText("EASY_FILL_PROPOSED_VALUES") : oRb.getText("EASY_FILL_PREVIOUS_VALUES")
-            });
-
-            for (const key of Object.keys(oDataFromLLM)) {
-                const oGroupElement = new GroupElement();
-                const oSmartField = new SmartField({
-                    value: "{" + key + "}",
-                    editable: isEditable,
-                    enabled: isEditable,
-                    contextEditable: isEditable,
-                    change: onSmartFieldChange.bind(this),
-                    editableChanged: (oEvent)=>{
-                        //SmartField changes the editable state of inner controls based on various factors, so to make sure that the editable state remains as per our requirement we are forcing it here
-                        if (oEvent.getParameter("editable") !== isEditable) {
-                            oEvent.getSource().setEditable(isEditable);
-                        }
-                    }
-                });
-                if (isEditable) {
-                    aEditableFields.push(oSmartField);
-                }
-                const oSmartLabel = new SmartLabel();
-                oSmartLabel.setLabelFor(oSmartField);
-                oGroupElement.addElement(oSmartField);
-                //Update the field only if its transient
-                if (isTransient) {
-                    oContext.setProperty(key,oDataFromLLM[key]);
-                }
-                oGroup.addGroupElement(oGroupElement);
-            }
-            oGroup.setBindingContext(oContext);
-            return oGroup;
-        }
+        // ─── Public API ──────────────────────────────────────────────────────────
 
         return {
+            /**
+             * Entry point called by the ObjectPage ControllerImplementation when the
+             * EasyFill button is pressed.
+             *
+             * If the ObjectPage is already in edit mode, the dialog opens immediately.
+             * If not, the page is transitioned to edit mode first and the resulting
+             * edit context is passed into the dialog.
+             *
+             * @param {sap.ui.base.Event} oEvent - the button click event
+             */
             onEasyFillButtonClick: function(oEvent) {
-                // Once you click on the easyFill button the OP should be going to editable state, if its already in editState then directly open the dialog
-                //If its in non-editable state we should wait until the context switch, We need that context to finally save the data from the AI into the newly came editable context of the OP
-                const isEditable = oController.getView().getModel("ui").getProperty("/editable");
-                if (isEditable) {
+                var bIsEditable = oController.getView().getModel("ui").getProperty("/editable");
+                if (bIsEditable) {
                     openEasyFillDialog(oEvent);
                 } else {
-                    oController._templateEventHandlers.onEditByEasyFill().then((oContext)=>{
-                        oObjectPageContext = oContext;
-                        openEasyFillDialog(oEvent,oContext);
-                    }).catch(Function.prototype);
+                    // Trigger the OP edit flow and wait for the new edit context before opening
+                    oController._templateEventHandlers.onEditByEasyFill()
+                        .then(function(oContext) {
+                            oObjectPageContext = oContext;
+                            openEasyFillDialog(oEvent, oContext);
+                        })
+                        .catch(Function.prototype);
                 }
             }
         };
     }
 
     return BaseObject.extend("sap.suite.ui.generic.template.lib.ai.EasyFill.EasyFillHandler", {
-        constructor: function (oState, oController, oTemplateUtils, oObjectPage) {
+        /**
+         * Constructor — wires all helpers and exposes the public API via getMethods.
+         *
+         * @param {object} oState - shared template state (passed from ControllerImplementation)
+         * @param {object} oController - MVC controller instance
+         * @param {object} oTemplateUtils - FE template utilities (services, common utils, etc.)
+         * @param {object} oObjectPage - the ObjectPage control instance
+         */
+        constructor: function(oState, oController, oTemplateUtils, oObjectPage) {
             extend(this, getMethods(oState, oController, oTemplateUtils, oObjectPage));
         }
     });
-    
 });