FIX: AI Helper UX bugs (#223)

assets/javascripts/discourse/connectors/after-d-editor/ai-helper-context-menu.js

@@ -121,7 +121,7 @@ export default class AiHelperContextMenu extends Component {
         )
       : "";
 
-    if (this.selectedText.length === 0) {
+    if (this.selectedText?.length === 0) {
       this.closeContextMenu();
       return;
     }

assets/stylesheets/modules/ai-helper/common/ai-helper.scss

@@ -206,11 +206,18 @@
 
 // Suggest Titles Related
 .suggest-titles-button {
-  position: absolute;
-  top: 1px;
-  right: 1px;
+  display: block;
+  align-self: baseline;
   background: var(--secondary);
-  border: none;
+  border: 1px solid var(--primary-medium);
+  border-left: none;
+}
+
+.title-input:has(.suggestion-button) {
+  // border on focus should be on top of suggestion button
+  input:focus {
+    z-index: 1;
+  }
 }
 
 .suggestion-button {
@@ -249,6 +256,21 @@
   position: relative;
 }
 
+// Prevent suggestion button from wrapping on smaller screens
+@media screen and (max-width: 768px) {
+  #reply-control {
+    .category-input:has(.suggestion-button) {
+      .category-chooser {
+        width: 10px;
+      }
+
+      + .mini-tag-chooser {
+        width: 10px;
+      }
+    }
+  }
+}
+
 .suggest-tags-button + .ai-suggestions-menu {
   top: 4.25rem;
 }