Add CI and fix screenshot selection

Author: AI Assistant

.github/workflows/agentux-ci.yml

@@ -0,0 +1,28 @@
+name: AgentUX CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+
+      - name: Setup Node.js
+        uses: actions/setup-node@v4
+        with:
+          node-version: "20"
+          cache: npm
+
+      - name: Install dependencies
+        run: npm ci
+
+      - name: Compile
+        run: npm run compile
+
+      - name: Unit tests
+        run: npm run test:unit

README.md

@@ -1,5 +1,7 @@
 # AgentUX — VS Code Extension for Screenshot UX Audits
 
+[![AgentUX CI](https://github.com/ai-craftsman404/AgentUX/actions/workflows/agentux-ci.yml/badge.svg)](https://github.com/ai-craftsman404/AgentUX/actions/workflows/agentux-ci.yml)
+
 AgentUX reviews static UI screenshots with OpenAI Vision, a strict JSON pipeline, and an interactive VS Code webview. It surfaces spacing, typography, contrast, navigation, interaction, and design-system insights backed by attention heatmaps and region overlays.
 
 ## Two Distinct Output Formats

src/commands/analyzeScreenshot.ts

@@ -171,30 +171,47 @@ const resolveLocalPath = (uri?: vscode.Uri): string | null => {
     return null;
   }
 
-  const normalized = path.normalize(fsPath);
-  return fs.existsSync(normalized) ? normalized : null;
+  return path.normalize(fsPath);
 };
 
 export const analyzeScreenshot = async (
   context: vscode.ExtensionContext,
   fileUri?: vscode.Uri,
 ): Promise<void> => {
   try {
-    let screenshotPath: string | null = resolveLocalPath(fileUri);
+    const INVALID_SELECTION_MESSAGE =
+      'Invalid file selection. Please select a local image file.';
 
-    if (!screenshotPath) {
-      screenshotPath = resolveLocalPath(getActiveImageEditor());
-    }
-
-    if (!screenshotPath) {
-      screenshotPath = resolveLocalPath(await pickScreenshot());
-    }
+    let screenshotPath: string | null = null;
 
-    if (!screenshotPath) {
-      vscode.window.showErrorMessage(
-        'Invalid or unreadable screenshot. Please select a local image file.',
-      );
-      return;
+    // If invoked from context menu with a URI, never fall back to the file picker.
+    if (fileUri) {
+      const candidate = resolveLocalPath(fileUri);
+      if (!candidate || !fs.existsSync(candidate)) {
+        vscode.window.showErrorMessage(INVALID_SELECTION_MESSAGE);
+        return;
+      }
+      screenshotPath = candidate;
+    } else {
+      // If an image is open in the active editor, use it (and do not show the picker).
+      const activeUri = getActiveImageEditor();
+      if (activeUri) {
+        const candidate = resolveLocalPath(activeUri);
+        if (!candidate || !fs.existsSync(candidate)) {
+          vscode.window.showErrorMessage(INVALID_SELECTION_MESSAGE);
+          return;
+        }
+        screenshotPath = candidate;
+      } else {
+        // Otherwise, prompt the user to pick a file.
+        const pickedUri = await pickScreenshot();
+        const candidate = resolveLocalPath(pickedUri);
+        if (!candidate || !fs.existsSync(candidate)) {
+          vscode.window.showErrorMessage(INVALID_SELECTION_MESSAGE);
+          return;
+        }
+        screenshotPath = candidate;
+      }
     }
 
     const metadata = await resolveMetadata();