Update app.py

app.py

@@ -1,45 +1,25 @@
+import streamlit as st
 from transformers import pipeline
-import gradio as gr
-from PIL import ExifTags
+from PIL import Image
 
-# Pakai model publik (gratis, tanpa login)
-classifier = pipeline("image-classification", model="orpatashnik/image-real-fake")
+# Gunakan model AI detector (gratis di HuggingFace)
+classifier = pipeline("image-classification", model="shuhuai/AI-image-detector")
 
-def detect_image(image):
-    # --- Cek metadata (EXIF) ---
-    exif_result = "⚠️ Tidak ada metadata kamera"
-    try:
-        raw_exif = image._getexif()
-        if raw_exif:
-            exif_result = "✅ Metadata kamera terdeteksi"
-    except:
-        pass
+st.title("Deteksi Foto Asli vs AI Generated")
+st.write("Upload foto untuk mengecek apakah asli atau AI generated.")
 
-    # --- Prediksi model ---
-    results = classifier(image)
-    label = results[0]["label"]
-    score = results[0]["score"] * 100
+uploaded_file = st.file_uploader("Upload gambar", type=["jpg","jpeg","png"])
 
-    if "fake" in label.lower():
-        verdict = "🚨 Kemungkinan besar Hasil AI"
-    else:
-        verdict = "📷 Kemungkinan besar Foto Asli"
-
-    return f"""{verdict}
+if uploaded_file:
+    image = Image.open(uploaded_file)
+    st.image(image, caption="Gambar yang diupload", use_column_width=True)
 
-Label Model: {label}
-Confidence: {score:.2f}%
+    result = classifier(image)
 
-Cek Metadata: {exif_result}
-"""
+    label = result[0]['label']
+    confidence = result[0]['score'] * 100
 
-# Gradio app
-iface = gr.Interface(
-    fn=detect_image,
-    inputs=gr.Image(type="pil"),
-    outputs="text",
-    title="AI Image Detector (Gratis)",
-    description="Upload gambar untuk mendeteksi apakah foto asli atau hasil AI"
-)
-
-iface.launch()
+    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}%")