Ultralytics yolo support + some fixes

Author: stranavad

.idea/API.iml

@@ -355,6 +355,14 @@ def delete_nn(self, sensor_name: str):
         if not isinstance(self.pipeline, NNPipeline):
             return
 
+        # Clear nn_config on the current sensor BEFORE the pipeline restart.
+        # reload_sensors() propagates the old sensor's nn_config to the new
+        # one on open(), so without this GET /nn keeps returning the deployed
+        # config after delete — client shows a stale "NN" badge.
+        existing = self.sensors.get(sensor_name)
+        if existing is not None:
+            existing.nn_config = None
+
         pipeline = self.pipeline
         self.close()
         pipeline.remove_nn(sensor_name)

robopipe_api/camera/camera.py

@@ -138,10 +138,10 @@ def hello_robopipe():
     return PlainTextResponse("Hello from Robopipe API!")
 
 
-@app.exception_handler(Exception)
-def global_exception_handler(request: Request, exc: Exception):
-    if isinstance(exc, (CameraNotFoundException, SensorNotFoundException)):
-        return JSONResponse(status_code=404, content=None)
+@app.exception_handler(CameraNotFoundException)
+@app.exception_handler(SensorNotFoundException)
+def not_found_exception_handler(request: Request, exc: Exception):
+    return JSONResponse(status_code=404, content=None)
 
 
 def app_setup():