fix: update reload action

Author: QianKuang8

extensions/intellij/src/main/kotlin/com/github/continuedev/continueintellijextension/actions/ContinuePluginActions.kt

@@ -95,25 +95,32 @@ class ReloadBrowserAction: ContinueToolbarAction() {
     override fun toolbarActionPerformed(project: Project) {
         val toolWindow = ToolWindowManager.getInstance(project).getToolWindow("Continue")
             ?: return
-        val contentManager = toolWindow.contentManager
         val browserService = project.service<ContinueBrowserService>()
-        browserService.reload()
 
-        val newBrowser = project.getBrowser() ?: return
-        val newBrowserComponent = newBrowser.getComponent()
-
-        contentManager.removeAllContents(true)
-        val newContent = contentManager.factory.createContent(
-            newBrowserComponent,
-            null,
-            false
-        )
-
-        contentManager.addContent(newContent)
-
-        contentManager.setSelectedContent(newContent)
-
-        toolWindow.activate(null)
+        // Perform the reload and UI update on the Event Dispatch Thread
+        ApplicationManager.getApplication().invokeLater {
+            // Reload the browser service to get a new browser instance
+            browserService.reload()
+
+            val newBrowser = project.getBrowser() ?: return@invokeLater
+            val newBrowserComponent = newBrowser.getComponent()
+
+            val contentManager = toolWindow.contentManager
+            contentManager.removeAllContents(true)
+
+            val newContent = contentManager.factory.createContent(
+                newBrowserComponent,
+                null,
+                false
+            )
+            contentManager.addContent(newContent)
+            contentManager.setSelectedContent(newContent, true) // Request focus
+
+            toolWindow.activate({
+                // After activation, ensure the browser's input field gets focus
+                newBrowser.focusOnInput()
+            }, true)
+        }
     }
 }
 

extensions/intellij/src/main/kotlin/com/github/continuedev/continueintellijextension/browser/ContinueBrowserService.kt

@@ -40,11 +40,18 @@ class ContinueBrowserService(val project: Project): Disposable {
      * This method is intended for use when browser is frozen (unresponsive).
      */
     fun reload() {
-        browser?.let {
-            Disposer.dispose(it)
-        }
+        // Store the old browser instance to be disposed later
+        val oldBrowser = browser
         browser = null
 
+        // Dispose the old browser after the new one is loaded and UI is updated.
+        // This avoids race conditions. We can do this on a background thread.
+        oldBrowser?.let {
+            ApplicationManager.getApplication().invokeLater {
+                Disposer.dispose(it)
+            }
+        }
+
         load()
     }