Create handler.py

handler.py

@@ -0,0 +1,35 @@
+from PIL import Image
+import base64
+import numpy as np
+from io import BytesIO
+
+class EndpointHandler:
+    def __init__(self, path=""):
+        print("Handler initialized")
+        # Здесь можно загрузить модель и веса
+
+    def __call__(self, data):
+        image_b64 = data.get("image")
+        if image_b64 is None:
+            return {"error": "No image provided"}
+
+        # Декодируем изображение
+        image = Image.open(BytesIO(base64.b64decode(image_b64)))
+        width, height = image.size
+
+        # Генерируем фейковую маску
+        mask = np.zeros((height, width), dtype=np.uint8)
+        mask[height//4: 3*height//4, width//4: 3*width//4] = 255
+
+        # Преобразуем маску в base64 PNG
+        mask_image = Image.fromarray(mask)
+        buffered = BytesIO()
+        mask_image.save(buffered, format="PNG")
+        mask_base64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
+
+        # Возвращаем результат
+        return {
+            "label": "mock-segmentation",
+            "mask": mask_base64,  # ⚠️ mask — строка, закодированная в base64
+            "score": 0.99
+        }