delete api

app.py

@@ -250,6 +250,9 @@
 # if __name__ == "__main__":
 #     demo.launch()
 
+
+
+
 import gradio as gr
 import cv2
 import numpy as np
@@ -571,16 +574,18 @@ class EnginePartDetector:
 
         # Layer 3: Latent Space Matching (Cosine Similarity to centroids)
         class_scores = []
-        for name, vectors in self.classes.items():
-            # Calculate centroid (neighborhood center)
-            centroid = np.mean(vectors, axis=0)
-            sim = self._cosine(query_feat, centroid)
-            class_scores.append((name, sim))
-        
-        class_scores.sort(key=lambda x: x[1], reverse=True)
-
-        best_class, best_score = class_scores[0]
-        matched = best_score >= threshold
+        if self.classes:
+            for name, vectors in self.classes.items():
+                # Calculate centroid (neighborhood center)
+                centroid = np.mean(vectors, axis=0)
+                sim = self._cosine(query_feat, centroid)
+                class_scores.append((name, sim))
+            
+            class_scores.sort(key=lambda x: x[1], reverse=True)
+            best_class, best_score = class_scores[0]
+            matched = best_score >= threshold
+        else:
+            best_class, best_score, matched = "UNCONFIGURED", 0.0, False
         status  = f"✅ CLASSIFIED AS: {best_class}" if matched else "❌ UNCERTAIN (below threshold)"
 
         lines = [
@@ -626,6 +631,17 @@ class EnginePartDetector:
             body.append(f"  • {name}: {len(vectors)} samples")
         return f"{header}\n" + "\n".join(body)
 
+    def delete_class(self, class_name: str) -> bool:
+        """Remove a class and all its samples from the detector."""
+        if class_name in self.classes:
+            del self.classes[class_name]
+            if class_name in self.class_rois:
+                del self.class_rois[class_name]
+            self._persist_data()
+            logger.info(f"Class '{class_name}' deleted from detector.")
+            return True
+        return False
+
 # ───────────────────────────────────────────────────────────────────────────────
 # Gradio Application
 # ───────────────────────────────────────────────────────────────────────────────
@@ -641,6 +657,12 @@ def add_sample(image, class_name):
 def list_classes():
     return detector.list_templates()
 
+def delete_class_api(class_name: str):
+    success = detector.delete_class(class_name)
+    if success:
+        return f"✅ Class '{class_name}' deleted successfully.", None
+    return f"❌ Class '{class_name}' not found.", None
+
 # Custom CSS for premium look
 custom_css = """
 .container { max-width: 1200px; margin: auto; }
@@ -678,6 +700,15 @@ with gr.Blocks(title="Engine Part CV System") as demo:
             api_name="detect_part",
         )
 
+        # Deletion API (Hidden from UI but accessible via API)
+        delete_btn_api = gr.Button("Delete Class (Internal)", visible=False)
+        delete_btn_api.click(
+            fn=delete_class_api,
+            inputs=[gr.Textbox(visible=False)], # Dummy input if needed, though often just call via API
+            outputs=[gr.Markdown()],
+            api_name="delete_class"
+        )
+
     with gr.Tab("💾 Train Latent Space"):
         with gr.Row():
             with gr.Column(scale=1):