Update handler.py

handler.py

@@ -21,7 +21,7 @@ class EndpointHandler:
                 task="image-classification",
                 model="yaya36095/ai-source-detector",
                 device=0 if torch.cuda.is_available() else -1,
-                torch_dtype=torch.float16  # يُفضل فقط لو الجهاز يدعم
+                torch_dtype=torch.float16
             )
 
             print("تم تحميل النموذج بنجاح")
@@ -39,27 +39,41 @@ class EndpointHandler:
                 print(f"فشلت المحاولة البديلة أيضًا: {e2}")
                 raise
 
-    def _decode_b64(self, b: bytes) -> Image.Image:
-        try:
-            print(f"فك ترميز base64. حجم البيانات: {len(b)} بايت")
-            return Image.open(io.BytesIO(base64.b64decode(b))).convert("RGB")
-        except Exception as e:
-            print(f"خطأ في فك الترميز: {e}")
-            raise
-
     def __call__(self, data: Any) -> List[Dict[str, Any]]:
         print(f"استدعاء __call__ مع نوع البيانات: {type(data)}")
 
         img = None
         try:
             if isinstance(data, Image.Image):
+                print("البيانات هي صورة PIL")
                 img = data
+
             elif isinstance(data, dict):
                 payload = data.get("inputs") or data.get("image")
-                if isinstance(payload, str):
-                    payload = payload.encode()
-                if isinstance(payload, bytes):
-                    img = self._decode_b64(payload)
+
+                if isinstance(payload, Image.Image):
+                    img = payload
+
+                elif isinstance(payload, bytes):
+                    try:
+                        img = Image.open(io.BytesIO(payload)).convert("RGB")
+                        print("تم فتح الصورة من Bytes بنجاح")
+                    except Exception as e:
+                        print(f"فشل في فتح الصورة من Bytes: {e}")
+
+                elif isinstance(payload, str):
+                    try:
+                        # محاولة فك base64
+                        try:
+                            img = Image.open(io.BytesIO(base64.b64decode(payload))).convert("RGB")
+                            print("تم فتح الصورة من base64 بنجاح")
+                        except Exception:
+                            # لو فشل، نحاول اعتباره مسار ملف محلي
+                            if os.path.exists(payload):
+                                img = Image.open(payload).convert("RGB")
+                                print("تم فتح الصورة من مسار محلي بنجاح")
+                    except Exception as e:
+                        print(f"فشل في فتح الصورة من النص: {e}")
 
             if img is None:
                 print("لم يتم العثور على صورة صالحة")