Upload 3 files

Dockerfile (1)

@@ -0,0 +1,20 @@
+# Usa una imagen base de Python
+FROM python:3.11.0
+# Establece el directorio de trabajo
+WORKDIR /code
+
+# Copia los archivos necesarios al contenedor
+COPY ./requirements.txt /code/requirements.txt
+RUN apt-get update && apt-get install -y libgl1-mesa-glx
+RUN pip install --no-cache-dir -r /code/requirements.txt
+RUN pip install fastapi uvicorn pillow opencv-python
+
+
+
+COPY . .
+
+RUN chmod -R 777 /code
+EXPOSE 7860
+
+# Comando para ejecutar la aplicación
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app (1).py

@@ -0,0 +1,42 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+import cv2
+import numpy as np
+from PIL import Image
+import io
+import base64
+
+app = FastAPI()
+
+@app.post("/detect/")
+async def detect_face(file: UploadFile = File(...)):
+    try:
+        image_bytes = await file.read()
+        image = Image.open(io.BytesIO(image_bytes))
+        img_np = np.array(image)
+
+        if img_np.shape[2] == 4:
+            img_np = cv2.cvtColor(img_np, cv2.COLOR_BGRA2BGR)
+
+        face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
+        gray = cv2.cvtColor(img_np, cv2.COLOR_BGR2GRAY)
+        faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))
+
+        if len(faces) == 0:
+            raise HTTPException(status_code=404, detail="No se detectaron rostros en la imagen.")
+
+        for (x, y, w, h) in faces:
+            cv2.rectangle(img_np, (x, y), (x+w, y+h), (255, 0, 0), 2)
+
+        result_image = Image.fromarray(cv2.cvtColor(img_np, cv2.COLOR_BGR2RGB))
+        img_byte_arr = io.BytesIO()
+        result_image.save(img_byte_arr, format='JPEG')
+        img_byte_arr = img_byte_arr.getvalue()
+
+        return {
+            "message": "Rostros detectados",
+            "rostros": len(faces),
+            "imagen_base64": base64.b64encode(img_byte_arr).decode('utf-8')
+        }
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

requirements (1).txt

@@ -0,0 +1,6 @@
+fastapi
+uvicorn
+opencv-python
+numpy
+Pillow
+python-multipart