Manejo de errores

app.py

@@ -22,12 +22,21 @@ async def genera_imagen(platillo: str = Form(...)):
     seconds_available = herramientas.obtenSegundosDisponibles()
     print("Los segundos disponibles obtenidos son: ", seconds_available)
 
-
     if seconds_available > globales.work_cost:
         print("GPU...")
         resultado = funciones.genera_platillo_gpu(platillo)
-        return FileResponse(resultado, media_type="image/png", filename="imagen.png")
+        if "Error" in resultado:
+            return resultado
+        else:
+            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")
+        print("El resultado de inference es: ", resultado)
+        print("Y su type es: ", type(resultado))
+        if type(resultado) is str:
+            return '{"Error":"500"}'
+        else:
+            return StreamingResponse(content=resultado, media_type="image/png")
+        
+        

archivos/last_timestamp.txt

@@ -1 +1 @@
-1744393289
+1744734269

archivos/seconds_available.txt

@@ -1 +1 @@
-1500
+288

funciones.py

@@ -35,14 +35,15 @@ def genera_platillo_gpu(platillo):
         #Cuando es GPU, debe de restar segundos disponibles de HF 
         herramientas.restaSegundosGPU(globales.work_cost)
 
-        print("Result[0] que es lo que regresa es: ", result[0])
-        time.sleep(18)
+        print("Lo que estoy regresando es result[0] o sea: ", result[0])
         
         return result[0]
 
     except Exception as e:     
         print("Excepción es: ", e)
-        return "default.png"
+        # return "default.png"
+        return '{"Error":"500"}'
+        return "Hola, error gpu..."
 
 
 def genera_platillo_inference(platillo):
@@ -74,7 +75,9 @@ def genera_platillo_inference(platillo):
         
     except Exception as e:
         print("Excepción es: ", e)
-        with open("default.png", "rb") as default_image_file:
-            img_io_default = io.BytesIO(default_image_file.read())
-            img_io_default.seek(0)
-            return img_io_default
+        # with open("default.png", "rb") as default_image_file:
+        #     img_io_default = io.BytesIO(default_image_file.read())
+        #     img_io_default.seek(0)
+        #     return img_io_default
+        return '{"Error":"500"}'
+        return "Hola error inference..."

globales.py

@@ -10,3 +10,5 @@ inferencia = "black-forest-labs/FLUX.1-dev"
 proveedor =  "hf-inference"
 
 work_cost = 12
+
+quota = 300

herramientas.py

@@ -95,8 +95,8 @@ def renuevaSegundosDisponibles():
     # Guardar el nuevo número en el archivo
     try:
         with open(archivo_ruta, 'w') as archivo:
-            archivo.write(str(1500))
-        print(f"Se renovaron los 1500 segundos disponibles.")
+            archivo.write(str(globales.quota))
+        print(f"Se renovaron los {globales.quota} segundos disponibles.")
         renuevaTimestampActual()
     except Exception as e:
         print(f"Error al escribir en el archivo '{archivo_ruta}': {e}")