Automatización de renovaciones de tiempos y modelos en todos los servers

app.py

@@ -32,8 +32,7 @@ async def echo_image(image: UploadFile = File(...)):
 @app.post("/genera-imagen/")
 async def genera_imagen(platillo: str = Form(...)):
 
-    seconds_available = herramientas.obtenSegundosDisponibles() #Ahora los segundos disponible son independientes. Por server no hay diferencia, pero si por cuenta (y ahora tenemos 3 PRO), es importante que no quede ese distintivo en código para que la sincronización sea fácil y rápida. Local es Moibe, las demás su respectivo server. Todos manejados en bridges y en hf variables.
-
+    seconds_available = herramientas.obtenSegundosDisponibles()
     print(herramientas.imprimeTimeNow())
     
     if seconds_available > globales.work_cost:
@@ -51,29 +50,4 @@ async def genera_imagen(platillo: str = Form(...)):
         if type(resultado) is str:
             return resultado
         else:
-            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")    
-#     seconds_available = herramientas.obtenSegundosDisponibles()
-#     #seconds_available = 0 
-#     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")                
+            return StreamingResponse(content=resultado, media_type="image/png")   

autenticacion.py

@@ -1,27 +1,29 @@
 import os
 import socket
 
+def local_check():
+    hostname = socket.gethostname()
+    #r-moibe-nowme
+    print("Hostname: ", hostname)
+    #Estoy usando el nombre de la app para identificar que estoy corriendola en HF. #Ésto independiente al servidor en uso.
+    if "-nowme" in hostname:
+        print("Ejecutando api en el servidor.")
+        return False
+    else:
+        print("Ejecutando api en local.")
+        return True
+
 def defineAmbiente():
     if local_check():
         print("Entorno Local...")        
         import bridges        
         llave = bridges.llave #Acceso a HF
         servidor = "moibe" #o Sosa Jhons, depende de donde deduciras el tiempo de gpu cuando pruebas en local.
+        hora_renovacion = 17
     else:
         print("Entorno remoto listo...")        
         llave = os.getenv("llave") #Acceso a HF
         servidor = os.getenv("servidor")
         print("El servidor remoto es: ", servidor)
-    return llave, servidor
-
-def local_check():
-    hostname = socket.gethostname()
-    #r-moibe-nowme
-    print("Hostname: ", hostname)
-    #Estoy usando el nombre de la app para identificar que estoy corriendola en HF. #Ésto independiente al servidor en uso.
-    if "-nowme" in hostname:
-        print("Ejecutando api en el servidor.")
-        return False
-    else:
-        print("Ejecutando api en local.")
-        return True
+        hora_renovacion = os.getenv("hora_renovacion")
+    return llave, servidor, hora_renovacion

funciones.py

@@ -32,8 +32,7 @@ def genera_platillo_gpu(platillo):
         print(f"La configuración completa para '{espacio_aleatorio_elegido}' es: {configuracion_espacio}")
 
         client = gradio_client.Client(configuracion_espacio['ruta'], hf_token=globales.llave)
-        #kwargs = selected_space_config['static_kwargs']
-
+        
         result = client.predict(
         #**kwargs,
         prompt=prompt,
@@ -93,10 +92,7 @@ def genera_platillo_inference(platillo):
             provider= provedor_seleccionado,
             model=configuracion_modelo['ruta'],
             api_key=globales.llave
-        )    
-    
-    print("Cree cliente: ", client)
-    time.sleep(0)
+        )
 
     try: 
         image = client.text_to_image(
@@ -114,9 +110,7 @@ def genera_platillo_inference(platillo):
         print("Excepción: ", e)
         if "Gateway Time-out" in str(e): 
             print("GATEWAY TIME-OUT 💀")
-            #modelo=globales.inferencia_backup #Con el nuevo paradigma ya no hay cambio de modelo.
-            #Escribe en txt el nuevo modelo.
-            #herramientas.modificaModeloActual(modelo)        
+                   
         error_impreso = f"Error: {e}"
         print(error_impreso)
         raise HTTPException(

globales.py

@@ -1,9 +1,8 @@
 import autenticacion
-version = "1.1.1" #Medición de segundos de procesamiento para cada servidor.
 
 #previo = "Una fotografía de un plato blanco con "
-previo = "A photograph of a white plate with a visually appealing culinary dish of  "  
-llave, servidor = autenticacion.defineAmbiente()
+previo = "A pic of a white plate with a visually appealing culinary dish of  "  
+llave, servidor, hora_renovacion = autenticacion.defineAmbiente()
 
 
 # espacio = [
@@ -24,17 +23,3 @@ proveedor_back = "fal-ai"
 work_cost = 1 #Los segundos que cuesta la generación de una imagen, para control de la capa gratuita.
 inference_cost = 1
 quota = 1500
-
-
-SERVER_NAMES = {
-    "google_us": "Google US Server",
-    #"azure_lite": "Azure Lite Supercomputer Server",
-    #"artemis" : "Artemis GPU Super cluster",
-    #"nb_dr" : "NebulaDrive Tensor Server",
-    #"pixelnet" : "PixelNet NPU Server",
-    #"nsfw_core" : "NSFW-Core: Uncensored Server",
-    
-    #"nsfw_core_2" : "NSFW-Core: Uncensored Server 2",
-    #"nsfw_core_3" : "NSFW-Core: Uncensored Server 3",
-    #"nsfw_core_4" : "NSFW-Core: Uncensored Server 4",
-}

herramientas.py

@@ -1,37 +1,31 @@
-import conexion_firebase
-import globales
+import time
 import pytz
 import random
-from datetime import datetime
+import globales
+import conexion_firebase
+from datetime import datetime 
+import time 
 
-servidor = globales.servidor
 
-# segundos = server_configs[globales.servidor]["segundos_var"]
+servidor = globales.servidor
 
 def obtenUltimoTimestamp():
     """
     Obtiene el último timestamp de renovación guardado.
     """
     resultado = conexion_firebase.obtenDato('nowme', servidor, 'timestamp')
-    print("Obten último timestamp obtenido: ", resultado)    
     
     return resultado
     
 def obtenSegundosDisponibles(): 
 
-    if esDiaSiguiente() == True:
-        renuevaSegundosDisponibles()
-        #renuevaModeloPrincipal() El modelo principal ya se renueva directo en base.
+    siEsDiaSiguienteRenueva()       
 
     #Finalmente obten los segundos disponibles después de las operaciones.
     return conexion_firebase.obtenDato('nowme', servidor, 'segundos')
 
 def obtenSegundosDisponiblesInference(): 
 
-    # if esDiaSiguiente() == True:
-    #     renuevaSegundosDisponibles()
-    #     renuevaModeloPrincipal()
-
     #Finalmente obten los segundos disponibles después de las operaciones.
     return conexion_firebase.obtenDato('nowme', servidor, 'inferencias')
     
@@ -45,7 +39,6 @@ def renuevaSegundosDisponibles():
 def renuevaTimestampActual(): 
 
     timestamp_actual = imprimeTimeNowFixed16h()
-    print("Ésto es el timestamp now de la renovación...: ", timestamp_actual)
     conexion_firebase.editaDato('nowme', servidor, 'timestamp', timestamp_actual)    
 
 def restaSegundosGPU(cuantos_segundos):
@@ -55,8 +48,7 @@ def restaSegundosGPU(cuantos_segundos):
     """
 
     segundos_disponibles = obtenSegundosDisponibles()    
-    print("Segundos disponibles: ", segundos_disponibles)
-
+    
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
     print("Segundos disponibles ahora: ", nuevos_segundos_disponibles)
@@ -69,8 +61,7 @@ def restaSegundosInference(cuantos_segundos):
     """
 
     segundos_disponibles = obtenSegundosDisponiblesInference()    
-    print("Segundos disponibles Inference: ", segundos_disponibles)
-    
+
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
     print("Segundos disponibles ahora en restaSegundosInference: ", nuevos_segundos_disponibles)
@@ -84,13 +75,7 @@ def modificaModeloActual(nuevo_modelo):
 
     modelo_actual = conexion_firebase.obtenDato('nowme', servidor, 'modelo_actual')
     conexion_firebase.editaDato('nowme', servidor, 'modelo_actual', nuevo_modelo)
-    print(f"Se actualizó el modelo actual: {modelo_actual} por {nuevo_modelo}.")
 
-# def renuevaModeloPrincipal():
-#     #Obten el modelo principal (default).   
-#     modelo_principal = conexion_firebase.obtenDato('nowme', servidor, 'modelo_principal')
-#     #Asignalo como modelo actual.
-#     conexion_firebase.editaDato('nowme', servidor, 'modelo_actual', modelo_principal)
 
 def imprimeTimeNow(): 
     """
@@ -145,53 +130,56 @@ def imprimeTimeNowFixed16h():
 
     return formatted_time
 
-def esDiaSiguiente(): 
-    """
-    Compara dos timestamps Unix y devuelve True si el día de timestamp_actual
-    es diferente al día de timestamp_registro.
-
-    Args:
-        timestamp_registro (int): Timestamp Unix del registro original (en segundos).
-        timestamp_actual (int): Timestamp Unix actual (en segundos).
-
-    Returns:
-        bool: True si el día de timestamp_actual es diferente al día de
-              timestamp_registro, False si es el mismo día.
-    """
+def siEsDiaSiguienteRenueva():    
 
     #Obtiene el último registro de fecha de la base de firestore.
-    fecha_registro_dt = obtenUltimoTimestamp()   
-    
-    print("Primero, el último timestamp obtenido de base es: ", fecha_registro_dt)
-    print("Y su tipo es: ", type(fecha_registro_dt))
+    fecha_registro_dt = obtenUltimoTimestamp()
+    print("Obtuve el último timestamp:  ", fecha_registro_dt)
    
     #Timestamp actual
-    fecha_actual_dt = imprimeTimeNow()
-    print("Segundo, la fecha_actual_dt quedó como: ", fecha_actual_dt)
-    print("Y su tipo es: ", type(fecha_actual_dt))
+    fecha_actual_dt = imprimeTimeNow()    
     
-
     formato = "%Y-%m-%d %H:%M:%S"
-    datetime_obj_1 = datetime.strptime(fecha_registro_dt, formato)
-    datetime_obj_2 = datetime.strptime(fecha_actual_dt, formato)   
-        
+    datetime_obj_1 = datetime.strptime(fecha_registro_dt, formato)    
+    datetime_obj_2 = datetime.strptime(fecha_actual_dt, formato)  
+            
     # Extraer solo la fecha de los objetos datetime
     fecha_registro = datetime_obj_1.date()
-    #print("Primera extracción: ", fecha_registro)
+    print("Primera extracción: ", fecha_registro)
     fecha_actual = datetime_obj_2.date()
-    #print("Segunda extracción: ", fecha_actual)    
-    
+    print("Segunda extracción: ", fecha_actual)
+
+    # Experimento de extracción de horas: 
+    hora_registro = datetime_obj_1.time()
+    print("Hora Primera extracción: ", hora_registro)   
 
     # Verificar si las fechas son diferentes
     resultado = fecha_actual > fecha_registro
-    print("Resultado de resta de fechas: ", resultado)    
-    
-    return resultado
-
-def getFluxServer(): 
-    print("In getFluxServer...")
-    server_keys = list(globales.SERVER_NAMES.values())
-    random_value = random.choice(server_keys)
-    print("Random key: ", random_value)
+    print("Resultado de resta de fechas: ", resultado)
+
+    print("Las fechas son del tipo: ", type(fecha_registro))
+    diferencia = fecha_actual - fecha_registro
+    print("Diferencia.days es: ", diferencia.days)    
+
+    if resultado == True:
+        print("El resultado es True, aquí checaríamos la hora.") 
+        if diferencia.days > 1:
+            print("Renovación directa por distancia superior a 1 día...")
+            print("Renovando segundos.")
+            renuevaSegundosDisponibles()
+            renuevaTimestampActual()
+        else: #Si la diferencia es de un solo día entonces si debe checar si ya rebaso la hora de renovación del servidor.        
+            hora_actual = datetime_obj_2.time()
+            print("Hora Segunda extracción: ", hora_actual)    
+            print("La hora exacta es: ", hora_actual.hour)
+            print("La hora renovación es: ", globales.hora_renovacion)
+            if hora_actual.hour > globales.hora_renovacion: 
+                print("Renovando segundos.")
+                renuevaSegundosDisponibles()
+                renuevaTimestampActual()
+            else:
+                print("La hora aún no llega al límite, no renovará.")
+    else: 
+        print("El resultado es false, no renovará...")
     
-    return random_value
+    return resultado