Manejo de envs local y hf

funciones.py

@@ -5,13 +5,11 @@ import io
 import globales
 
 
-previo = "Una fotografía de un plato blanco con "
-
 def genera_platillo_gpu(platillo):
 
-    client = gradio_client.Client(globales.espacio, hf_token=bridges.hug)
+    client = gradio_client.Client(globales.espacio, hf_token=globales.llave)
 
-    prompt = previo + platillo
+    prompt = globales.previo + platillo
 
     print("Eso es el prompt final:", prompt)
 
@@ -44,10 +42,10 @@ def genera_platillo_inference(platillo):
 
     client = InferenceClient(
             provider= globales.proveedor,
-            api_key=bridges.hug
+            api_key=globales.llave
         )
 
-    prompt = previo + platillo
+    prompt = globales.previo + platillo
 
     try: 
         image = client.text_to_image(

globales.py

@@ -1,3 +1,8 @@
+import herramientas
+
+previo = "Una fotografía de un plato blanco con "
+llave = herramientas.obtenAccesoHF()
+
 seconds_available = 0
 
 #espacio = "black-forest-labs/FLUX.1-schnell"

herramientas.py

@@ -0,0 +1,29 @@
+import os
+import socket
+
+def obtenAccesoHF():
+    if local_check():
+        print("Estoy en LOCAL...")        
+        import bridges
+        llave = bridges.llave
+    else:
+        print("Estoy en REMOTO...")
+        llave = os.getenv("llave")
+
+        print("Ésto es llave:", llave)
+
+
+    return llave
+
+def local_check(): 
+
+    hostname = socket.gethostname()
+    #r-moibe-nowme
+    print("Dentro de local_check... , el hostname es: ", hostname)
+
+    if "r-moibe-nowme" in hostname:
+        print("Ejecutando en el servidor")
+        return False
+    else:
+        print("Ejecutando en local")
+        return True