Estadísticas Diarias y por Servidor

conexion_firebase.py

@@ -131,6 +131,8 @@ def buscar_log_por_prompt(valor_prompt):
 def incrementar_campo_numerico(collection_name, document_id, field_name, amount=1):
     """
     Incrementa un campo numérico en un documento de Firestore de forma atómica.
+    Si el documento no existe, lo crea e inicializa el campo con el 'amount'.
+    Si el campo no existe en un documento existente, lo inicializa y aplica el incremento.
 
     Args:
         collection_name (str): El nombre de la colección.
@@ -141,10 +143,14 @@ def incrementar_campo_numerico(collection_name, document_id, field_name, amount=
     doc_ref = db.collection(collection_name).document(document_id)
 
     try:
-        # Usa firestore.Increment(cantidad) para un incremento atómico
-        doc_ref.update({
-            field_name: firestore.Increment(amount)
-        })
-        print(f"✔️ Campo '{field_name}' en el documento '{document_id}' incrementado en {amount}.")
+        # Usamos .set() con merge=True para comportamiento de "upsert".
+        # Si el documento no existe, lo crea.
+        # Si el campo no existe, lo crea e inicializa con 'amount'.
+        # Si el campo ya existe, lo incrementa con 'amount'.
+        doc_ref.set(
+            {field_name: firestore.Increment(amount)},
+            merge=True  # Esta es la clave para que se cree si no existe y no sobrescriba otros campos
+        )
+        print(f"✔️ Campo '{field_name}' en el documento '{document_id}' actualizado/creado e incrementado en {amount}.")
     except Exception as e:
-        print(f"❌ Error al incrementar campo '{field_name}' en '{document_id}': {e}")
+        print(f"❌ Error al operar en el campo '{field_name}' del documento '{document_id}': {e}")

funciones.py

@@ -1,10 +1,10 @@
 import io
-import time
-import globales
 import random
+import globales
 import herramientas
 import gradio_client
 import conexion_firebase
+from datetime import datetime
 from fastapi import HTTPException, status
 from huggingface_hub import InferenceClient
 
@@ -47,9 +47,20 @@ def genera_platillo_gpu(platillo):
 
         print("Platillo generado:", platillo)
         conexion_firebase.registraLog(servidor, platillo, espacio_aleatorio_elegido, 42)
+        
+        #Generales
         conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'imagenes_totales', amount=1)
         conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'endpoint-' + servicio, amount=1)
-        conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'imagenes_gpu', amount=1)
+        conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'gpu-totales', amount=1)
+        conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'gpu-' + servicio, amount=1)
+        #Día a Día
+        now = datetime.now() 
+        document_id_fecha_hora = now.strftime("%Y-%m-%d")
+        nombre_doc = servidor + " - " + document_id_fecha_hora
+        conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'imagenes_totales', amount=1)
+        conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'endpoint-' + servicio, amount=1)
+        conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'gpu-totales', amount=1)
+        conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'gpu-' + servicio, amount=1)    
 
         return result[0]
 
@@ -69,13 +80,11 @@ def genera_platillo_inference(platillo):
         ]
     
     print(f"Modelos habilitados: {modelos_habilitados}")
-
     modelo_aleatorio_elegido = random.choice(modelos_habilitados)
     configuracion_modelo = dict_modelos[modelo_aleatorio_elegido]   
 
     creditos_restantes_inference = conexion_firebase.obtenDato('nowme', servidor, 'inferencias')
     
-    #print("Los créditos restantes de hf-inference que tienes son: ", creditos_restantes_inference)
     if creditos_restantes_inference > 0:
         provedor_seleccionado = globales.proveedor
     else:
@@ -98,9 +107,8 @@ def genera_platillo_inference(platillo):
         height=560, #568
         num_inference_steps=12,
         seed=42
-        ) 
+        )
 
-        #Detenido momentaneamente por cambio a firebase.
         herramientas.restaSegundosInference(globales.inference_cost)              
         
     except Exception as e:
@@ -119,7 +127,18 @@ def genera_platillo_inference(platillo):
     img_io.seek(0)
     print("Platillo generado:", platillo)
     conexion_firebase.registraLog(servidor, platillo, modelo_aleatorio_elegido, 42)
+    #Generales
     conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'imagenes_totales', amount=1)
     conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'endpoint-' + servicio, amount=1)
-    conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'imagenes_inference', amount=1)
+    conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'inference-totales', amount=1)
+    conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'inference-' + servicio, amount=1)
+    #Día a Día
+    now = datetime.now() 
+    document_id_fecha_hora = now.strftime("%Y-%m-%d")
+    nombre_doc = servidor + " - " + document_id_fecha_hora    
+    conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'imagenes_totales', amount=1)
+    conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'endpoint-' + servicio, amount=1)
+    conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'inference-totales', amount=1)
+    conexion_firebase.incrementar_campo_numerico('estadisticas', nombre_doc, 'inference-' + servicio, amount=1)
+
     return img_io