import gradio as gr
from transformers import AutoImageProcessor, AutoModelForImageClassification
from PIL import Image

# تحميل المعالج والنموذج من Hugging Face Hub
model = AutoModelForImageClassification.from_pretrained("Alhdrawi/x_alhdrawi")
processor = AutoImageProcessor.from_pretrained("Alhdrawi/x_alhdrawi")

# دالة التنبؤ
def predict(image):
    inputs = processor(images=image, return_tensors="pt")
    outputs = model(**inputs)
    logits = outputs.logits
    predicted_class_idx = logits.argmax(-1).item()
    label = model.config.id2label[predicted_class_idx]
    return f"🔍 Prediction: {label}"

# واجهة Gradio
iface = gr.Interface(
    fn=predict,
    inputs=gr.Image(type="pil", label="ارفع صورة شعاعية"),
    outputs=gr.Textbox(label="النتيجة"),
    title="🩻 نموذج التشخيص الشعاعي - Alhdrawi",
    description="نموذج ذكاء اصطناعي لتشخيص الصور الشعاعية مثل X-ray باستخدام Transformers من Hugging Face."
)

iface.launch()