added delete image

app/data.py

@@ -93,6 +93,18 @@ class Data:
             self.data[image]["labels"] = []
         self._save_data()
 
+    def delete_image(self, image: str) -> None:
+        if image in self.data:
+            if "image" in self.data[image]:
+                Path(self.data[image]["image"]).unlink(missing_ok=True)
+            if "emb" in self.data[image]:
+                Path(self.data[image]["emb"]).unlink(missing_ok=True)
+            if "masks" in self.data[image]:
+                for label_path in self.data[image]["masks"]:
+                    Path(label_path).unlink(missing_ok=True)
+            del self.data[image]
+            self._save_data()
+
     def get_all_images(self) -> list:
         return list(self.data.keys())
 

app/main.py

@@ -273,6 +273,15 @@ async def upload_image(image: str, image_data: ImageData) -> Response:
     return Response(content=image_id, media_type="text/plain")
 
 
+@app.delete("/autolabeler/v1/delete_image/{image}")
+async def delete_image(image: str) -> Response:
+    if image not in DATA:
+        raise HTTPException(status_code=404, detail="Image not found")
+
+    DATA.delete_image(image)
+    return Response(content="deleted", media_type="text/plain")
+
+
 @app.get("/v1/get_all_images")
 async def get_all_images() -> Response:
     return JSONResponse(content={"images": DATA.get_all_images()})

instance-labeler/app/page.tsx

@@ -22,6 +22,18 @@ export default function Home() {
     setDisplayImage(`data:${contentType};base64,${res.data}`);
   }
 
+  const handleFileDelete = async (imageName: string) => {
+    try {
+      const res = await axios.delete(`v1/delete_image/${imageName}`);
+      getAllImages();
+      setDisplayIndex(0);
+      setDisplayImage('/loading.png');
+    } catch (err) {
+      console.log(err);
+      alert('Error deleting image');
+    }
+  }
+
   const handleFileUpload = async (e: ChangeEvent<HTMLInputElement>) => {
     const reader = new FileReader();
     reader.addEventListener("load", async () => {
@@ -66,6 +78,8 @@ export default function Home() {
           }}>Next</button>
       </div>
       <br />
+      <button className="bg-blue-500 hover:bg-blue-700 text-white font-bold py-2 px-2"
+        onClick={() => handleFileDelete(images[displayIndex])}>Delete Image</button>&nbsp;
       <input type="file" id="image" accept="image/png, image/jpeg" onChange={handleFileUpload} />
     </>
   )