Prod Ready

app.py

@@ -3,6 +3,7 @@ from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import StreamingResponse, FileResponse
 from io import BytesIO
 import funciones, globales
+import herramientas
 
 app = FastAPI()
 
@@ -17,7 +18,12 @@ async def echo_image(image: UploadFile = File(...)):
 @app.post("/genera-imagen/")
 async def genera_imagen(platillo: str = Form(...)):
 
-    if globales.seconds_available > 25:
+    #Obtengo antes los segundos disponibles para saber si tengo GPU disponible o me voy a inferencia.
+    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")

archivos/last_timestamp.txt

@@ -0,0 +1 @@
+1744393289

archivos/seconds_available.txt

@@ -0,0 +1 @@
+1488

config.ini

@@ -3,7 +3,6 @@ llave = herramientas.obtenAccesoHF()
 
 [hugging_face]
 previo = "Una fotografía de un plato blanco con "
-seconds_available = 1500
 espacio = "black-forest-labs/FLUX.1-schnell"
 inferencia = "black-forest-labs/FLUX.1-dev"
 proveedor =  "hf-inference"

funciones.py

@@ -2,6 +2,7 @@ from huggingface_hub import InferenceClient
 import gradio_client
 import io
 import globales
+import herramientas
 
 
 def genera_platillo_gpu(platillo):
@@ -29,6 +30,9 @@ def genera_platillo_gpu(platillo):
         # num_inference_steps=28,
         # api_name="/infer"       
         )
+
+        #Cuando es GPU, debe de restar segundos disponibles de HF 
+        herramientas.restaSegundosGPU(globales.work_cost)
         
         return result[0]
 

globales.py

@@ -3,10 +3,10 @@ import herramientas
 previo = "Una fotografía de un plato blanco con "
 llave = herramientas.obtenAccesoHF()
 
-seconds_available = 1500
-
 espacio = "black-forest-labs/FLUX.1-schnell"
 #espacio = "black-forest-labs/FLUX.1-dev"
 inferencia = "black-forest-labs/FLUX.1-dev"
 
-proveedor =  "hf-inference"
+proveedor =  "hf-inference"
+
+work_cost = 12

herramientas.py

@@ -1,5 +1,6 @@
 import os
 import socket
+import time
 
 def obtenAccesoHF():
     if local_check():
@@ -12,7 +13,6 @@ def obtenAccesoHF():
 
         print("Ésto es llave:", llave)
 
-
     return llave
 
 def local_check(): 
@@ -26,4 +26,113 @@ def local_check():
         return False
     else:
         print("Ejecutando en local")
-        return True
+        return True
+
+def obtenUltimoTimestamp():
+    """
+    Obtiene el último timestamp de renovación guardado.
+    """
+
+    archivo_ruta = "archivos/last_timestamp.txt"
+
+    try:
+        with open(archivo_ruta, 'r') as archivo:
+            contenido = archivo.readline().strip()
+            ultimo_timestamp = int(contenido)
+            return ultimo_timestamp
+    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 nuevoDia():
+    """
+    Compara dos timestamps (en formato string ISO) y devuelve True si han
+    pasado más de 24 horas entre ellos, False en caso contrario.
+    """
+
+    timestamp_original = obtenUltimoTimestamp()
+    timestamp_actual = int(time.time())
+
+    try:
+        segundos_en_24_horas = 24 * 60 * 60
+        diferencia_en_segundos = abs(timestamp_actual - timestamp_original)
+        return diferencia_en_segundos > segundos_en_24_horas
+    except ValueError:
+        print("Error: Formato de timestamp incorrecto.")
+        return False
+    
+def obtenSegundosDisponibles(): 
+
+    print("Estoy en obten segundos disponibles...")
+    
+    if nuevoDia() == True: 
+        renuevaSegundosDisponibles()
+
+    archivo_ruta = "archivos/seconds_available.txt"
+
+    try:
+        # Leer el número actual de segundos disponibles
+        with open(archivo_ruta, 'r') as archivo:
+            contenido = archivo.readline().strip()
+            segundos_disponibles = int(contenido)
+            return segundos_disponibles
+    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 renuevaSegundosDisponibles(): 
+
+    print("Estoy en renuevaSegundosDisponibles...")
+
+    archivo_ruta = "archivos/seconds_available.txt"
+
+    # 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.")
+        renuevaTimestampActual()
+    except Exception as e:
+        print(f"Error al escribir en el archivo '{archivo_ruta}': {e}") 
+
+def renuevaTimestampActual(): 
+
+    print("Estoy en renuevatimestmap actual...")
+    archivo_ruta = "archivos/last_timestamp.txt"
+
+    timestamp_actual = int(time.time())
+
+    # Guardar el nuevo número en el archivo
+    try:
+        with open(archivo_ruta, 'w') as archivo:
+            archivo.write(str(timestamp_actual))
+        print(f"Se renovó por el timestamp de éste momento.")
+    except Exception as e:
+        print(f"Error al escribir en el archivo '{archivo_ruta}': {e}")     
+
+def restaSegundosGPU(segundos):
+    """
+    Lee el número de segundos disponibles desde seconds_available.txt,
+    resta los segundos dados como parámetro y guarda el nuevo valor en el archivo.
+    """
+
+    print("Estoy en resta segundos...")
+    archivo_ruta = "archivos/seconds_available.txt"
+    segundos_disponibles = obtenSegundosDisponibles()    
+
+    # Restar los segundos
+    nuevos_segundos_disponibles = segundos_disponibles - segundos
+
+    # Guardar el nuevo número en el archivo
+    try:
+        with open(archivo_ruta, 'w') as archivo:
+            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}") 

test.py

@@ -0,0 +1,3 @@
+import herramientas
+
+print(herramientas.nuevoDia())