{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": []
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "code",
      "source": [
        "import os\n",
        "from IPython.utils import capture\n",
        "\n",
        "def check_colab_subscription():\n",
        "    return (os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES') / (1024. ** 3) <= 20)\n",
        "free_plan = check_colab_subscription()\n",
        "\n",
        "def detect_environment():\n",
        "    environments = {\n",
        "        'COLAB_GPU': ('Google Colab', \"/root\" if free_plan else \"/content\"),\n",
        "        'KAGGLE_URL_BASE': ('Kaggle', \"/kaggle/working/content\"),\n",
        "        'SAGEMAKER_INTERNAL_IMAGE_URI': ('SageMaker Studio Lab', \"/home/studio-lab-user/content\")\n",
        "    }\n",
        "\n",
        "    for env_var, (environment, path) in environments.items():\n",
        "        if env_var in os.environ:\n",
        "            return environment, path\n",
        "\n",
        "    print(\"\\033[31mError: an unsupported runtime environment was detected.\\n\\033[34mSupported environments:\\033[0m Google Colab, Kaggle, Sagemaker Studio Lab\")\n",
        "    return None, None\n",
        "\n",
        "env, root_path = detect_environment()\n",
        "if env and root_path:\n",
        "    webui_path = f\"{root_path}/sdw\"\n",
        "\n",
        "    print(f\"Runtime environment: \\033[33m{env}\\033[0m\")\n",
        "    # Colab Plan\n",
        "    if env == \"Google Colab\":\n",
        "        print(f\"Colab Pro subscription: \\033[34m{not free_plan}\\033[0m\")\n",
        "        print(f\"File location: \\033[32m{root_path}\\033[0m\")\n",
        "\n",
        "    print(\"Please wait for the files to download.... 👀\", end='')\n",
        "    with capture.capture_output() as cap:\n",
        "        files = [f'widgets_{lang}.py', f'downloading_{lang}.py', f'launch_{lang}.py', f'auto-cleaner_{lang}.py']\n",
        "        folder_path = f'{root_path}/file_cell'\n",
        "\n",
        "        if os.path.exists(folder_path):\n",
        "            !rm -rf {folder_path}\n",
        "\n",
        "        !mkdir -p {folder_path}\n",
        "\n",
        "        for file in files:\n",
        "            !wget -q https://huggingface.co/NagisaNao/fast_repo/resolve/main/files_cells/python/{lang}/{file} -O {folder_path}/{file}\n",
        "\n",
        "    del cap\n",
        "    print(\"\\rDone! Now you can run the cells below. ☄️\" + \" \"*30)"
      ],
      "metadata": {
        "id": "osk658gR0Wu7"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}