Update app.py

app.py

@@ -17,7 +17,7 @@ workspace = os.getenv("ROBOFLOW_WORKSPACE")
 project_name = os.getenv("ROBOFLOW_PROJECT")
 model_version = int(os.getenv("ROBOFLOW_MODEL_VERSION"))
 
-# CountGD Config (Replace DINO-X)
+# CountGD Config
 COUNTGD_API_KEY = os.getenv("COUNTGD_API_KEY")
 
 # Inisialisasi YOLO Model dari Roboflow
@@ -86,7 +86,7 @@ def detect_combined(image):
         url = "https://api.landing.ai/v1/tools/text-to-object-detection"
         headers = {"Authorization": f"Basic {COUNTGD_API_KEY}"}
         competitor_boxes = []
-        COUNTGD_PROMPTS = ["cans", "bottle", "beverage"]
+        COUNTGD_PROMPTS = ["cans", "bottle", "beverage", "mixed box"]
         
         for prompt in COUNTGD_PROMPTS:
             with open(temp_path, "rb") as f:
@@ -122,13 +122,13 @@ def detect_combined(image):
             pt2 = (int(x + w/2), int(y + h/2))
             cv2.rectangle(img, pt1, pt2, (0, 255, 0), 2)
             cv2.putText(img, pred['class'], (pt1[0], pt1[1]-10),
-                        cv2.FONT_HERSHEY_SIMPLEX, 1.0, (0,255,0), 3)
+                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0,255,0), 3)
         # Gambar bounding box CountGD (merah)
         for box in competitor_boxes:
             x1, y1, x2, y2 = box
             cv2.rectangle(img, (int(x1), int(y1)), (int(x2), int(y2)), (0, 0, 255), 2)
             cv2.putText(img, "unclassified", (int(x1), int(y1)-10),
-                        cv2.FONT_HERSHEY_SIMPLEX, 1.0, (0,0,255), 3)
+                        cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0,0,255), 3)
         
         output_path = "/tmp/combined_output.jpg"
         cv2.imwrite(output_path, img)