Update app.py

app.py

@@ -1,56 +1,52 @@
 import gradio as gr
 import cv2
+import numpy as np
 from ultralytics import YOLO
 
-# --- CONFIGURATION ---
-# Pastikan file 'best.pt' sudah ada di tab 'Files' di Hugging Face Space Anda
+# --- LOAD MODEL ---
+# Pastikan file 'best.pt' ada di Files Hugging Face
 MODEL_PATH = "best.pt"
-
-print("🔄 Sedang memuat model...")
 try:
     model = YOLO(MODEL_PATH)
     print("✅ Model berhasil dimuat!")
 except Exception as e:
-    print(f"❌ Gagal memuat model: {e}")
     model = None
+    print(f"❌ Error Load Model: {e}")
 
-def detect_objects(frame):
+def detect_manual(image):
     """
-    Fungsi sederhana: Terima gambar -> Deteksi -> Kembalikan gambar hasil deteksi
+    Fungsi ini hanya berjalan saat tombol ditekan.
     """
-    if frame is None:
+    if image is None:
         return None
     
     if model is None:
-        return frame # Kembalikan frame asli jika model gagal load
+        return image # Kembalikan gambar asli jika model rusak
 
-    # Gradio memberikan gambar dalam format RGB, OpenCV/YOLO butuh BGR (kadang otomatis, tapi aman dikonversi)
-    # Namun ultralytics .plot() mengembalikan BGR numpy array, jadi kita perlu bolak-balik warnanya agar pas di web.
-    
-    # 1. Lakukan Prediksi
-    results = model.predict(frame, conf=0.4) # Conf 0.4 agar tidak terlalu sensitif
+    # 1. Deteksi
+    results = model.predict(image, conf=0.4)
     
-    # 2. Gambar Kotak (Bounding Boxes)
-    # plot() mengembalikan array BGR
-    annotated_frame = results[0].plot()
-
-    # 3. Konversi BGR ke RGB agar warnanya benar di Browser
-    output_frame = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
+    # 2. Ambil Plot (Gambar hasil)
+    # Ultralytics mengembalikan BGR, kita perlu convert ke RGB
+    res_plotted = results[0].plot()
+    output_image = cv2.cvtColor(res_plotted, cv2.COLOR_BGR2RGB)
     
-    return output_frame
+    return output_image
 
-# --- GRADIO INTERFACE ---
+# --- TAMPILAN WEB ---
 with gr.Blocks() as demo:
-    gr.Markdown("# 🛠️ Test Mode: YOLO Best.pt Only")
-    gr.Markdown("Kode ini hanya menguji apakah model custom `best.pt` berjalan.")
+    gr.Markdown("# 📸 Test Mode: Foto & Deteksi")
+    gr.Markdown("Cara pakai: Klik ikon kamera untuk ambil foto, lalu tekan tombol 'Mulai Deteksi'.")
     
     with gr.Row():
-        input_cam = gr.Image(sources=["webcam"], streaming=True, label="Kamera Anda")
-        output_cam = gr.Image(label="Hasil Deteksi")
+        # streaming=False artinya mode ambil foto biasa
+        inp = gr.Image(sources=["webcam"], label="Ambil Foto Di Sini", streaming=False)
+        out = gr.Image(label="Hasil AI")
+    
+    btn = gr.Button("🔍 Mulai Deteksi", variant="primary")
     
-    # Jalankan fungsi setiap kali ada frame baru dari kamera
-    input_cam.change(fn=detect_objects, inputs=input_cam, outputs=output_cam)
+    # AI hanya jalan saat tombol ditekan
+    btn.click(fn=detect_manual, inputs=inp, outputs=out)
 
-# Matikan SSR agar webcam lebih lancar
 if __name__ == "__main__":
     demo.launch(ssr_mode=False)