Model backup

app.py

@@ -38,6 +38,6 @@ async def genera_imagen(platillo: str = Form(...)):
         resultado = funciones.genera_platillo_inference(platillo)
         print("El resultado de inference es: ", resultado)
         if type(resultado) is str:
-            return '{"Error":"500"}'
+            return resultado
         else:
             return StreamingResponse(content=resultado, media_type="image/png")          

archivos/last_timestamp.txt

@@ -1 +1 @@
-1745452456
+1745539537

archivos/modelo_actual.txt

@@ -0,0 +1 @@
+black-forest-labs/FLUX.1-dev

archivos/modelo_principal.txt

@@ -0,0 +1 @@
+black-forest-labs/FLUX.1-dev

archivos/seconds_available.txt

@@ -1 +1 @@
-288
+294

funciones.py

@@ -7,7 +7,8 @@ from huggingface_hub import InferenceClient
 def genera_platillo_gpu(platillo):
 
     prompt = globales.previo + platillo
-    print("Platillo generado:", platillo)
+    print("Platillo enviado:", platillo)
+    
 
     kwargs = {
             "prompt": prompt,
@@ -32,6 +33,7 @@ def genera_platillo_gpu(platillo):
         #Cuando es GPU, debe de restar segundos disponibles de HF 
         herramientas.restaSegundosGPU(globales.work_cost)
 
+        print("Platillo generado:", platillo)
         print("Resultado regresado en result[0] es: ", result[0])        
         return result[0]
 
@@ -44,9 +46,19 @@ def genera_platillo_gpu(platillo):
 
 def genera_platillo_inference(platillo):
 
+    print("Proveedor:", globales.proveedor)
+    modelo_actual = herramientas.obtenModeloActual()
+
+    #modelo = globales.inferencia
+
+    modelo = modelo_actual
+
+
+    print("Modelo:", modelo)
+
     prompt = globales.previo + platillo
-    print("Platillo generado:", platillo)
-   
+    print("Platillo enviado:", platillo)
+       
     client = InferenceClient(
             provider= globales.proveedor,
             api_key=globales.llave
@@ -55,24 +67,22 @@ def genera_platillo_inference(platillo):
     try: 
         image = client.text_to_image(
         prompt,
-        model=globales.inferencia,
-        #seed=42, 
-        #guidance_scale=7.5,
-        #num_inference_steps=50,
-        #width=1024, #El default es 1024 x 1024 y quizá 1024*768, el max es 1536. 
-        #height=1024 #El límite de replicate es 1024.        
-        )
-
-        img_io = io.BytesIO()
-        image.save(img_io, "PNG")
-        img_io.seek(0)
-        return img_io         
+        model=modelo
+        )                 
         
     except Exception as e:
         print("Excepción es: ", e)
-        # Opción de envío de imagen genérica.
-        # 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"}'
+        if "Gateway Time-out" in str(e): 
+            print("GATEWAY TIME-OUT 💀")
+            modelo=globales.inferencia_backup
+            #Escribe en txt el nuevo modelo.
+            herramientas.modificaModeloActual(modelo)
+        
+        return f"Error: {e}"
+    
+    img_io = io.BytesIO()
+    image.save(img_io, "PNG")
+    img_io.seek(0)
+    print("Platillo generado:", platillo)
+    return img_io
+    

globales.py

@@ -6,9 +6,10 @@ llave = herramientas.obtenAccesoHF()
 espacio = "black-forest-labs/FLUX.1-schnell"
 #espacio = "black-forest-labs/FLUX.1-dev"
 inferencia = "black-forest-labs/FLUX.1-dev"
+inferencia_backup = "black-forest-labs/FLUX.1-schnell"
 
 proveedor =  "hf-inference"
 
-work_cost = 12
+work_cost = 6
 
 quota = 300

herramientas.py

@@ -47,7 +47,7 @@ def obtenUltimoTimestamp():
         print(f"Error: El contenido del archivo '{archivo_ruta}' no es un número entero válido.")
         return
 
-def nuevoDia():
+def esNuevoDia():
     """
     Compara dos timestamps (en formato string ISO) y devuelve True si han
     pasado más de 24 horas entre ellos, False en caso contrario.
@@ -68,7 +68,7 @@ def obtenSegundosDisponibles():
 
     print("Estoy en obten segundos disponibles...")
 
-    if nuevoDia() == True: 
+    if esNuevoDia() == True: 
         renuevaSegundosDisponibles()
 
     archivo_ruta = "archivos/seconds_available.txt"
@@ -101,6 +101,7 @@ def renuevaSegundosDisponibles():
             archivo.write(str(quota_total))
         print(f"Se renovaron los {quota_total} segundos disponibles.")
         renuevaTimestampActual()
+        renuevaModeloPrincipal()
     except Exception as e:
         print(f"Error al escribir en el archivo '{archivo_ruta}': {e}") 
 
@@ -138,4 +139,78 @@ def restaSegundosGPU(segundos):
             archivo.write(str(nuevos_segundos_disponibles))
         print(f"Se restaron {segundos} segundos. Ahora quedan {nuevos_segundos_disponibles} segundos disponibles.")
     except Exception as e:
-        print(f"Error al escribir en el archivo '{archivo_ruta}': {e}") 
+        print(f"Error al escribir en el archivo '{archivo_ruta}': {e}") 
+
+
+def modificaModeloActual(nuevo_modelo):
+    """
+    Actualiza el archivo archivos/modelo_actual.txt con el modelo funcional en caso de
+    problemas con el actual.
+    """
+
+    print("Estoy en actualiza modelo actual...")
+    archivo_ruta = "archivos/modelo_actual.txt"
+    modelo_actual = obtenModeloActual()    
+ 
+
+    # Guardar el nuevo número en el archivo
+    try:
+        with open(archivo_ruta, 'w') as archivo:
+            archivo.write(nuevo_modelo)
+        print(f"Se actualizó el modelo actual: {modelo_actual} por {nuevo_modelo}.")
+    except Exception as e:
+        print(f"Error al escribir en el archivo '{archivo_ruta}': {e}") 
+
+def obtenModeloActual():
+    """
+    Obtiene el último timestamp de renovación guardado.
+    """
+    
+    archivo_ruta = "archivos/modelo_actual.txt"
+
+    try:
+        with open(archivo_ruta, 'r') as archivo:
+            contenido = archivo.readline().strip()
+            modelo_actual = str(contenido)
+            return modelo_actual
+    except FileNotFoundError:
+        print(f"Error: El archivo '{archivo_ruta}' no fue encontrado.")
+        return 
+    except ValueError:
+        print(f"Error: El contenido del archivo '{archivo_ruta}' no es un número entero válido.")
+        return
+    
+def obtenModeloPrincipal():
+    """
+    Obtiene el modelo principal (default).
+    """
+    
+    archivo_ruta = "archivos/modelo_principal.txt"
+
+    try:
+        with open(archivo_ruta, 'r') as archivo:
+            contenido = archivo.readline().strip()
+            modelo_principal = str(contenido)
+            return modelo_principal
+    except FileNotFoundError:
+        print(f"Error: El archivo '{archivo_ruta}' no fue encontrado.")
+        return 
+    except ValueError:
+        print(f"Error: El contenido del archivo '{archivo_ruta}' no es un número entero válido.")
+        return
+
+
+def renuevaModeloPrincipal():
+
+    archivo_ruta = "archivos/modelo_actual.txt"   
+    modelo_actual = obtenModeloActual() 
+    modelo_principal = obtenModeloPrincipal()
+
+    # Guardar el nuevo número en el archivo
+    try:
+        with open(archivo_ruta, 'w') as archivo:
+            archivo.write(modelo_principal)
+        print(f"Se actualizó el modelo principal: {modelo_actual} por {modelo_principal}.")
+    except Exception as e:
+        print(f"Error al escribir en el archivo '{archivo_ruta}': {e}") 
+