|
|
from fastapi import FastAPI, Form |
|
|
from fastapi import FastAPI, File, UploadFile |
|
|
from fastapi.responses import StreamingResponse, FileResponse |
|
|
from io import BytesIO |
|
|
import funciones, globales |
|
|
|
|
|
app = FastAPI() |
|
|
|
|
|
@app.post("/echo-image/") |
|
|
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) |
|
|
|
|
|
@app.post("/genera-imagen/") |
|
|
async def genera_imagen(platillo: str = Form(...)): |
|
|
|
|
|
if globales.seconds_available > 25: |
|
|
print("GPU...") |
|
|
resultado = funciones.genera_platillo_gpu(platillo) |
|
|
return FileResponse(resultado, media_type="image/png", filename="imagen.png") |
|
|
else: |
|
|
print("Inference...") |
|
|
resultado = funciones.genera_platillo_inference(platillo) |
|
|
return StreamingResponse(content=resultado, media_type="image/png") |
