Upload 2 files

install.py

@@ -1,42 +1,28 @@
 import subprocess
 try:
     import requests
+    from huggingface_hub import HFApi
 except:
-    subprocess.run(["pip3", "install", "requests"])
+    subprocess.run(["pip3", "install", "requests", "huggingface_hub"])
     import requests
+    from huggingface_hub import HFApi
 import os
 import zipfile
 import io
 
-# Obtener todas las variables de entorno como un diccionario
-variables = os.environ
-url = ""
-# Imprimir todas las variables de entorno
-for key, value in variables.items():
-    print(f"{key}: {value}")
-    if key == "URL":
-        url = value
-        break
-    print("[INFO] NO SE ENCONTRÓ URL")
+api = HfApi()
+url = api.get_secret("URL")
 
 def process(url):
-    # Descargar el archivo desde la URL
     respuesta = requests.get(url)
-    
-    # Verificar si la descarga fue exitosa
     if respuesta.status_code != 200:
         raise Exception(f"Error al descargar el archivo. Código de estado: {respuesta.status_code}")
-    
-    # Verificar si el archivo es un ZIP
     if 'zip' in respuesta.headers.get('Content-Type', '').lower():
-        # Crear un objeto BytesIO para manejar el contenido en memoria
         with zipfile.ZipFile(io.BytesIO(respuesta.content)) as archivo_zip:
-            # Extraer todos los archivos del ZIP en el directorio actual
             archivo_zip.extractall()
             print(f"Archivos extraídos en: {os.getcwd()}")
     else:
-        # Si no es un ZIP, guardar el archivo directamente
-        nombre_archivo = url.split('/')[-1]  # Obtener el nombre del archivo desde la URL
+        nombre_archivo = url.split('/')[-1]
         with open(nombre_archivo, 'wb') as archivo:
             archivo.write(respuesta.content)
         print(f"Archivo descargado: {nombre_archivo}")