Medición de segundos de procesamiento para cada servidor.

app.py

@@ -32,7 +32,7 @@ async def echo_image(image: UploadFile = File(...)):
 @app.post("/genera-imagen/")
 async def genera_imagen(platillo: str = Form(...)):
 
-    seconds_available = herramientas.obtenSegundosDisponibles()
+    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.
 
     print(herramientas.imprimeTimeNow())
     

autenticacion.py

@@ -6,7 +6,7 @@ def defineAmbiente():
         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.
+        servidor = "sosa" #o Sosa Jhons, depende de donde deduciras el tiempo de gpu cuando pruebas en local.
     else:
         print("Entorno remoto listo...")        
         llave = os.getenv("llave") #Acceso a HF
@@ -18,7 +18,7 @@ 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.
+    #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

globales.py

@@ -33,6 +33,7 @@ SERVER_NAMES = {
     #"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,23 +1,18 @@
-from datetime import datetime
-from datetime import datetime
 import conexion_firebase
 import globales
 import pytz
-import time
 import random
+from datetime import datetime
 
-if globales.servidor == "moibe":
-    segundos = "segundos_dev"
-    last_timestamp = "last_timestamp_dev"
-else: 
-    segundos = "segundos_prod"
-    last_timestamp = "last_timestamp_prod"
+servidor = globales.servidor
+
+# segundos = server_configs[globales.servidor]["segundos_var"]
 
 def obtenUltimoTimestamp():
     """
     Obtiene el último timestamp de renovación guardado.
     """
-    resultado = conexion_firebase.obtenDato('nowme', 'huggingface', last_timestamp)
+    resultado = conexion_firebase.obtenDato('nowme', servidor, 'timestamp')
     print("Obten último timestamp obtenido: ", resultado)    
     
     return resultado
@@ -29,7 +24,7 @@ def obtenSegundosDisponibles():
         renuevaModeloPrincipal()
 
     #Finalmente obten los segundos disponibles después de las operaciones.
-    return conexion_firebase.obtenDato('nowme', 'huggingface', segundos)
+    return conexion_firebase.obtenDato('nowme', servidor, 'segundos')
 
 def obtenSegundosDisponiblesInference(): 
 
@@ -38,20 +33,20 @@ def obtenSegundosDisponiblesInference():
     #     renuevaModeloPrincipal()
 
     #Finalmente obten los segundos disponibles después de las operaciones.
-    return conexion_firebase.obtenDato('nowme', 'huggingface', 'hfInference')
+    return conexion_firebase.obtenDato('nowme', servidor, 'hfInference')
     
     
 def renuevaSegundosDisponibles():
 
     #Segundos de cuota total gratuita disponibles al momento.
-    conexion_firebase.editaDato('nowme', 'huggingface', segundos, globales.quota)
+    conexion_firebase.editaDato('nowme', servidor, 'segundos', globales.quota)
     renuevaTimestampActual()
 
 def renuevaTimestampActual(): 
 
     timestamp_actual = imprimeTimeNowFixed16h()
     print("Ésto es el timestamp now de la renovación...: ", timestamp_actual)
-    conexion_firebase.editaDato('nowme', 'huggingface', last_timestamp, timestamp_actual)    
+    conexion_firebase.editaDato('nowme', servidor, 'timestamp', timestamp_actual)    
 
 def restaSegundosGPU(cuantos_segundos):
     """
@@ -65,7 +60,7 @@ def restaSegundosGPU(cuantos_segundos):
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
     print("Segundos disponibles ahora: ", nuevos_segundos_disponibles)
-    conexion_firebase.editaDato('nowme', 'huggingface', segundos, nuevos_segundos_disponibles)
+    conexion_firebase.editaDato('nowme', servidor, 'segundos', nuevos_segundos_disponibles)
 
 def restaSegundosInference(cuantos_segundos):
     """
@@ -79,7 +74,7 @@ def restaSegundosInference(cuantos_segundos):
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
     print("Segundos disponibles ahora en restaSegundosInference: ", nuevos_segundos_disponibles)
-    conexion_firebase.editaDato('nowme', 'huggingface', 'hfInference', nuevos_segundos_disponibles)    
+    conexion_firebase.editaDato('nowme', servidor, 'hfInference', nuevos_segundos_disponibles)    
 
 def modificaModeloActual(nuevo_modelo):
     """
@@ -87,15 +82,15 @@ def modificaModeloActual(nuevo_modelo):
     problemas con el actual.
     """
 
-    modelo_actual = conexion_firebase.obtenDato('nowme', 'huggingface', 'modelo_actual')
-    conexion_firebase.editaDato('nowme', 'huggingface', 'modelo_actual', 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', 'huggingface', 'modelo_principal')
+    modelo_principal = conexion_firebase.obtenDato('nowme', servidor, 'modelo_principal')
     #Asignalo como modelo actual.
-    conexion_firebase.editaDato('nowme', 'huggingface', 'modelo_actual', modelo_principal)
+    conexion_firebase.editaDato('nowme', servidor, 'modelo_actual', modelo_principal)
 
 def imprimeTimeNow(): 
     """
@@ -145,15 +140,6 @@ def imprimeTimeNowFixed16h():
         )
     )
     
-    # Nota sobre .localize() vs .astimezone():
-    # .localize() se usa para tomar un datetime "naive" (sin zona horaria)
-    # y asignarle una zona horaria.
-    # .astimezone() se usa para convertir un datetime "timezone-aware"
-    # de una zona horaria a otra.
-    # Aquí creamos un datetime "naive" con la fecha correcta y la hora 16:00:00
-    # y luego lo "localizamos" en la zona horaria de México.
-
-
     # 5. Formatear la fecha y hora
     formatted_time = fixed_time_dt.strftime("%Y-%m-%d %H:%M:%S")