Update handler.py

handler.py

@@ -1,7 +1,7 @@
 import base64
 import io
 import os
-from typing import Dict, Any
+from typing import Dict, Any, List
 
 import torch
 from PIL import Image
@@ -81,11 +81,14 @@ class EndpointHandler:
     # --------------------------------------------------
     # 3) الدالة الرئيسة
     # --------------------------------------------------
-    def __call__(self, data: Any) -> Dict[str, float]:
+    def __call__(self, data: Any) -> List[Dict[str, Any]]:
         """
         يدعم:
         • Widget (PIL.Image)
         • REST (base64 فى data["inputs"] أو data["image"])
+        
+        يعيد:
+        • مصفوفة من القواميس بتنسيق [{label: string, score: number}, ...]
         """
         img: Image.Image | None = None
 
@@ -99,10 +102,21 @@ class EndpointHandler:
                 img = self._decode_b64(payload)
 
         if img is None:
-            return {"error": "No image provided"}
+            return [{"label": "error", "score": 1.0}]
 
         with torch.no_grad():
             logits = self.model(self._img_to_tensor(img))
             probs = torch.nn.functional.softmax(logits.squeeze(0), dim=0)
 
-        return {self.labels[i]: float(probs[i]) for i in range(len(self.labels))}
+        # تحويل النتائج إلى التنسيق المطلوب: Array<label: string, score:number>
+        results = []
+        for i, label in enumerate(self.labels):
+            results.append({
+                "label": label,
+                "score": float(probs[i])
+            })
+        
+        # ترتيب النتائج تنازلياً حسب درجة الثقة
+        results.sort(key=lambda x: x["score"], reverse=True)
+        
+        return results