from ultralytics import YOLO
import gradio as gr
from PIL import Image

# โหลดโมเดลที่เทรนไว้
model = YOLO("pokemon_best.pt")

def detect(image):
    results = model(image)
    return results[0].plot()  # คืนภาพพร้อมกล่องตรวจจับ

# สร้าง Gradio UI
demo = gr.Interface(
    fn=detect,
    inputs=gr.Image(type="pil", label="อัปโหลดภาพโปเกมอน"),
    outputs=gr.Image(label="ผลลัพธ์การตรวจจับ"),
    title="Pokémon Object Detection (YOLOv8)",
    description="อัปโหลดภาพโปเกมอน แล้วระบบจะตรวจจับว่ามีโปเกมอนตัวไหนบ้างในภาพ"
)

demo.launch()