Update app.py

app.py

@@ -670,8 +670,12 @@ HTML_TEMPLATE = """
                     `;
                     
                     if (numObstacles && numObstacles !== '0') {
-                        successMessage += `<br>Saved as Obstacle ${numObstacles}`;
-                        successMessage += `<br><a href="/obstacles" target="_blank" style="color: #007bff; text-decoration: underline;">View Obstacles Display</a>`;
+                        if (predictedId === '-1') {
+                            successMessage += `<br><span style="color: #ffc107;">⚠️ Obstacle ${numObstacles} not saved - No valid detection (ID: -1)</span>`;
+                        } else {
+                            successMessage += `<br><span style="color: #28a745;">✓ Saved as Obstacle ${numObstacles}</span>`;
+                            successMessage += `<br><a href="/obstacles" target="_blank" style="color: #007bff; text-decoration: underline;">View Obstacles Display</a>`;
+                        }
                     }
                     
                     successMessage += `</div>`;
@@ -1002,15 +1006,15 @@ OBSTACLES_HTML_TEMPLATE = """
         function updateStatus() {
             const existingCount = Object.values(obstaclesData).filter(o => o && o.exists).length;
             const statusEl = document.getElementById('status');
-            statusEl.textContent = `Displaying ${existingCount}/8 obstacles`;
             
             if (existingCount === 0) {
                 statusEl.style.backgroundColor = '#fff3cd';
                 statusEl.style.color = '#856404';
-                statusEl.textContent = 'No obstacles detected yet. Upload images with NUM_OBSTACLES parameter (1-8).';
+                statusEl.textContent = 'No obstacles detected yet. Upload images with NUM_OBSTACLES parameter (1-8) and valid detections (ID ≠ -1).';
             } else {
                 statusEl.style.backgroundColor = '#d4edda';
                 statusEl.style.color = '#155724';
+                statusEl.textContent = `Displaying ${existingCount}/8 obstacles with valid detections`;
             }
         }
         
@@ -1102,8 +1106,16 @@ def image_predict():
     class_id, detection_result = process_file(file_path, signal, task_type, filename)
     
     # Handle NUM_OBSTACLES parameter for obstacle display
-    if num_obstacles and num_obstacles.isdigit() and 1 <= int(num_obstacles) <= 8:
+    # Only save if detection is valid (image_id != '-1')
+    if (num_obstacles and num_obstacles.isdigit() and 1 <= int(num_obstacles) <= 8 
+        and detection_result and detection_result.get('image_id') != '-1'):
         save_obstacle_image(detection_result, int(num_obstacles))
+        print(f"Obstacle {num_obstacles} saved with valid detection (ID: {detection_result.get('image_id')})")
+    elif (num_obstacles and num_obstacles.isdigit() and 1 <= int(num_obstacles) <= 8 
+          and detection_result and detection_result.get('image_id') == '-1'):
+        print(f"Obstacle {num_obstacles} NOT saved - invalid detection (ID: -1)")
+    elif num_obstacles and num_obstacles.isdigit() and 1 <= int(num_obstacles) <= 8:
+        print(f"Obstacle {num_obstacles} NOT saved - no detection result")
     
     if detection_result is None:
         return jsonify({'error': 'Failed to process image'}), 500