Update api.py

api.py

@@ -45,9 +45,6 @@ async def root():
 
 @app.post("/predict/")
 async def predict(file: UploadFile = File(...)):
-    """
-    Returns JSON with detections (no image).
-    """
     try:
         contents = await file.read()
         image = Image.open(io.BytesIO(contents)).convert("RGB")
@@ -88,9 +85,6 @@ async def predict(file: UploadFile = File(...)):
 
 @app.post("/predict_image/")
 async def predict_image(file: UploadFile = File(...)):
-    """
-    Returns the annotated image only (PNG).
-    """
     try:
         contents = await file.read()
         image = Image.open(io.BytesIO(contents)).convert("RGB")
@@ -108,3 +102,37 @@ async def predict_image(file: UploadFile = File(...)):
 
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
+
+# ✅ Nouvelle route de test sur Hugging Face
+@app.get("/test_request/")
+async def test_request():
+    """
+    Teste l'API déployée sur Hugging Face : envoie une image vers /predict et /predict_image,
+    puis sauvegarde les résultats.
+    """
+    try:
+        file_path = "test.jpg"  # mets ton image dans ton Space
+        base_url = "https://stroke-ia-api.hf.space"  # ⚠️ adapte au nom exact de ton Space
+
+        # 1) Test JSON (détection)
+        url_predict = f"{base_url}/predict/"
+        files = {"file": open(file_path, "rb")}
+        response = requests.post(url_predict, files=files)
+        json_result = response.json()
+
+        # 2) Test image annotée
+        url_img = f"{base_url}/predict_image/"
+        files = {"file": open(file_path, "rb")}
+        response_img = requests.post(url_img, files=files)
+
+        with open("result.png", "wb") as f:
+            f.write(response_img.content)
+
+        return {
+            "message": "✅ Test request executed on Hugging Face API. Results saved.",
+            "json_result": json_result,
+            "saved_image": "result.png"
+        }
+
+    except Exception as e:
+        return {"error": str(e)}