Update detector/gesture_detector.py

detector/gesture_detector.py

@@ -3,13 +3,16 @@ from PIL import Image
 import cv2
 import os
 from datetime import datetime
-import base64
 
-# Load model YOLO hanya sekali
+# Load YOLO model
 model = YOLO("best.pt")
 
-# Buat folder hasil
-os.makedirs("detected_images", exist_ok=True)
+# Folder aman untuk Hugging Face
+SAVE_DIR = "/tmp/detected_images"
+os.makedirs(SAVE_DIR, exist_ok=True)
+
+# Base URL Hugging Face kamu (ganti sesuai nama Space kamu)
+BASE_URL = "https://maulidaaa-hellome.hf.space/detected_images"
 
 def detect_gesture(image_file):
     image = Image.open(image_file.stream).convert("RGB")
@@ -34,17 +37,14 @@ def detect_gesture(image_file):
                 }
             })
 
-    img_with_boxes = results[0].plot()
+    # Simpan gambar hasil
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    saved_filename = f"detected_images/detection_{timestamp}.jpg"
-    cv2.imwrite(saved_filename, img_with_boxes)
-
-    _, buffer = cv2.imencode(".jpg", img_with_boxes)
-    img_base64 = base64.b64encode(buffer).decode("utf-8")
+    filename = f"detection_{timestamp}.jpg"
+    saved_path = os.path.join(SAVE_DIR, filename)
+    cv2.imwrite(saved_path, results[0].plot())
 
     return {
-        "saved_to": saved_filename,
-        "image_base64": img_base64,
+        "saved_to": f"{BASE_URL}/{filename}",
         "detections": detections,
         "total_detections": len(detections)
     }