Update app.py

app.py

@@ -5,6 +5,7 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from collections import defaultdict
 import uvicorn
+import uuid
 import os
 
 app = FastAPI()
@@ -12,12 +13,11 @@ app = FastAPI()
 class InputData(BaseModel):
     image_base64: str
 
-def save_base64_image_cv(base64_str, output_path="final.png"):
+def save_base64_image_cv(base64_str, output_path):
     image_data = base64.b64decode(base64_str)
     nparr = np.frombuffer(image_data, np.uint8)
     img = cv2.imdecode(nparr, cv2.IMREAD_UNCHANGED)
 
-    # jika RGBA → tempelkan ke background putih
     if img is not None and img.shape[2] == 4:
         alpha = img[:, :, 3] / 255.0
         rgb = img[:, :, :3]
@@ -25,9 +25,8 @@ def save_base64_image_cv(base64_str, output_path="final.png"):
         img = (rgb * alpha[:, :, None] + white_bg * (1 - alpha[:, :, None])).astype(np.uint8)
 
     cv2.imwrite(output_path, img)
-    return output_path
 
-def extract_icon_positions(image_path="final.png"):
+def extract_icon_positions(image_path):
     img = cv2.imread(image_path)
     gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
 
@@ -69,29 +68,36 @@ def find_rarest_icon_position(icon_features, positions):
 
     rarest_group = min(groups.values(), key=len)
     idx = rarest_group[0]
-
     return positions[idx][0], positions[idx][1]
 
 @app.post("/predict")
 async def predict(data: InputData):
+
+    # file unik → aman untuk paralel
+    tmp_file = f"{uuid.uuid4().hex}.png"
+
     try:
-        save_base64_image_cv(data.image_base64, "final.png")
+        save_base64_image_cv(data.image_base64, tmp_file)
 
-        icon_features, positions = extract_icon_positions("final.png")
+        icon_features, positions = extract_icon_positions(tmp_file)
 
         if not positions:
             return {"error": "No icons detected"}
 
         x, y = find_rarest_icon_position(icon_features, positions)
 
-        # hapus file temp
-        if os.path.exists("final.png"):
-            os.remove("final.png")
-
-        return {"x": x, "y": y}
+        return {
+            "x": x,
+            "y": y,
+            "count": len(positions)  # opsional
+        }
 
     except Exception as e:
         return {"error": str(e)}
 
+    finally:
+        if os.path.exists(tmp_file):
+            os.remove(tmp_file)
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)