Image processing

Dockerfile

@@ -9,5 +9,3 @@ RUN pip install --no-cache-dir -r /code/requirements.txt
 COPY . .
 
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
-
-#CMD ["fastapi", "run"]

app.py

@@ -26,4 +26,12 @@ async def echo_image(image: UploadFile = File(...)):
     if not image.content_type.startswith("image/"):
         return {"error": "El archivo no es una imagen"}
     contents = await image.read()
-    return StreamingResponse(BytesIO(contents), media_type=image.content_type)
+    return StreamingResponse(BytesIO(contents), media_type=image.content_type)
+
+@app.post(
+        "/genera_dni/",        
+        summary="Procesamiento de DNI")
+async def comidas(image: UploadFile = File(...)):
+    if not image.content_type.startswith("image/"):
+        return {"error": "El archivo no es una imagen"}
+    return funciones.procesa_dni(image)

funciones.py

@@ -1,9 +1,15 @@
 from gradio_client import Client, handle_file
 
-client = Client("Moibe/api_rapicash")
-result = client.predict(
-		img=handle_file('dnid.png'),
-		lang="en",
-		api_name="/predict"
-)
-print(result)
+def procesa_dni(image):
+
+	print(image)
+
+	client = Client("Moibe/api_rapicash")
+	result = client.predict(
+			img=handle_file('dnid.png'),
+			lang="en",
+			api_name="/predict"
+	)
+	print(result)
+
+	return result