Servicio Secundario para Procesamiento de Lotes

app.py

@@ -51,4 +51,29 @@ async def genera_imagen(platillo: str = Form(...)):
         if type(resultado) is str:
             return resultado
         else:
-            return StreamingResponse(content=resultado, media_type="image/png")          
+            return StreamingResponse(content=resultado, media_type="image/png")    
+
+@app.post("/procesador-lotes/")
+async def procesa_lote(platillo: str = Form(...)):
+
+    print("Servicio Secundario de Procesamiento de Lotes")
+    #Obtengo los segundos disponibles de procesamiento para saber si tengo GPU disponible de la capa gratuita o me voy a método cobrado.
+    seconds_available = herramientas.obtenSegundosDisponibles()
+    print(herramientas.imprimeTimeNow())
+    
+    if seconds_available > globales.work_cost:
+        print("Usando GPU (capa gratuita)...")
+        resultado = funciones.genera_platillo_gpu(platillo)
+        if "Error" in resultado:
+            return resultado
+        else:
+            return FileResponse(resultado, media_type="image/png", filename="imagen.png")
+    else: 
+        
+        print("Usando Inference...")
+        resultado = funciones.genera_platillo_inference(platillo)
+        print("El resultado de inference es: ", resultado)
+        if type(resultado) is str:
+            return resultado
+        else:
+            return StreamingResponse(content=resultado, media_type="image/png")