app.py

import streamlit as st
from PIL import Image
import requests
import torch
from torchvision import transforms
from torchvision.models import resnet18

# Judul aplikasi
st.title("🔍 Deteksi Gambar AI vs Foto Asli")

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

if uploaded_file is not None:
    image = Image.open(uploaded_file).convert("RGB")
    st.image(image, caption="Gambar yang Diupload", use_column_width=True)

    if st.button("Deteksi Apakah Gambar Ini AI?"):
        st.write("🔄 Memproses...")

        # Gunakan model dummy dulu (untuk testing)
        # Nanti bisa ganti dengan model pendeteksi AI

        # Sementara: deteksi sederhana pakai metadata
        if "AIPrompt" in uploaded_file.name or "generated" in uploaded_file.name.lower():
            st.error("❌ Gambar ini kemungkinan besar dibuat oleh AI.")
        else:
            st.success("✅ Gambar ini kemungkinan besar adalah foto asli.")

        st.caption("Catatan: Ini adalah demo sederhana. Versi penuh akan menggunakan model AI untuk pendeteksian lebih akurat.")