Timezone optimization

app.py

@@ -34,7 +34,6 @@ async def genera_imagen(platillo: str = Form(...)):
 
     #Obtengo los segundos disponibles de procesamiento para saber si tengo GPU disponible de la capa gratuita o me voy a método cobrado.
     seconds_available = herramientas.obtenSegundosDisponibles()
-    print("Los segundos de procesamiento disponibles el día de hoy son: ", seconds_available)
     print(herramientas.imprimeTime())
     
     if seconds_available > globales.work_cost:

conexion_firebase.py

@@ -28,8 +28,7 @@ def editaDato(coleccion, dato, info, contenido):
         
     #Primero debemos definir la referencia al documento, o sea a la hoja de usuario.
     doc_ref = db.collection(coleccion).document(dato)
-    print("Doc ref es: ", doc_ref)
-    
+        
     doc_ref.update({
         # 'quote': quote,
         info: contenido,

funciones.py

@@ -75,6 +75,8 @@ def genera_platillo_inference(platillo):
             herramientas.modificaModeloActual(modelo)        
         return f"Error: {e}"
     
+    #Cuando es GPU, debe de restar segundos disponibles de HF 
+    herramientas.restaSegundosGPU(globales.work_cost)
     img_io = io.BytesIO()
     image.save(img_io, "PNG")
     img_io.seek(0)

herramientas.py

@@ -3,6 +3,7 @@ from datetime import datetime
 from datetime import datetime
 import conexion_firebase
 import globales
+import pytz
 
 if globales.servidor == "moibe":
     segundos = "segundos_dev"
@@ -15,14 +16,10 @@ def obtenUltimoTimestamp():
     """
     Obtiene el último timestamp de renovación guardado.
     """
-    print("Estoy en obtenUltimoTimestamp")
-    print("La var last_timestamp es : ", last_timestamp)
     resultado = conexion_firebase.obtenDato('nowme', 'huggingface', last_timestamp)
 
-    print(f"El resultado es: {resultado} y su tipo es {type(resultado)}... ")
     return resultado
 
-
 def esNuevoDia():
     """
     Compara dos timestamps (en formato string ISO) y devuelve True si han
@@ -42,8 +39,6 @@ def esNuevoDia():
     
 def obtenSegundosDisponibles(): 
 
-    print("Obteniendo segundos disponibles...")
-
     if esDiaSiguiente() == True:
         renuevaSegundosDisponibles()
         renuevaModeloPrincipal()
@@ -58,7 +53,6 @@ def renuevaSegundosDisponibles():
     conexion_firebase.editaDato('nowme', 'huggingface', segundos, quota_total)
     renuevaTimestampActual()
 
-
 def renuevaTimestampActual(): 
 
     timestamp_actual = int(time.time())
@@ -71,15 +65,12 @@ def restaSegundosGPU(cuantos_segundos):
     """
 
     segundos_disponibles = obtenSegundosDisponibles()    
-    print("Los segundos obtenidos de obtenSegundosDisponibles son: ", segundos_disponibles)
+    print("Segundos disponibles: ", segundos_disponibles)
 
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
-    print("La resta o los nuevos_segundos_disponibles son: ", nuevos_segundos_disponibles)
-
     conexion_firebase.editaDato('nowme', 'huggingface', segundos, nuevos_segundos_disponibles)
 
-
 def modificaModeloActual(nuevo_modelo):
     """
     Actualiza el archivo archivos/modelo_actual.txt con el modelo funcional en caso de
@@ -90,7 +81,6 @@ def modificaModeloActual(nuevo_modelo):
     conexion_firebase.editaDato('nowme', 'huggingface', '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')
@@ -98,10 +88,24 @@ def renuevaModeloPrincipal():
     conexion_firebase.editaDato('nowme', 'huggingface', 'modelo_actual', modelo_principal)
 
 def imprimeTime(): 
-    timestamp_actual = int(time.time())    
-    fecha_hora = datetime.fromtimestamp(timestamp_actual)
-    fecha_hora_legible = fecha_hora.strftime('%Y-%m-%d %H:%M:%S')
-    return fecha_hora_legible
+    """
+    Devuelve la fecha y hora actual en la zona horaria de la Ciudad de México (GMT-6).
+    """
+    # 1. Definir la zona horaria de la Ciudad de México
+    # Puedes usar 'America/Mexico_City' para que pytz maneje el horario de verano automáticamente.
+    mexico_city_tz = pytz.timezone('America/Mexico_City')
+
+    # 2. Obtener la hora actual en UTC
+    utc_now = datetime.now(pytz.utc)
+
+    # 3. Convertir la hora UTC a la zona horaria deseada
+    mexico_city_now = utc_now.astimezone(mexico_city_tz)
+
+    # 4. Formatear la fecha y hora
+    # El formato que deseas es "YYYY-MM-DD HH:MM:SS"
+    formatted_time = mexico_city_now.strftime("%Y-%m-%d %H:%M:%S")
+
+    return formatted_time
 
 def esDiaSiguiente(): 
     """
@@ -132,5 +136,4 @@ def esDiaSiguiente():
     print(f"Estoy comparando fecha registro {fecha_registro} con fecha actual {fecha_actual}...")
     
     # Verificar si las fechas son diferentes
-    return fecha_actual > fecha_registro
-
+    return fecha_actual > fecha_registro

nextTest.py

@@ -1,4 +0,0 @@
-import herramientas
-
-print("Resultado: ")
-print(herramientas.esDiaSiguiente(1746553638))

requirements.txt

@@ -3,4 +3,5 @@ fastapi[standard]
 huggingface_hub
 gradio_client
 Pillow
-firebase_admin
+firebase_admin
+pytz