@blueharford/scrypted-spatial-awareness 0.6.6 → 0.6.7

package/out/main.nodejs.js

@@ -40054,7 +40054,7 @@ class SpatialAwarenessPlugin extends sdk_1.ScryptedDeviceBase {
             });
         }
     }
-    handleTrainingEndRequest(response) {
+    async handleTrainingEndRequest(response) {
         if (!this.trackingEngine) {
             response.send(JSON.stringify({ error: 'Tracking engine not running' }), {
                 code: 500,
@@ -40064,6 +40064,39 @@ class SpatialAwarenessPlugin extends sdk_1.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' },
             });