Update app.py

app.py

@@ -102,31 +102,54 @@ def estimate_quantity(pred):
 # =====================
 # ROBOFLOW (HTTP)
 # =====================
+import base64
+
 def detect(image_path):
-    url = "https://serverless.roboflow.com/rishab-5ghrt/detect-count-and-visualize-4/run"
+    api_key = os.getenv("ROBOFLOW_API_KEY")
+
+    if not api_key:
+        return "❌ API KEY NOT FOUND"
+
+    url = "https://serverless.roboflow.com/rishab-5ghrt/detect-count-and-visualize-4"
 
     try:
+        # convert image → base64
         with open(image_path, "rb") as f:
-            response = requests.post(
-                url,
-                files={"file": f},
-                params={"api_key": ROBOFLOW_API_KEY},
-                timeout=15
-            )
+            image_bytes = f.read()
+            image_base64 = base64.b64encode(image_bytes).decode("utf-8")
+
+        payload = {
+            "inputs": {
+                "image": {
+                    "type": "base64",
+                    "value": image_base64
+                }
+            }
+        }
+
+        response = requests.post(
+            url,
+            json=payload,
+            headers={
+                "Authorization": f"Bearer {api_key}",
+                "Content-Type": "application/json"
+            },
+            timeout=15
+        )
 
         if response.status_code != 200:
             return f"❌ Roboflow Error: {response.text}"
 
         data = response.json()
 
-        if "predictions" in data:
-            return data["predictions"]
+        # extract predictions
+        if "outputs" in data:
+            return data["outputs"][0].get("predictions", [])
 
         return []
 
     except Exception as e:
         return f"❌ Request Failed: {str(e)}"
-
 # =====================
 # MAIN FUNCTION
 # =====================