import gradio as gr
import numpy as np
from PIL import Image
from tensorflow.keras.applications.mobilenet_v2 import (
    MobileNetV2,
    preprocess_input,
    decode_predictions
)

# Load model sekali saja
model = MobileNetV2(weights="imagenet")

def predict_image(img):
    if img is None:
        return "Silakan upload gambar terlebih dahulu"

    img = img.convert("RGB")
    img = img.resize((224, 224))

    x = np.array(img)
    x = np.expand_dims(x, axis=0)
    x = preprocess_input(x)

    preds = model.predict(x)
    decoded = decode_predictions(preds, top=5)[0]

    result = "🤖 Hasil Tebakan AI:\n\n"
    for i, p in enumerate(decoded):
        result += f"{i+1}. {p[1]} — {p[2]*100:.2f}%\n"

    return result

demo = gr.Interface(
    fn=predict_image,
    inputs=gr.Image(type="pil"),
    outputs=gr.Textbox(lines=7),
    title="🤖 AI Tebak Gambar",
    description="Upload gambar, AI akan menebak isinya",
    allow_flagging="never"
)

demo.launch()