🎨 add boolean for colab

Author: enryH

docs/vuegen_basic_case_study.ipynb

@@ -54,13 +54,17 @@
    "source": [
     "### 0.1. Installing libraries and creating global variables for platform and working directory\n",
     "\n",
-    "To run this notebook locally, you should create a virtual environment with the required libraries and set the `platform` variable to `local_notebook`. If you are running this notebook on Google Colab, set the `platform` variable to `colab`."
+    "To run this notebook locally, you should create a virtual environment with the required libraries. If you are running this notebook on Google Colab, everything should be set."
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "tags": [
+     "hide-output"
+    ]
+   },
    "outputs": [],
    "source": [
     "# Vuegen library \n",
@@ -73,7 +77,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "platform = 'local_notebook' #@param ['colab', 'local_notebook'] {allow-input: true}"
+    "IN_COLAB = \"COLAB_GPU\" in os.environ"
    ]
   },
   {
@@ -83,11 +87,11 @@
    "outputs": [],
    "source": [
     "# Set working directory\n",
-    "if platform == 'colab':\n",
+    "if IN_COLAB:\n",
     "    # Clone the repository in Colab\n",
     "    !git clone --depth=1 https://github.com/Multiomics-Analytics-Group/vuegen.git\n",
     "    base_output_dir = \"vuegen/docs/example_data/Basic_example_vuegen_demo_notebook/\"\n",
-    "elif platform == 'local_notebook':\n",
+    "else:\n",
     "    # Output directory for local execution\n",
     "    base_output_dir = \"./example_data/Basic_example_vuegen_demo_notebook/\""
    ]
@@ -99,7 +103,7 @@
    "outputs": [],
    "source": [
     "# Optional library to launch a streamlit app from colab\n",
-    "if platform == 'colab':\n",
+    "if IN_COLAB:\n",
     "    !npm install localtunnel"
    ]
   },
@@ -112,7 +116,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 27,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -122,7 +126,7 @@
     "from vuegen import report_generator\n",
     "from vuegen.utils import get_logger, load_yaml_config\n",
     "\n",
-    "if platform == 'colab':\n",
+    "if IN_COLAB:\n",
     "    import urllib"
    ]
   },
@@ -310,7 +314,7 @@
    "outputs": [],
    "source": [
     "# Launch the Streamlit report depneding on the platform\n",
-    "#if platform == \"local_notebook\":\n",
+    "#if not IN_COLAB:\n",
     "#    !streamlit run streamlit_report/sections/report_manager.py\n",
     "#elif platform == \"colab\":\n",
     "#    print(\"Password/Enpoint IP for localtunnel is:\",urllib.request.urlopen('https://ipv4.icanhazip.com').read().decode('utf8').strip(\"\\n\"))\n",