Update app.py

app.py

@@ -47,31 +47,41 @@ def has_camera_exif(image):
 # DETEKSI HYBRID
 # ----------------------------
 def detect_image(image):
+    output_lines = []
+
+    # -------- HF AI-detector --------
     hf_score = 0
+    hf_label = "N/A"
+    hf_conf = 0
     if hf_detector:
         try:
             result = hf_detector(image)
-            label = result[0]['label']
-            conf = result[0]['score'] * 100
-            if "fake" in label.lower() or "artificial" in label.lower():
-                hf_score = conf
-            elif "human" in label.lower():
-                hf_score = 100 - conf  # human confidence rendah → kemungkinan AI
+            hf_label = result[0]['label']
+            hf_conf = result[0]['score'] * 100
+            if any(x in hf_label.lower() for x in ["fake", "ai", "artificial"]):
+                hf_score = hf_conf
+            elif "human" in hf_label.lower():
+                hf_score = 100 - hf_conf  # semakin besar human → kecil skor AI
         except:
             hf_score = 0
 
+    # -------- General model --------
     general_score = 0
+    general_label = "N/A"
+    general_conf = 0
     if general_model:
         try:
             result2 = general_model(image)
-            label2 = result2[0]['label']
-            conf2 = result2[0]['score'] * 100
-            if any(x in label2.lower() for x in ["anime","cartoon","illustration","3d","maya"]):
-                general_score = conf2
+            general_label = result2[0]['label']
+            general_conf = result2[0]['score'] * 100
+            if any(x in general_label.lower() for x in ["anime","cartoon","illustration","3d","maya","render"]):
+                general_score = general_conf
+            else:
+                general_score = 100 - general_conf
         except:
             general_score = 0
 
-    # Analisis lokal
+    # -------- Analisis lokal --------
     blur_score = calculate_blur(image)
     noise_score = calculate_noise(image)
     exif_present = has_camera_exif(image)
@@ -81,29 +91,27 @@ def detect_image(image):
     if not exif_present:
         local_score += 10
 
-    # Weighted score
+    # -------- Weighted Score --------
     weighted_score = hf_score*0.6 + general_score*0.2 + local_score*0.2
-    weighted_score = min(max(weighted_score, 0), 100)  # clamp 0-100
+    weighted_score = min(max(weighted_score, 0), 100)  # pastikan 0–100
 
-    # Output final
+    # -------- Output Final --------
     if weighted_score >= 95:
-        final_result = f"⚠️ Gambar ini hasil AI ({weighted_score:.2f}%)"
+        final_result = "🤖 Gambar ini hasil AI"
     elif weighted_score <= 5:
-        final_result = f"✅ Gambar ini asli ({100-weighted_score:.2f}%)"
+        final_result = "✅ Gambar ini asli"
     else:
-        final_result = f"AI: {weighted_score:.2f}%, Asli: {100-weighted_score:.2f}%"
+        final_result = f"⚖️ Gambar ini {weighted_score:.2f}% AI, {100-weighted_score:.2f}% asli"
 
-    output = f"""
-### Hasil Deteksi:
-{final_result}
+    output_lines.append(f"### Hasil Deteksi:\n{final_result}")
+    output_lines.append(f"Weighted Skor: {weighted_score:.2f}")
+    output_lines.append(f"HF AI-detector: {hf_label} ({hf_conf:.2f}%)")
+    output_lines.append(f"General Model: {general_label} ({general_conf:.2f}%)")
+    output_lines.append(f"Blur Score: {blur_score:.2f}")
+    output_lines.append(f"Noise Score: {noise_score:.2f}")
+    output_lines.append(f"Metadata Kamera: {'Ada' if exif_present else 'Tidak Ada'}")
 
-HF AI-detector: {label if hf_detector else 'N/A'} ({conf if hf_detector else 0:.2f}%)
-General Model: {label2 if general_model else 'N/A'} ({conf2 if general_model else 0:.2f}%)
-Blur Score: {blur_score:.2f}
-Noise Score: {noise_score:.2f}
-Metadata Kamera: {'Ada' if exif_present else 'Tidak Ada'}
-"""
-    return output
+    return "\n".join(output_lines)
 
 # ----------------------------
 # Gradio Interface
@@ -113,7 +121,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs="markdown",
     title="AI vs Foto Asli Detector (Hybrid)",
-    description="Unggah gambar, sistem akan mendeteksi apakah gambar kemungkinan besar asli atau dihasilkan AI."
+    description="Unggah gambar, sistem akan mendeteksi persentase AI vs asli. 100% AI → hasil AI, 100% asli → asli."
 )
 
 if __name__ == "__main__":