Diccionario de Modelos Disponibles

app.py

@@ -34,8 +34,6 @@ async def genera_imagen(platillo: str = Form(...)):
 
     seconds_available = herramientas.obtenSegundosDisponibles()
 
-    #seconds_available = 0 
-
     print(herramientas.imprimeTimeNow())
     
     if seconds_available > globales.work_cost:

espacios.py

@@ -0,0 +1,41 @@
+import herramientas
+
+server =  herramientas.getFluxServer()
+print("Server elegido: ", server)
+print("El type es: ", type(server))
+
+espacio = [
+    {
+        "id": "black-forest-labs/FLUX.1-dev",
+        "static_kwargs": { 
+            "api_name": "/infer",
+        }
+    },
+    # { 
+    #     "id": "black-forest-labs/FLUX.1-schnell",
+    #     "static_kwargs": {
+    #         "api_name": "/infer",
+    #     }
+    # },
+    # {
+    #     "id": "NihalGazi/FLUX-Pro-Unlimited",
+    #     "static_kwargs": {
+    #         "api_name": "/generate_image", 
+    #         "server_choice": server,
+    #     }
+    # },
+    # { 
+    #     "id": "HiDream-ai/HiDream-I1-Dev",
+    #     "static_kwargs": {
+    #         "api_name": "/infer",
+    #      
+    #     }
+    # },
+    # { 
+    #     "id": "stabilityai/stable-diffusion",
+    #     "static_kwargs": {
+    #         "api_name": "/generate",
+    #         
+    #     }
+    # }
+]

funciones.py

@@ -1,25 +1,33 @@
 import io
 import globales
+import espacios
 import herramientas
 import gradio_client
 from huggingface_hub import InferenceClient
 import conexion_firebase
+import random
+import time
 
 def genera_platillo_gpu(platillo):
 
     prompt = globales.previo + platillo
-    print("Platillo enviado:", platillo)    
-
-    kwargs = {
-            "prompt": prompt,
-            "api_name": "/infer"
-        }    
+    print("Platillo enviado:", platillo) 
 
     try:
 
-        client = gradio_client.Client(globales.espacio, hf_token=globales.llave)
+        selected_space_config = random.choice(espacios.espacio)
+
+        client = gradio_client.Client(selected_space_config['id'], hf_token=globales.llave)
+
+        # kwargs = {
+        #     "prompt": prompt,
+        #     "api_name": "/infer"
+        # } 
+
+        kwargs = selected_space_config['static_kwargs']
+
         result = client.predict(**kwargs,
-        # prompt=prompt,
+        prompt=prompt,
         # negative_prompt="",
         # seed=42,
         # randomize_seed=True,
@@ -30,6 +38,8 @@ def genera_platillo_gpu(platillo):
         # api_name="/infer"       
         )
 
+        #print("Result inmediato es: ", result)
+
         #Cuando es GPU, debe de restar segundos disponibles de HF 
         herramientas.restaSegundosGPU(globales.work_cost)
 
@@ -37,7 +47,7 @@ def genera_platillo_gpu(platillo):
         return result[0]
 
     except Exception as e:     
-        print("Excepción: ", e)
+        print("Excepción: ", e)        
         # Opción para regresar imagen genérica. (ya no porque se envía desde backend.)
         # return "default.png"
         return '{"Error 500": e}'
@@ -49,15 +59,13 @@ def genera_platillo_inference(platillo):
     modelo_actual = conexion_firebase.obtenDato('nowme', 'huggingface', 'modelo_actual')
     modelo = modelo_actual
 
-    #modelo = globales.inferencia
-
     print("Modelo:", modelo)
     
     #Elegir proveedor de inferencia
     creditos_restantes_inference = conexion_firebase.obtenDato('nowme', 'huggingface', 'hfInference')
     #creditos_restantes_inference = 5000
 
-    print("Los créditos restantes de hf-inference que tienes son: ", creditos_restantes_inference)
+    #print("Los créditos restantes de hf-inference que tienes son: ", creditos_restantes_inference)
     if creditos_restantes_inference > 0:
         provedor_seleccionado = globales.proveedor
     else:

globales.py

@@ -1,10 +1,21 @@
 import autenticacion
-version = "0.0.0"
+version = "0.0.1"
 
 #previo = "Una fotografía de un plato blanco con "
 previo = "A photograph of a white plate with "  
 llave, servidor = autenticacion.defineAmbiente()
-espacio = "black-forest-labs/FLUX.1-schnell"
+
+
+# espacio = [
+#             "black-forest-labs/FLUX.1-dev",
+#             #"black-forest-labs/FLUX.1-schnell",
+#             "NihalGazi/FLUX-Pro-Unlimited",
+#             #"HiDream-ai/HiDream-I1-Dev",
+#             #"stabilityai/stable-diffusion"
+
+#           ]
+
+
 #espacio = "black-forest-labs/FLUX.1-dev"
 inferencia = "black-forest-labs/FLUX.1-dev"
 inferencia_backup = "black-forest-labs/FLUX.1-schnell"
@@ -13,3 +24,16 @@ 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
+
+
+SERVER_NAMES = {
+    "google_us": "Google US Server",
+    #"azure_lite": "Azure Lite Supercomputer Server",
+    #"artemis" : "Artemis GPU Super cluster",
+    #"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

@@ -4,6 +4,7 @@ import conexion_firebase
 import globales
 import pytz
 import time
+import random
 
 if globales.servidor == "moibe":
     segundos = "segundos_dev"
@@ -17,7 +18,7 @@ def obtenUltimoTimestamp():
     Obtiene el último timestamp de renovación guardado.
     """
     resultado = conexion_firebase.obtenDato('nowme', 'huggingface', last_timestamp)
-    print("Obten último timestamp obtenido: ", resultado)    
+    #print("Obten último timestamp obtenido: ", resultado)    
     
     return resultado
     
@@ -30,16 +31,6 @@ def obtenSegundosDisponibles():
     #Finalmente obten los segundos disponibles después de las operaciones.
     return conexion_firebase.obtenDato('nowme', 'huggingface', segundos)
 
-def obtenSegundosDisponibles(): 
-
-    if esDiaSiguiente() == True:
-        renuevaSegundosDisponibles()
-        renuevaModeloPrincipal()
-
-    #Finalmente obten los segundos disponibles después de las operaciones.
-    return conexion_firebase.obtenDato('nowme', 'huggingface', segundos)
-    
-
 def obtenSegundosDisponiblesInference(): 
 
     # if esDiaSiguiente() == True:
@@ -73,7 +64,7 @@ def restaSegundosGPU(cuantos_segundos):
 
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
-    print("Segundos disponibles ahora: ", nuevos_segundos_disponibles)
+    Segundos disponibles ahora: ", nuevos_segundos_disponibles)
     conexion_firebase.editaDato('nowme', 'huggingface', segundos, nuevos_segundos_disponibles)
 
 def restaSegundosInference(cuantos_segundos):
@@ -83,7 +74,7 @@ def restaSegundosInference(cuantos_segundos):
     """
 
     segundos_disponibles = obtenSegundosDisponiblesInference()    
-    print("Segundos disponibles Inference: ", segundos_disponibles)
+    #print("Segundos disponibles Inference: ", segundos_disponibles)
     
     # Restar los segundos
     nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
@@ -207,7 +198,14 @@ def esDiaSiguiente():
 
     # Verificar si las fechas son diferentes
     resultado = fecha_actual > fecha_registro
-    print("Resultado de resta de fechas: ", resultado)
+    print("Resultado de resta de fechas: ", resultado)    
     
+    return resultado
+
+def getFluxServer(): 
+    print("In getFluxServer...")
+    server_keys = list(globales.SERVER_NAMES.values())
+    random_value = random.choice(server_keys)
+    print("Random key: ", random_value)
     
-    return resultado
+    return random_value