import subprocess
try:
    import requests
except:
    subprocess.run(["pip3", "install", "requests"])
    import requests
import os
import zipfile
import io

url = os.getenv("URL")

def process(url):
    respuesta = requests.get(url)
    if respuesta.status_code != 200:
        raise Exception(f"Error al descargar el archivo. Código de estado: {respuesta.status_code}")
    if 'zip' in respuesta.headers.get('Content-Type', '').lower():
        with zipfile.ZipFile(io.BytesIO(respuesta.content)) as archivo_zip:
            archivo_zip.extractall()
            print(f"Archivos extraídos en: {os.getcwd()}")
    else:
        nombre_archivo = url.split('/')[-1]
        with open(nombre_archivo, 'wb') as archivo:
            archivo.write(respuesta.content)
        print(f"Archivo descargado: {nombre_archivo}")
process(url)