diff --git a/.github/workflows/docs.py b/.github/workflows/docs.py
new file mode 100644
index 000000000..ef3f3c598
--- /dev/null
+++ b/.github/workflows/docs.py
@@ -0,0 +1,33 @@
+"""Script to generate docs based on given input directory (first argument)
+and output directory (second argument)"""
+import os
+import subprocess
+import sys
+import glob
+import nbformat
+
+code_path = sys.argv[1]
+out_path = sys.argv[2]
+
+for notebook_path in glob.glob(f"{code_path}/examples/PySDM_examples/*/*.ipynb"):
+    with open(notebook_path, encoding="utf8") as fin:
+        with open(notebook_path + ".badges.md", "w", encoding="utf8") as fout:
+            fout.write(nbformat.read(fin, nbformat.NO_CONVERT).cells[0].source)
+subprocess.run(
+    [
+        sys.executable,
+        "-We",
+        "-m",
+        "pdoc",
+        "-o",
+        f"{out_path}/html",
+        "PySDM",
+        "examples/PySDM_examples",
+        "-t",
+        "docs/templates",
+        "--math",
+        "--mermaid",
+    ],
+    env={**os.environ, "PDOC_ALLOW_EXEC": "1"},
+    check=True,
+)
diff --git a/.github/workflows/pdoc.yml b/.github/workflows/pdoc.yml
index 03dd1bb34..72cb8fc80 100644
--- a/.github/workflows/pdoc.yml
+++ b/.github/workflows/pdoc.yml
@@ -31,17 +31,11 @@ jobs:
           pip install -e .
           pip install -e examples
 
-          python - <<EOF
-          import glob, nbformat
-          for notebook_path in glob.glob('examples/PySDM_examples/*/*.ipynb'):
-            with open(notebook_path, encoding="utf8") as fin:
-              with open(notebook_path + ".badges.md", 'w') as fout:
-                fout.write(nbformat.read(fin, nbformat.NO_CONVERT).cells[0].source)
-          EOF
-          PDOC_ALLOW_EXEC=1 python -We -m pdoc -o html PySDM examples/PySDM_examples -t docs/templates --math --mermaid
+          python -We .github/workflows/docs.py . .
+        
       - if: ${{ github.ref == 'refs/heads/main' && matrix.platform == 'ubuntu-latest' }}
         uses: JamesIves/github-pages-deploy-action@4.1.1
         with:
           branch: pdoc
           folder: html
-          clean: true
\ No newline at end of file
+          clean: true