[AiAssistance] Focus text input after screenshot actions

Fixed: 399782817
Change-Id: Ie9b1efc079b12c1e1dd595a3fbf5a422410f5d00
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6311454
Auto-Submit: Ergün Erdoğmuş <[email protected]>
Reviewed-by: Alex Rudenko <[email protected]>
Commit-Queue: Alex Rudenko <[email protected]>
Commit-Queue: Ergün Erdoğmuş <[email protected]>

Author: ergunsh

front_end/panels/ai_assistance/AiAssistancePanel.ts

@@ -1134,12 +1134,18 @@ export class AiAssistancePanel extends UI.Panel.Panel {
     if (bytes) {
       this.#imageInput = bytes;
       this.requestUpdate();
+      void this.updateComplete.then(() => {
+        this.#viewOutput.chatView?.focusTextInput();
+      });
     }
   }
 
   #handleRemoveImageInput(): void {
     this.#imageInput = '';
     this.requestUpdate();
+    void this.updateComplete.then(() => {
+      this.#viewOutput.chatView?.focusTextInput();
+    });
   }
 
   #handleTextInputChange(value: string): void {