Update app.py

app.py

@@ -44,68 +44,63 @@ def has_camera_exif(image):
     return False
 
 # ----------------------------
-# DETEKSI HYBRID
+# DETEKSI HYBRID DENGAN PERSENTASE
 # ----------------------------
 def detect_image(image):
-    # Prediksi HF AI-detector
     hf_score = 0
-    hf_label = "N/A"
-    hf_conf = 0
+    general_score = 0
+    local_score = 0
+
+    # -------- HF AI-detector --------
+    hf_label, hf_conf = "N/A", 0
     if hf_detector:
         try:
             result = hf_detector(image)
             hf_label = result[0]['label']
             hf_conf = result[0]['score'] * 100
+            # Jika label mengandung tanda AI → skor AI = confidence
             if any(x in hf_label.lower() for x in ["fake", "ai", "artificial"]):
                 hf_score = hf_conf
-            else:
-                hf_score = 100 - hf_conf  # jika human, kurangi
         except:
             hf_score = 0
 
-    # Prediksi general model
-    general_score = 0
-    general_label = "N/A"
-    general_conf = 0
+    # -------- General model --------
+    general_label, general_conf = "N/A", 0
     if general_model:
         try:
             result2 = general_model(image)
             general_label = result2[0]['label']
             general_conf = result2[0]['score'] * 100
             if any(x in general_label.lower() for x in ["anime","cartoon","illustration","maya","3d"]):
-                general_score = general_conf
+                general_score = general_conf * 0.7  # dikurangi bobot karena cadangan
         except:
             general_score = 0
 
-    # Analisis lokal
+    # -------- Analisis lokal --------
     blur_score = calculate_blur(image)
     noise_score = calculate_noise(image)
     exif_present = has_camera_exif(image)
-    local_score = 0
 
-    # Blur rendah atau noise rendah → kemungkinan AI
     if blur_score < 100 or noise_score < 10:
         local_score += 50
-    # Tidak ada metadata kamera → kemungkinan AI
     if not exif_present:
-        local_score += 40
-    else:
-        local_score -= 20  # ada metadata kamera → lebih asli
+        local_score += 10
 
-    # Weighted hybrid
-    weighted_score = hf_score*0.4 + general_score*0.2 + local_score*0.4
+    # -------- Weighted Score --------
+    total_ai_score = hf_score*0.7 + general_score*0.2 + local_score*0.1
+    total_ai_score = min(max(total_ai_score, 0), 100)  # clamp 0–100
+    total_real_score = 100 - total_ai_score
 
-    # Konversi ke persen AI / Asli
-    if weighted_score >= 95:
-        final_result = "🖼️ Gambar ini hasil AI (100%)"
-    elif weighted_score <= 5:
-        final_result = "🖼️ Gambar ini asli (100%)"
+    # -------- Output --------
+    if total_ai_score == 100:
+        final_text = "🤖 Gambar ini hasil AI"
+    elif total_real_score == 100:
+        final_text = "✅ Gambar ini asli"
     else:
-        final_result = f"🖼️ Gambar ini {round(weighted_score,2)}% AI / {round(100-weighted_score,2)}% Asli"
+        final_text = f"🖼️ {total_ai_score:.2f}% AI / {total_real_score:.2f}% Asli"
 
-    # Output
     output_lines = [
-        f"### Hasil Deteksi:\n{final_result}",
+        f"### Hasil Deteksi:\n{final_text}",
         f"HF AI-detector: {hf_label} ({hf_conf:.2f}%)",
         f"General Model: {general_label} ({general_conf:.2f}%)",
         f"Blur Score: {blur_score:.2f}",
@@ -122,8 +117,8 @@ iface = gr.Interface(
     fn=detect_image,
     inputs=gr.Image(type="pil"),
     outputs="markdown",
-    title="AI vs Foto Asli Detector (Hybrid)",
-    description="Unggah gambar, sistem akan mendeteksi persentase AI vs Asli."
+    title="Hybrid AI vs Foto Asli Detector (Gratis)",
+    description="Unggah gambar, sistem akan mendeteksi persentase AI dan persentase asli. Tidak ada kategori tidak pasti."
 )
 
 if __name__ == "__main__":