Update app.py

app.py

@@ -44,30 +44,45 @@ def has_camera_exif(image):
     return False
 
 # ----------------------------
-# DETEKSI HYBRID WEIGHTED
+# DETEKSI HYBRID HF PRIORITAS
 # ----------------------------
 def detect_image(image):
-    hf_score = 0
-    general_score = 0
-    local_score = 0
+    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'].lower()
-            conf = result[0]['score'] * 100
-            hf_score = conf if any(x in label for x in ["fake", "ai", "artificial"]) else 0
+            hf_label = result[0]['label']
+            hf_conf = result[0]['score'] * 100
+            # tentukan HF score AI relevan
+            if any(x in hf_label.lower() for x in ["fake", "ai", "artificial"]):
+                hf_score = hf_conf
         except:
             hf_score = 0
 
+    # Jika HF confidence tinggi → langsung AI
+    if hf_score > 60:
+        final_result = "🤖 AI Detected"
+        weighted_score = hf_score
+        output_lines.append(f"### Hasil Deteksi:\n{final_result}")
+        output_lines.append(f"HF AI-detector: {hf_label} ({hf_conf:.2f}%)")
+        return "\n".join(output_lines)
+
     # -------- General model --------
+    general_score = 0
+    general_label = "N/A"
+    general_conf = 0
     if general_model:
         try:
             result2 = general_model(image)
-            label2 = result2[0]['label'].lower()
-            conf2 = result2[0]['score'] * 100
-            general_score = conf2 if any(x in label2 for x in ["anime","cartoon","illustration","maya"]) else 0
+            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"]):
+                general_score = general_conf
         except:
             general_score = 0
 
@@ -81,26 +96,24 @@ def detect_image(image):
     if not exif_present:
         local_score += 10
 
-    # -------- Weighted Score --------
-    final_score = hf_score*0.6 + general_score*0.25 + local_score*0.15
+    # -------- Weighted Score (general + local) --------
+    weighted_score = general_score*0.6 + local_score*0.4
 
-    if final_score > 50:
+    if weighted_score > 50:
         final_result = "🤖 AI Detected"
     else:
         final_result = "✅ Foto Asli"
 
-    output = f"""
-### Hasil Deteksi:
-{final_result}
+    # -------- Output --------
+    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'}")
 
-**Weighted Skor:** {final_score:.2f}
-**HF AI-detector:** {result[0]['label']} ({result[0]['score']*100:.2f}%)
-**General Model:** {result2[0]['label']} ({result2[0]['score']*100:.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
@@ -109,8 +122,8 @@ iface = gr.Interface(
     fn=detect_image,
     inputs=gr.Image(type="pil"),
     outputs="markdown",
-    title="AI vs Foto Asli Detector (Weighted Hybrid)",
-    description="Unggah gambar, sistem hybrid akan mendeteksi apakah gambar kemungkinan besar asli atau dihasilkan AI."
+    title="AI vs Foto Asli Detector (HF Prioritas)",
+    description="Unggah gambar, sistem akan mendeteksi apakah gambar kemungkinan besar asli atau dihasilkan AI."
 )
 
 if __name__ == "__main__":