fix: return to settings view after migration wizard when opened from settings

packages/kilo-vscode/webview-ui/src/App.tsx

@@ -164,6 +164,7 @@ export const LanguageBridge: Component<{ children: any }> = (props) => {
 // Inner app component that uses the contexts
 const AppContent: Component = () => {
   const [currentView, setCurrentView] = createSignal<ViewType>("newTask")
+  const [migrationReturnView, setMigrationReturnView] = createSignal<ViewType>("newTask") // legacy-migration
   const session = useSession()
   const server = useServer()
 
@@ -248,12 +249,20 @@ const AppContent: Component = () => {
           />
         </Match>
         <Match when={currentView() === "settings"}>
-          <Settings onMigrateClick={() => setCurrentView("migration")} />
+          <Settings
+            onMigrateClick={() => {
+              setMigrationReturnView("settings")
+              setCurrentView("migration")
+            }}
+          />
           {/* legacy-migration */}
         </Match>
         {/* legacy-migration start */}
         <Match when={currentView() === "migration"}>
-          <MigrationWizard onBack={() => setCurrentView("newTask")} onComplete={() => setCurrentView("newTask")} />
+          <MigrationWizard
+            onBack={() => setCurrentView(migrationReturnView())}
+            onComplete={() => setCurrentView(migrationReturnView())}
+          />
         </Match>
         {/* legacy-migration end */}
       </Switch>