diff --git a/tdx-enhanced.js b/tdx-enhanced.js
index 81893d3..ba1991d 100644
--- a/tdx-enhanced.js
+++ b/tdx-enhanced.js
@@ -713,6 +713,13 @@
         document.querySelectorAll("div.moreToggle *:not(a), div.lessToggle *:not(a), div.wrap-text *:not(a)").forEach(textBlock=>{
             textBlock.style.color = "var(--txt-1)"
         })
+        document.querySelectorAll("#ToggleAIAgentAssist").forEach(AIAgentButton=>{     
+        let selected = settings('get','aiBehavior');
+            console.log("AI Behavior selected:",selected.value)
+            if (selected == "aiDisable") {
+                AIAgentButton.style.display = "none"
+            }  
+        })
 
         let desktopLayout = settings('get','layout')
         let desktop = document.querySelector("#divContent")
@@ -1466,6 +1473,13 @@
                     <input type="radio" id="highlightBehavior-text" name="highlightBehavior" value="text">
                     <label for="highlightBehavior-block">Text</label><br>
                 </div>
+                <div>
+                    <h3>AI Agent</h3>
+                    <input type="radio" id="aiBehavior-enable" name="aiBehavior" value="aiEnable">
+                    <label for="aiBehavior-enable">AI Enable</label><br>
+                    <input type="radio" id="aiBehavior-disable" name="aiBehavior" value="aiDisable">
+                    <label for="aiBehavior-disable">AI Disable</label>
+                </div>
             </form>
 
         </div>