@blueharford/scrypted-spatial-awareness 0.6.6 → 0.6.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@blueharford/scrypted-spatial-awareness",
-  "version": "0.6.6",
+  "version": "0.6.7",
   "description": "Cross-camera object tracking for Scrypted NVR with spatial awareness",
   "author": "Joshua Seidel <blueharford>",
   "license": "Apache-2.0",

package/src/main.ts

@@ -1584,7 +1584,7 @@ export class SpatialAwarenessPlugin extends ScryptedDeviceBase
     }
   }
 
-  private handleTrainingEndRequest(response: HttpResponse): void {
+  private async handleTrainingEndRequest(response: HttpResponse): Promise<void> {
     if (!this.trackingEngine) {
       response.send(JSON.stringify({ error: 'Tracking engine not running' }), {
         code: 500,
@@ -1595,6 +1595,44 @@ export class SpatialAwarenessPlugin extends ScryptedDeviceBase
 
     const session = this.trackingEngine.endTrainingSession();
     if (session) {
+      // Get unique visited cameras
+      const visitedCameraIds = [...new Set(session.visits.map(v => v.cameraId))];
+
+      // Auto-run discovery on visited cameras to detect landmarks and zones
+      if (this.discoveryEngine && visitedCameraIds.length > 0) {
+        this.console.log(`[Training] Running discovery analysis on ${visitedCameraIds.length} visited cameras...`);
+
+        let landmarksFound = 0;
+        let zonesFound = 0;
+
+        for (const cameraId of visitedCameraIds) {
+          try {
+            const analysis = await this.discoveryEngine.analyzeScene(cameraId);
+            if (analysis.isValid) {
+              landmarksFound += analysis.landmarks.length;
+              zonesFound += analysis.zones.length;
+              this.console.log(`[Training] ${cameraId}: Found ${analysis.landmarks.length} landmarks, ${analysis.zones.length} zones`);
+            }
+          } catch (e) {
+            this.console.warn(`[Training] Failed to analyze ${cameraId}:`, e);
+          }
+        }
+
+        // Get all pending suggestions and auto-accept them
+        const suggestions = this.discoveryEngine.getPendingSuggestions();
+        for (const suggestion of suggestions) {
+          this.applyDiscoverySuggestion(suggestion);
+          this.discoveryEngine.acceptSuggestion(suggestion.id);
+        }
+
+        // Persist topology after applying suggestions
+        if (suggestions.length > 0 && this.trackingEngine) {
+          const updatedTopology = this.trackingEngine.getTopology();
+          await this.storageSettings.putSetting('topology', JSON.stringify(updatedTopology));
+          this.console.log(`[Training] Auto-applied ${suggestions.length} discoveries (${landmarksFound} landmarks, ${zonesFound} zones)`);
+        }
+      }
+
       response.send(JSON.stringify(session), {
         headers: { 'Content-Type': 'application/json' },
       });