Update app.py

app.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi import FastAPI, File, UploadFile, HTTPException, Query
 from fastapi.responses import HTMLResponse
 from pydantic import BaseModel
 from typing import List
@@ -9,7 +9,7 @@ from io import BytesIO
 
 app = FastAPI()
 
-def buscar_existe(imagen):
+def buscar_existe(imagen, mensaje):
     face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')
     #imagen = cv2.imread('carro.jpg')
     gray = cv2.cvtColor(imagen, cv2.COLOR_BGR2GRAY)# convierte a tono de gris
@@ -19,15 +19,15 @@ def buscar_existe(imagen):
         existe = "Si"
         break
         
-    return existe
+    return existe+" "+mensaje
 
 # Ruta de predicción
 @app.post('/predict/')
-async def predict(file: UploadFile = File(...)):
+async def predict(file: UploadFile = File(...), mensaje: str = Query(...)):
     try:
         image = Image.open(BytesIO(await file.read()))
         image = np.asarray(image)
-        prediction = buscar_existe(image)
+        prediction = buscar_existe(image, mensaje)
         return {"prediction": prediction}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))