Falai provider

funciones.py

@@ -58,7 +58,6 @@ def genera_platillo_inference(platillo):
     else:
         provedor_seleccionado = globales.proveedor_back
 
-
     prompt = globales.previo + platillo
     print("Platillo enviado:", platillo)
        
@@ -71,7 +70,9 @@ def genera_platillo_inference(platillo):
         image = client.text_to_image(
         prompt,
         model=modelo
-        )                 
+        )   
+
+        herramientas.restaSegundosInference(globales.inference_cost)              
         
     except Exception as e:
         print("Excepción: ", e)

globales.py

@@ -9,6 +9,7 @@ espacio = "black-forest-labs/FLUX.1-schnell"
 #inferencia = "black-forest-labs/FLUX.1-dev"
 #inferencia_backup = "black-forest-labs/FLUX.1-schnell"
 proveedor =  "hf-inference" #falai
-proveedor_back = "falai"
-work_cost = 6 #Los segundos que cuesta la generación de una imagen, para control de la capa gratuita.
+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

herramientas.py

@@ -28,6 +28,24 @@ def obtenSegundosDisponibles():
 
     #Finalmente obten los segundos disponibles después de las operaciones.
     return conexion_firebase.obtenDato('nowme', 'huggingface', segundos)
+
+def obtenSegundosDisponiblesInference(): 
+
+    # if esDiaSiguiente() == True:
+    #     renuevaSegundosDisponibles()
+    #     renuevaModeloPrincipal()
+
+    #Finalmente obten los segundos disponibles después de las operaciones.
+    return conexion_firebase.obtenDato('nowme', 'huggingface', 'hfInference')
+
+def obtenSegundosDisponiblesInference(): 
+
+    if esDiaSiguiente() == True:
+        renuevaSegundosDisponibles()
+        renuevaModeloPrincipal()
+
+    #Finalmente obten los segundos disponibles después de las operaciones.
+    return conexion_firebase.obtenDato('nowme', 'huggingface', segundos)
     
 def renuevaSegundosDisponibles():
 
@@ -55,6 +73,20 @@ def restaSegundosGPU(cuantos_segundos):
     print("Segundos disponibles ahora: ", nuevos_segundos_disponibles)
     conexion_firebase.editaDato('nowme', 'huggingface', segundos, nuevos_segundos_disponibles)
 
+def restaSegundosInference(cuantos_segundos):
+    """
+    Lee el número de segundos disponibles desde seconds_available.txt,
+    resta los segundos dados como parámetro y guarda el nuevo valor en el archivo.
+    """
+
+    segundos_disponibles = obtenSegundosDisponiblesInference()    
+    print("Segundos disponibles: ", segundos_disponibles)
+
+    # Restar los segundos
+    nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
+    print("Segundos disponibles ahora: ", nuevos_segundos_disponibles)
+    conexion_firebase.editaDato('nowme', 'huggingface', 'hfInference', nuevos_segundos_disponibles)
+
 def modificaModeloActual(nuevo_modelo):
     """
     Actualiza el archivo archivos/modelo_actual.txt con el modelo funcional en caso de