Manejo centralizado de disponibilidad de espacios y modelos

app.py

@@ -45,11 +45,9 @@ async def genera_imagen(request: Request, platillo: str = Form(...)):
             return resultado
         else:
             return FileResponse(resultado, media_type="image/png", filename="imagen.png")
-    else: 
-        
+    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:

autenticacion.py

@@ -22,7 +22,7 @@ def defineAmbiente():
         hora_renovacion = 17
         servicio = "desarrollo"
     else:
-        print("Entorno remoto listo...")        
+        #print("Entorno remoto listo...")        
         llave = os.getenv("llave") #Acceso a HF
         servidor = os.getenv("servidor")
         print("El servidor remoto es: ", servidor)

buscaLog.py

@@ -15,5 +15,5 @@ if __name__ == "__main__":
     else:
         print("❌ Búsqueda finalizada sin resultados.")
 
-    print("\n--- Probando una búsqueda que no debería encontrar nada ---")
+    #print("\n--- Probando una búsqueda que no debería encontrar nada ---")
     conexion_firebase.buscar_log_por_prompt("Prompt que no existe")

cloudfunction.py

@@ -45,7 +45,7 @@ def incrementar_segundos_python(event: pubsub_fn.CloudEvent) -> None:
         doc_ref.update({
             "segundos": firestore.FieldValue.increment(1500)
         })
-        print("Campo 'segundos' actualizado con éxito (+1500) en el documento 'nowme/sosa'.")
+        # print("Campo 'segundos' actualizado con éxito (+1500) en el documento 'nowme/sosa'.")
 
     except exceptions.NotFound:
         print("Error: Documento nowme/sosa no encontrado. No se pudo actualizar.")
@@ -55,10 +55,8 @@ def incrementar_segundos_python(event: pubsub_fn.CloudEvent) -> None:
         # doc_ref.set({"segundos": 1500}, merge=True) # Ejemplo de cómo crear/actualizar
 
     except Exception as e:
-        print(f"Error al actualizar el documento nowme/sosa: {e}")
+        #print(f"Error al actualizar el documento nowme/sosa: {e}")
         # Puedes lanzar la excepción si quieres que Cloud Functions lo reporte como un fallo.
         # raise e
         pass # O simplemente pasar para que la Function no se marque como fallo en este caso
 
-    print("Fin de la ejecución de la Function.")
-

conexion_firebase.py

@@ -10,12 +10,11 @@ firebase_admin.initialize_app(firebase_cred)
 db = firestore.client(database_id='nowme')
 #dato es el dato que traes  como el nombre del user. 
 #info es la info de ese dato que estás buscando, como token.
-def obtenDato(coleccion, dato, info):
+def obtenDato(coleccion, documento, info):
     #Future: Tentativamente ésta parte podría solo hacerse una vez y vivir en la app para ser reutilizado.
     
     #Primero debemos definir la referencia al documento, o sea a la hoja de usuario.
-    doc_ref = db.collection(coleccion).document(dato)    
-    #Éste es el documento que tiene los datos de ella.
+    doc_ref = db.collection(coleccion).document(documento)    
     documento = doc_ref.get()
     
     #Recuerda la conversión a diccionario.
@@ -23,13 +22,23 @@ def obtenDato(coleccion, dato, info):
     diccionario = documento.to_dict()
     return diccionario.get(info)
 
+def obtenAllDatos(coleccion, documento):
+
+    doc_ref = db.collection(coleccion).document(documento) 
+    documento = doc_ref.get()
+
+    #Recuerda la conversión a diccionario.
+    documento = doc_ref.get() 
+    diccionario = documento.to_dict()
+
+    return diccionario
+
 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)
         
     doc_ref.update({
-        # 'quote': quote,
         info: contenido,
     })
 
@@ -89,7 +98,7 @@ def registraLog(servidor_val, prompt_val, modelo_val, seed_val):
         # milisegundos al ID o usar un ID automático y un campo de timestamp.
         doc_ref.set(datos_log)
 
-        print(f"✔️ Log agregado exitosamente en 'logs/{document_id_fecha_hora}'.")
+        #print(f"✔️ Log agregado exitosamente en 'logs/{document_id_fecha_hora}'.")
         print(f"   Datos: Servidor='{servidor_val}', Prompt='{prompt_val}'...")
 
     except Exception as e:
@@ -153,4 +162,39 @@ def incrementar_campo_numerico(collection_name, document_id, field_name, amount=
         )
         print(f"✔️ Campo '{field_name}' en el documento '{document_id}' actualizado/creado e incrementado en {amount}.")
     except Exception as e:
-        print(f"❌ Error al operar en el campo '{field_name}' del documento '{document_id}': {e}")
+        print(f"❌ Error al operar en el campo '{field_name}' del documento '{document_id}': {e}")
+
+def obtener_documentos_habilitados(coleccion):
+    """
+    Obtiene una lista de los IDs de todos los documentos en la colección
+    cuyo campo 'habilitado' es true.
+
+    Args:
+        coleccion (str): El nombre de la colección a consultar.
+
+    Returns:
+        list: Una lista de strings, donde cada string es el ID de un documento
+              habilitado.
+    """
+    print(f"--- Buscando documentos habilitados en la colección '{coleccion}' ---")
+    
+    try:
+        # 1. Referencia a la colección con el nombre del parámetro
+        query_results = db.collection(coleccion).where(
+            filter=firestore.FieldFilter('habilitado', '==', True)
+        ).get()
+        
+        # 2. Itera sobre los resultados y almacena solo el ID en una lista
+        # Usamos una lista por comprensión para una mayor eficiencia y simplicidad
+        documentos_habilitados_ids = [doc.id for doc in query_results]
+        
+        if not documentos_habilitados_ids:
+            print("❌ No se encontraron documentos habilitados.")
+            return []
+            
+        print(f"✔️ Se encontraron {len(documentos_habilitados_ids)} IDs de documentos.")
+        return documentos_habilitados_ids
+        
+    except Exception as e:
+        print(f"❌ Error al obtener documentos habilitados: {e}")
+        return []

funciones.py

@@ -16,19 +16,14 @@ def genera_platillo_gpu(platillo):
     prompt = globales.previo + platillo
     
     try:
-        dict_espacios = conexion_firebase.obtenDato('nowme', servidor, 'espacios')
+        espacios = conexion_firebase.obtener_documentos_habilitados('espacios')
 
-        espacios_habilitados = [
-        nombre for nombre, config in dict_espacios.items()
-        if config.get("habilitado", False) # Usamos .get() para evitar KeyError si 'habilitado' no existe
-        ]
-
-        print(f"Espacios habilitados: {espacios_habilitados}")
-        
-        espacio_aleatorio_elegido = random.choice(espacios_habilitados)
-        configuracion_espacio = dict_espacios[espacio_aleatorio_elegido]
-        client = gradio_client.Client(configuracion_espacio['ruta'], hf_token=globales.llave)
+        print(f"Espacios habilitados: {espacios}")
+        espacio_elegido = random.choice(espacios)
+        datos_espacio = conexion_firebase.obtenAllDatos('espacios', espacio_elegido)
         
+        client = gradio_client.Client(datos_espacio['ruta'], hf_token=globales.llave)
+     
         result = client.predict(
         #**kwargs,
         prompt=prompt,
@@ -39,14 +34,14 @@ def genera_platillo_gpu(platillo):
         height=568,
         # guidance_scale=3.5,
         # num_inference_steps=28,
-        api_name=configuracion_espacio['api_name']       
+        api_name=datos_espacio['api_name']       
         )
 
         #Cuando es GPU, debe de restar segundos disponibles de HF 
         herramientas.restaSegundosGPU(globales.work_cost)
 
         print("Platillo generado:", platillo)
-        conexion_firebase.registraLog(servidor, platillo, espacio_aleatorio_elegido, 42)
+        conexion_firebase.registraLog(servidor, platillo, espacio_elegido, 42)
         
         #Generales
         conexion_firebase.incrementar_campo_numerico('estadisticas', 'imagenes', 'imagenes_totales', amount=1)
@@ -73,15 +68,13 @@ def genera_platillo_gpu(platillo):
 
 def genera_platillo_inference(platillo):
     
-    dict_modelos = conexion_firebase.obtenDato('nowme', servidor, 'modelos')    
-    modelos_habilitados = [
-        nombre for nombre, config in dict_modelos.items()
-        if config.get("habilitado", False) # Usamos .get() para evitar KeyError si 'habilitado' no existe
-        ]
+    modelos = conexion_firebase.obtener_documentos_habilitados('modelos')
+
+    print(f"Espacios habilitados: {modelos}")
+    modelo_elegido = random.choice(modelos)
+    datos_modelo = conexion_firebase.obtenAllDatos('espacios', modelo_elegido)
     
-    print(f"Modelos habilitados: {modelos_habilitados}")
-    modelo_aleatorio_elegido = random.choice(modelos_habilitados)
-    configuracion_modelo = dict_modelos[modelo_aleatorio_elegido]   
+    client = gradio_client.Client(datos_modelo['ruta'], hf_token=globales.llave)
 
     creditos_restantes_inference = conexion_firebase.obtenDato('nowme', servidor, 'inferencias')
     
@@ -95,7 +88,7 @@ def genera_platillo_inference(platillo):
        
     client = InferenceClient(
             provider= provedor_seleccionado,
-            model=configuracion_modelo['ruta'],
+            model=datos_modelo['ruta'],
             api_key=globales.llave
         )
 
@@ -126,7 +119,7 @@ def genera_platillo_inference(platillo):
     image.save(img_io, "PNG")
     img_io.seek(0)
     print("Platillo generado:", platillo)
-    conexion_firebase.registraLog(servidor, platillo, modelo_aleatorio_elegido, 42)
+    conexion_firebase.registraLog(servidor, platillo, modelo_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)

herramientas.py

@@ -60,7 +60,7 @@ def restaSegundosInference(cuantos_segundos):
 
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
-    print("Procesado, segundos disponibles ahora:  ", nuevos_segundos_disponibles)
+    #print("Procesado, segundos disponibles ahora:  ", nuevos_segundos_disponibles)
     conexion_firebase.editaDato('nowme', servidor, 'inferencias', nuevos_segundos_disponibles)    
 
 def modificaModeloActual(nuevo_modelo):
@@ -114,14 +114,12 @@ def siEsDiaSiguienteRenueva():
 
     if resultado == True:        
         if diferencia.days > 1:
-            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()
                          
             if int(hora_actual.hour) > int(globales.hora_renovacion):               
-                print("Renovando segundos.")
                 renuevaSegundosDisponibles()
                 renuevaTimestampActual()
             else:
@@ -150,7 +148,7 @@ def despliegaInfoCliente(request):
     user_id = "unauthenticated"
     if auth_header and auth_header.startswith("Bearer "):
         token = auth_header.split(" ")[1]
-        # Aquí normalmente decodificarías el token JWT para obtener el user_id
+        # Aquí se haría decodificación del token JWT para obtener el user_id
         # Ejemplo simulado:
         if token == "mysecrettoken123":
             user_id = "user123_authenticated"
@@ -160,4 +158,4 @@ def despliegaInfoCliente(request):
 
     print(f"Petición desde: {client_ip}")
     print(f"User-Agent: {user_agent}")
-    print(f"Usuario (Auth): {user_id}")
+    #print(f"Usuario (Auth): {user_id}")

test.py

@@ -0,0 +1,4 @@
+import conexion_firebase 
+
+datos = conexion_firebase.obtenAllDatos('modelos', 'flux-dev')
+print(datos)