Fastapi

.gitignore

@@ -1 +1,2 @@
 /venv/
+/__pycache__/

app.py

@@ -1,9 +1,29 @@
-from gradio_client import Client, handle_file
+from io import BytesIO
+import funciones
+from fastapi import FastAPI, Form
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import StreamingResponse, FileResponse, JSONResponse
 
-client = Client("Moibe/api_rapicash")
-result = client.predict(
-		img=handle_file('dnid.png'),
-		lang="en",
-		api_name="/predict"
-)
-print(result)
+app = FastAPI()
+
+# Nuevo endpoint para Health Check
+@app.get("/health",
+         tags=["Health Check"],
+         description="Verifica el estado de salud de la API.",
+         summary="Health Check"
+         )
+async def health_check():
+    """
+    Este endpoint devuelve una respuesta 200 OK para indicar que la API está funcionando.
+    """
+    return JSONResponse(content={"status": "ok"}, status_code=200)
+
+@app.post("/echo-image/",
+          description="Test endpoint que recibe y regresa la misma imagen, para probar envío, recepción y problemas con api o red.",
+          summary="Summary"
+          )
+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)

funciones.py

@@ -0,0 +1,9 @@
+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)

requirements.txt

@@ -1 +1,2 @@
-gradio_client
+gradio_client
+fastapi