Changes in demos to accomodate colab

Author: orzel320

notebooks/demo.ipynb

@@ -12,7 +12,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
+   "id": "93482e50",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from google.colab import output\n",
+    "output.enable_custom_widget_manager()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
    "id": "973e71b6",
    "metadata": {},
    "outputs": [
@@ -4430,29 +4441,24 @@
     "from nbody_sim.api import Simulation, ureg\n",
     "from nbody_sim.render import create_plotly_animation\n",
     "\n",
-    "print(\"Initializing N-Body Physics Engine...\")\n",
+    "print(\"Initializing n-body physics engine\")\n",
     "sim = Simulation()\n",
     "\n",
-    "# 1. Generate 500 procedural asteroids\n",
     "np.random.seed(42)\n",
     "n_bodies = 500\n",
     "\n",
-    "# Using your Pint unit registry for strict physical definitions\n",
     "positions = np.random.uniform(-5.0, 5.0, (n_bodies, 3)) * ureg.astronomical_unit\n",
     "velocities = np.random.uniform(-0.5, 0.5, (n_bodies, 3)) * (ureg.astronomical_unit / ureg.nbody_time)\n",
     "masses = np.random.uniform(0.1, 2.0, n_bodies) * ureg.solar_mass\n",
     "\n",
-    "# Add them to the Barnes-Hut tree\n",
     "sim.add_bodies(positions, velocities, masses)\n",
     "\n",
-    "# Generate procedural colours (RGBA floats) and sizes for the renderer\n",
     "colours = np.random.uniform(0.4, 1.0, (n_bodies, 4))\n",
     "colours[:, 3] = 1.0 # Set full opacity\n",
     "sizes = np.random.uniform(0.5, 2.0, n_bodies)\n",
     "\n",
-    "print(\"Baking 3D WebGL animation. Please wait...\")\n",
+    "print(\"Preparing animation\")\n",
     "\n",
-    "# 2. Render the interactive K3D plot\n",
     "plot = create_plotly_animation(\n",
     "    sim,\n",
     "    dt=0.01,\n",
@@ -4461,7 +4467,6 @@
     "    steps=150\n",
     ")\n",
     "\n",
-    "# Display the widget!\n",
     "plot.show()"
    ]
   }

notebooks/demo2.ipynb

@@ -29,6 +29,17 @@
     "!pip install --pre --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ \"nbody-sim-RO[vis]==1.0.4.dev0\""
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e0a76d28",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from google.colab import output\n",
+    "output.enable_custom_widget_manager()"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -71,29 +82,24 @@
     "from nbody_sim.api import Simulation, ureg\n",
     "from nbody_sim.render import create_k3d_animation\n",
     "\n",
-    "print(\"Initializing N-Body Physics Engine...\")\n",
+    "print(\"Initializing n-body physics engine\")\n",
     "sim = Simulation()\n",
     "\n",
-    "# 1. Generate 500 procedural asteroids\n",
     "np.random.seed(42)\n",
     "n_bodies = 500\n",
     "\n",
-    "# Using your Pint unit registry for strict physical definitions\n",
     "positions = np.random.uniform(-5.0, 5.0, (n_bodies, 3)) * ureg.astronomical_unit\n",
     "velocities = np.random.uniform(-0.5, 0.5, (n_bodies, 3)) * (ureg.astronomical_unit / ureg.nbody_time)\n",
     "masses = np.random.uniform(0.1, 2.0, n_bodies) * ureg.solar_mass\n",
     "\n",
-    "# Add them to the Barnes-Hut tree\n",
     "sim.add_bodies(positions, velocities, masses)\n",
     "\n",
-    "# Generate procedural colours (RGBA floats) and sizes for the renderer\n",
     "colours = np.random.uniform(0.4, 1.0, (n_bodies, 4))\n",
     "colours[:, 3] = 1.0 # Set full opacity\n",
     "sizes = np.random.uniform(0.5, 2.0, n_bodies)\n",
     "\n",
-    "print(\"Baking 3D WebGL animation. Please wait...\")\n",
+    "print(\"Preparing animation\")\n",
     "\n",
-    "# 2. Render the interactive K3D plot\n",
     "plot = create_k3d_animation(\n",
     "    sim,\n",
     "    dt=0.01,\n",
@@ -102,7 +108,6 @@
     "    steps=150\n",
     ")\n",
     "\n",
-    "# Display the widget!\n",
     "plot.display()"
    ]
   }

pyproject.toml

@@ -13,27 +13,27 @@ authors = [
     {name = "orzel320", email = "orzellll320@gmail.com"}
 ]
 dependencies = [
-    "numpy>=1.22.0",
-    "numba>=0.57.0",
-    "pint>=0.22"
+    "numpy",
+    "numba",
+    "pint"
 ]
 
 [project.optional-dependencies]
 vis = [
-    "matplotlib>=3.10.8",
-    "ipympl>=0.9.3",
-    "k3d>=2.16.0",
-    "jupyter>=1.1.1"
+    "matplotlib",
+    "ipympl",
+    "k3d",
+    "jupyter"
 ]
 
 test = [
-    "pytest>=9.0.3",
-    "pytest-cov>=7.1.0"
+    "pytest",
+    "pytest-cov"
 ]
 
 docs = [
-    "mkdocs>=1.6.1",
-    "mkdocs-material>=9.7.6"
+    "mkdocs",
+    "mkdocs-material"
 ]
 
 dev = [