import streamlit as st
from transformers import pipeline
from PIL import Image

# Gunakan model AI detector (gratis di HuggingFace)
classifier = pipeline("image-classification", model="shuhuai/AI-image-detector")

st.title("Deteksi Foto Asli vs AI Generated")
st.write("Upload foto untuk mengecek apakah asli atau AI generated.")

uploaded_file = st.file_uploader("Upload gambar", type=["jpg","jpeg","png"])

if uploaded_file:
    image = Image.open(uploaded_file)
    st.image(image, caption="Gambar yang diupload", use_column_width=True)

    result = classifier(image)

    label = result[0]['label']
    confidence = result[0]['score'] * 100

    if label.lower() in ["fake", "ai-generated"]:
        st.error(f"🚨 Kemungkinan besar **AI Generated**\n\nLabel: {label}\nConfidence: {confidence:.2f}%")
    else:
        st.success(f"✅ Kemungkinan besar **Foto Asli**\n\nLabel: {label}\nConfidence: {confidence:.2f}%")