appium-mcp 1.6.1 → 1.7.0

package/dist/ui/mcp-ui-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-ui-utils.js","sourceRoot":"","sources":["../../src/ui/mcp-ui-utils.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH;;;;;GAKG;AACH,MAAM,UAAU,gBAAgB,CAC9B,GAAW,EACX,WAAmB;IASnB,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,QAAQ,EAAE;YACR,GAAG;YACH,QAAQ,EAAE,WAAW;YACrB,IAAI,EAAE,WAAW;SAClB;KACF,CAAC;AACJ,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,oBAAoB,CAClC,OAKE,EACF,QAA2B,EAC3B,UAAiC;IAEjC,MAAM,eAAe,GACnB,QAAQ,KAAK,KAAK,IAAI,UAAU;QAC9B,CAAC,CAAC,UAAU,KAAK,WAAW;YAC1B,CAAC,CAAC,gBAAgB;YAClB,CAAC,CAAC,aAAa;QACjB,CAAC,CAAC,iBAAiB,CAAC;IAExB,MAAM,WAAW,GAAG,OAAO;SACxB,GAAG,CACF,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;0CACiB,MAAM,CAAC,IAAI,iBAAiB,KAAK;;cAE7D,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI;UAC9B,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,6BAA6B,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,MAAM,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE;;;0CAGrE,MAAM,CAAC,IAAI;UAC3C,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,6BAA6B,MAAM,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,EAAE;;iEAEV,MAAM,CAAC,IAAI;;;;GAIzE,CACE;SACA,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO;;;;;;2BAMkB,eAAe;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;sBAoHpB,eAAe;iBACpB,OAAO,CAAC,MAAM,UAAU,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;;;QAGhE,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,iDAAiD;;;;;;;;;;;yBAWnE,QAAQ;cACnB,UAAU,CAAC,CAAC,CAAC,gBAAgB,UAAU,IAAI,CAAC,CAAC,CAAC,EAAE;;;;;;;;;;;;;;;;;;GAkB3D,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,wBAAwB,CACtC,gBAAwB,EACxB,QAAgB;IAEhB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iCA2HwB,QAAQ;;;;;;;;wCAQD,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;yBA0C/B,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,gBAAgB;;;;;;;;;;;;;;;;;;;;;;;GAuBnE,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CAAC,WAOxC;IACC,qCAAqC;IACrC,MAAM,YAAY,GAChB,OAAO,WAAW,CAAC,SAAS,KAAK,QAAQ;QACvC,CAAC,CAAC,WAAW,CAAC,SAAS;QACvB,CAAC,CAAC,MAAM,CAAC,WAAW,CAAC,SAAS,IAAI,SAAS,CAAC,CAAC;IAEjD,gEAAgE;IAChE,MAAM,gBAAgB,GACpB,YAAY,CAAC,MAAM,GAAG,CAAC;QACrB,CAAC,CAAC,GAAG,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK;QACtC,CAAC,CAAC,YAAY,CAAC;IAEnB,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;mBAuHU,gBAAgB;;;;mBAIhB,WAAW,CAAC,QAAQ;;;;mBAIpB,WAAW,CAAC,cAAc;;UAGnC,WAAW,CAAC,UAAU;QACpB,CAAC,CAAC;;;mBAGK,WAAW,CAAC,UAAU;;SAEhC;QACG,CAAC,CAAC,EACN;UAEE,WAAW,CAAC,eAAe;QACzB,CAAC,CAAC;;;mBAGK,WAAW,CAAC,eAAe;;SAErC;QACG,CAAC,CAAC,EACN;UAEE,WAAW,CAAC,IAAI;QACd,CAAC,CAAC;;;kDAGoC,WAAW,CAAC,IAAI;;SAEzD;QACG,CAAC,CAAC,EACN;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAkEL,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CACtC,QASE;IAEF,MAAM,YAAY,GAAG,QAAQ;SAC1B,GAAG,CACF,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC;4CACkB,KAAK;;cAEnC,OAAO,CAAC,OAAO;;YAEjB,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,sDAAsD,CAAC,CAAC,CAAC,EAAE;YAC/E,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,kDAAkD,CAAC,CAAC,CAAC,EAAE;YACzE,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,sDAAsD,CAAC,CAAC,CAAC,EAAE;;;QAGnF,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,kDAAkD,OAAO,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,EAAE;QACxF,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,0DAA0D,OAAO,CAAC,WAAW,MAAM,CAAC,CAAC,CAAC,EAAE;QAC9G,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,+DAA+D,OAAO,CAAC,UAAU,aAAa,CAAC,CAAC,CAAC,EAAE;;UAEtH,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC;SAC/B,GAAG,CACF,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,EAAE,CAAC;;qCAED,QAAQ;qCACR,QAAQ;6DACgB,QAAQ,QAAQ,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC;;SAEjG,CACE;SACA,IAAI,CAAC,EAAE,CAAC;;;GAGhB,CACE;SACA,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBAgIQ,QAAQ,CAAC,MAAM,wBAAwB,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;;;QAGhF,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,0BAA0B;;;;;;;;;;;;;;;;;;;GAmBpE,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,2BAA2B,CAAC,UAAkB;IAC5D,+BAA+B;IAC/B,MAAM,aAAa,GAAG,UAAU;SAC7B,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC;SACtB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC;SACvB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAE3B,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;2BAoGkB,UAAU,CAAC,MAAM;;;;;;;;;;+CAUG,aAAa;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;gCAsC5B,aAAa;;;;;;;;;;;;GAY1C,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,uBAAuB,CACrC,QAAkB,EAClB,cAA6B;IAE7B,MAAM,YAAY,GAAG,QAAQ;SAC1B,GAAG,CACF,OAAO,CAAC,EAAE,CAAC;+BACc,OAAO,KAAK,cAAc,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE;mCACtC,OAAO;;cAE5B,OAAO;UACX,OAAO,KAAK,cAAc,CAAC,CAAC,CAAC,0CAA0C,CAAC,CAAC,CAAC,EAAE;;;UAG5E,OAAO,KAAK,YAAY,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,YAAY;;oFAEiB,OAAO;UACjF,OAAO,KAAK,cAAc,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;;;GAGxD,CACE;SACA,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBA0GQ,QAAQ,CAAC,MAAM,WAAW,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;;;QAGnE,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,8BAA8B;;;;;;;;;;;;;;;;;;GAkBxE,CAAC;AACJ,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,eAAe,CAC7B,IAAsD;IAEtD,MAAM,QAAQ,GAAG,IAAI;SAClB,GAAG,CACF,GAAG,CAAC,EAAE,CAAC;0CAC6B,GAAG,CAAC,WAAW;;cAE3C,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,WAAW;;;6CAGC,GAAG,CAAC,WAAW;;;gEAGI,GAAG,CAAC,WAAW;mEACZ,GAAG,CAAC,WAAW;gEAClB,GAAG,CAAC,WAAW;;;GAG5E,CACE;SACA,IAAI,CAAC,EAAE,CAAC,CAAC;IAEZ,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;mBAyHU,IAAI,CAAC,MAAM,OAAO,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE;;;;;QAKzD,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,sBAAsB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4DxD,CAAC;AACJ,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,sBAAsB,CACpC,IAAY,EACZ,WAAmB,MAAM;IAEzB,+BAA+B;IAC/B,MAAM,WAAW,GAAG,IAAI;SACrB,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC;SACtB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,MAAM,CAAC;SACrB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC;SACvB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAE3B,OAAO;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCA4F4B,QAAQ;oDACO,IAAI,CAAC,MAAM,gBAAgB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM;;;;;;;;;iDAStD,WAAW;;;;;;;;;;;;;;;;+BAgB7B,QAAQ,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;;;;;;;;;;;;;;;;;;;;;;GAsB/D,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CACrC,QAA6D,EAC7D,UAA+C;IAE/C,OAAO;QACL,OAAO,EAAE,CAAC,GAAG,QAAQ,CAAC,OAAO,EAAE,UAAU,CAAC;KAC3C,CAAC;AACJ,CAAC"}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "appium-mcp",
   "mcpName": "io.github.appium/appium-mcp",
-  "version": "1.6.1",
+  "version": "1.7.0",
   "type": "module",
   "repository": {
     "type": "git",
@@ -17,10 +17,10 @@
   "scripts": {
     "build": "rimraf dist && tsc && chmod +x dist/index.js && npm run copy-docs",
     "copy-docs": "mkdir -p dist/tools/documentation/uploads && cp -f src/tools/documentation/uploads/documents.json dist/tools/documentation/uploads/documents.json 2>/dev/null || true",
-    "start": "node src/index.js",
-    "start:stdio": "node src/index.js",
-    "start:httpStream": "node src/index.js --httpStream",
-    "start:httpStream:port": "node src/index.js --httpStream --port=",
+    "start": "node dist/index.js",
+    "start:stdio": "node dist/index.js",
+    "start:httpStream": "node dist/index.js --httpStream",
+    "start:httpStream:port": "node dist/index.js --httpStream --port=",
     "dev": "npx fastmcp dev src/index.js",
     "inspect": "npx fastmcp inspect src/index.js",
     "test": "NODE_OPTIONS=--experimental-vm-modules jest",

package/src/tools/app-management/list-apps.ts

@@ -1,6 +1,11 @@
 import { FastMCP } from 'fastmcp/dist/FastMCP.js';
 import { z } from 'zod';
 import { getDriver, getPlatformName } from '../../session-store.js';
+import {
+  createUIResource,
+  createAppListUI,
+  addUIResourceToResponse,
+} from '../../ui/mcp-ui-utils.js';
 
 async function listAppsFromDevice(): Promise<any[]> {
   const driver = await getDriver();
@@ -42,7 +47,7 @@ export default function listApps(server: FastMCP): void {
     execute: async () => {
       try {
         const apps = await listAppsFromDevice();
-        return {
+        const textResponse = {
           content: [
             {
               type: 'text',
@@ -50,6 +55,14 @@ export default function listApps(server: FastMCP): void {
             },
           ],
         };
+
+        // Add interactive app list UI
+        const uiResource = createUIResource(
+          `ui://appium-mcp/app-list/${Date.now()}`,
+          createAppListUI(apps)
+        );
+
+        return addUIResourceToResponse(textResponse, uiResource);
       } catch (err: any) {
         return {
           content: [

package/src/tools/context/get-contexts.ts

@@ -1,6 +1,11 @@
 import { FastMCP } from 'fastmcp/dist/FastMCP.js';
 import { z } from 'zod';
 import { getDriver } from '../../session-store.js';
+import {
+  createUIResource,
+  createContextSwitcherUI,
+  addUIResourceToResponse,
+} from '../../ui/mcp-ui-utils.js';
 
 export default function getContexts(server: FastMCP): void {
   server.addTool({
@@ -35,7 +40,7 @@ export default function getContexts(server: FastMCP): void {
           };
         }
 
-        return {
+        const textResponse = {
           content: [
             {
               type: 'text',
@@ -43,6 +48,14 @@ export default function getContexts(server: FastMCP): void {
             },
           ],
         };
+
+        // Add interactive context switcher UI
+        const uiResource = createUIResource(
+          `ui://appium-mcp/context-switcher/${Date.now()}`,
+          createContextSwitcherUI(contexts, currentContext)
+        );
+
+        return addUIResourceToResponse(textResponse, uiResource);
       } catch (err: any) {
         return {
           content: [

package/src/tools/interactions/get-page-source.ts

@@ -1,6 +1,11 @@
 import { FastMCP } from 'fastmcp/dist/FastMCP.js';
 import { z } from 'zod';
 import { getDriver } from '../../session-store.js';
+import {
+  createUIResource,
+  createPageSourceInspectorUI,
+  addUIResourceToResponse,
+} from '../../ui/mcp-ui-utils.js';
 
 export default function getPageSource(server: FastMCP): void {
   server.addTool({
@@ -24,7 +29,7 @@ export default function getPageSource(server: FastMCP): void {
           throw new Error('Page source is empty or null');
         }
 
-        return {
+        const textResponse = {
           content: [
             {
               type: 'text',
@@ -36,6 +41,14 @@ export default function getPageSource(server: FastMCP): void {
             },
           ],
         };
+
+        // Add interactive page source inspector UI
+        const uiResource = createUIResource(
+          `ui://appium-mcp/page-source-inspector/${Date.now()}`,
+          createPageSourceInspectorUI(pageSource)
+        );
+
+        return addUIResourceToResponse(textResponse, uiResource);
       } catch (err: any) {
         return {
           content: [

package/src/tools/interactions/screenshot.ts

@@ -2,6 +2,11 @@ import { FastMCP } from 'fastmcp/dist/FastMCP.js';
 import { getDriver } from '../../session-store.js';
 import { writeFile, mkdir } from 'fs/promises';
 import { join, isAbsolute } from 'path';
+import {
+  createUIResource,
+  createScreenshotViewerUI,
+  addUIResourceToResponse,
+} from '../../ui/mcp-ui-utils.js';
 
 /**
  * Resolves the screenshot directory path.
@@ -66,7 +71,7 @@ export async function executeScreenshot(
     // Save screenshot to disk
     await deps.writeFile(filepath, screenshotBuffer);
 
-    return {
+    const textResponse = {
       content: [
         {
           type: 'text',
@@ -74,6 +79,14 @@ export async function executeScreenshot(
         },
       ],
     };
+
+    // Add interactive screenshot viewer UI
+    const uiResource = createUIResource(
+      `ui://appium-mcp/screenshot-viewer/${Date.now()}`,
+      createScreenshotViewerUI(screenshotBase64, filepath)
+    );
+
+    return addUIResourceToResponse(textResponse, uiResource);
   } catch (err: any) {
     return {
       content: [

package/src/tools/session/create-session.ts

@@ -19,6 +19,11 @@ import {
 } from './select-device.js';
 import { IOSManager } from '../../devicemanager/ios-manager.js';
 import log from '../../logger.js';
+import {
+  createUIResource,
+  createSessionDashboardUI,
+  addUIResourceToResponse,
+} from '../../ui/mcp-ui-utils.js';
 
 // Define capabilities type
 interface Capabilities {
@@ -250,18 +255,39 @@ export default function createSession(server: any): void {
         const sessionId = await createDriverSession(driver, finalCapabilities);
         setSession(driver, sessionId);
 
+        // Safely convert sessionId to string for display
+        const sessionIdStr =
+          typeof sessionId === 'string'
+            ? sessionId
+            : String(sessionId || 'Unknown');
+
         log.info(
-          `${platform.toUpperCase()} session created successfully with ID: ${sessionId}`
+          `${platform.toUpperCase()} session created successfully with ID: ${sessionIdStr}`
         );
 
-        return {
+        const textResponse = {
           content: [
             {
               type: 'text',
-              text: `${platform.toUpperCase()} session created successfully with ID: ${sessionId}\nPlatform: ${finalCapabilities.platformName}\nAutomation: ${finalCapabilities['appium:automationName']}\nDevice: ${finalCapabilities['appium:deviceName']}`,
+              text: `${platform.toUpperCase()} session created successfully with ID: ${sessionIdStr}\nPlatform: ${finalCapabilities.platformName}\nAutomation: ${finalCapabilities['appium:automationName']}\nDevice: ${finalCapabilities['appium:deviceName']}`,
             },
           ],
         };
+
+        // Add interactive session dashboard UI
+        const uiResource = createUIResource(
+          `ui://appium-mcp/session-dashboard/${sessionIdStr}`,
+          createSessionDashboardUI({
+            sessionId: sessionIdStr,
+            platform: finalCapabilities.platformName,
+            automationName: finalCapabilities['appium:automationName'],
+            deviceName: finalCapabilities['appium:deviceName'],
+            platformVersion: finalCapabilities['appium:platformVersion'],
+            udid: finalCapabilities['appium:udid'],
+          })
+        );
+
+        return addUIResourceToResponse(textResponse, uiResource);
       } catch (error: any) {
         log.error('Error creating session:', error);
         throw new Error(`Failed to create session: ${error.message}`);

package/src/tools/session/select-device.ts

@@ -5,6 +5,11 @@ import { ADBManager } from '../../devicemanager/adb-manager.js';
 import { IOSManager } from '../../devicemanager/ios-manager.js';
 import { z } from 'zod';
 import log from '../../logger.js';
+import {
+  createUIResource,
+  createDevicePickerUI,
+  addUIResourceToResponse,
+} from '../../ui/mcp-ui-utils.js';
 
 // Store selected device globally
 let selectedDeviceUdid: string | null = null;
@@ -92,7 +97,7 @@ function formatAndroidListResponse(devices: any[]): any {
     .map((device, index) => `  ${index + 1}. ${device.udid}`)
     .join('\n');
 
-  return {
+  const textResponse = {
     content: [
       {
         type: 'text',
@@ -100,6 +105,14 @@ function formatAndroidListResponse(devices: any[]): any {
       },
     ],
   };
+
+  // Add interactive UI picker
+  const uiResource = createUIResource(
+    `ui://appium-mcp/device-picker/android-${Date.now()}`,
+    createDevicePickerUI(devices, 'android')
+  );
+
+  return addUIResourceToResponse(textResponse, uiResource);
 }
 
 /**
@@ -202,7 +215,7 @@ function formatIOSListResponse(
     )
     .join('\n');
 
-  return {
+  const textResponse = {
     content: [
       {
         type: 'text',
@@ -210,6 +223,14 @@ function formatIOSListResponse(
       },
     ],
   };
+
+  // Add interactive UI picker
+  const uiResource = createUIResource(
+    `ui://appium-mcp/device-picker/ios-${iosDeviceType}-${Date.now()}`,
+    createDevicePickerUI(devices, 'ios', iosDeviceType)
+  );
+
+  return addUIResourceToResponse(textResponse, uiResource);
 }
 
 /**

package/src/tools/test-generation/locators.ts

@@ -13,6 +13,11 @@ import { z } from 'zod';
 import { getDriver } from '../../session-store.js';
 import { generateAllElementLocators } from '../../locators/generate-all-locators.js';
 import log from '../../logger.js';
+import {
+  createUIResource,
+  createLocatorGeneratorUI,
+  addUIResourceToResponse,
+} from '../../ui/mcp-ui-utils.js';
 
 export default function generateLocators(server: any): void {
   server.addTool({
@@ -56,7 +61,8 @@ export default function generateLocators(server: any): void {
               fetchableOnly: true,
             }
           );
-          return {
+
+          const textResponse = {
             content: [
               {
                 type: 'text',
@@ -69,6 +75,14 @@ export default function generateLocators(server: any): void {
               },
             ],
           };
+
+          // Add interactive locator generator UI
+          const uiResource = createUIResource(
+            `ui://appium-mcp/locator-generator/${Date.now()}`,
+            createLocatorGeneratorUI(interactableElements)
+          );
+
+          return addUIResourceToResponse(textResponse, uiResource);
         } catch (parseError: any) {
           log.error('Error parsing XML:', parseError);
           throw new Error(`Failed to parse XML: ${parseError.message}`);