from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline

class EndpointHandler:
    def __init__(self, path=""):
        self.model = AutoModelForSequenceClassification.from_pretrained(
            path,
            ignore_mismatched_sizes=True  # مهم جدًا
        )
        self.tokenizer = AutoTokenizer.from_pretrained(path)
        self.pipeline = pipeline("text-classification", model=self.model, tokenizer=self.tokenizer)

    def __call__(self, data):
        try:
            inputs = data.get("inputs", "") if isinstance(data, dict) else data

            if not isinstance(inputs, str) or not inputs.strip():
                return [{"error": "Input must be a non-empty string."}]  # نرجعها داخل List

            result = self.pipeline(inputs)

            # نتاكد إنه list
            return result

        except Exception as e:
            return [{"error": str(e)}]  # نرجع الخطأ داخل List