From a41b9db51ec497f0ff525f289aee232a56fe28f0 Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Thu, 26 May 2022 17:34:05 +0200
Subject: [PATCH 01/10] Initial version of gallery using nbsphinx with a hack
 to make it work with myst_nb.

---
 examples/gallery.rst | 26 ++++++--------------------
 1 file changed, 6 insertions(+), 20 deletions(-)

diff --git a/examples/gallery.rst b/examples/gallery.rst
index 0627a4629..d72ff950d 100644
--- a/examples/gallery.rst
+++ b/examples/gallery.rst
@@ -1,20 +1,7 @@
 Gallery
 =======
 
-
-(Generalized) Linear and Hierarchical Linear Models
--------------------------
-
-.. nbgallery::
-    :caption: This is a thumbnail gallery:
-    :name: generalized_linear_models
-    :glob:
-    :reversed:
-
-    generalized_linear_models/*
-
-
-Case Studies
+Case studies
 ------------
 
 .. nbgallery::
@@ -25,17 +12,16 @@ Case Studies
 
     case_studies/*
 
-Diagnostics and Model Criticism
--------------------------------
+Generalized Linear Models
+-------------------------
 
 .. nbgallery::
     :caption: This is a thumbnail gallery:
-    :name: diagnostics_and_criticism
+    :name: generalized_linear_models
     :glob:
     :reversed:
 
-    diagnostics_and_criticism/*
-
+    generalized_linear_models/*
 
 Gaussian Processes
 ------------------
@@ -112,4 +98,4 @@ Variational Inference
     :glob:
     :reversed:
 
-    variational_inference/*
\ No newline at end of file
+    variational_inference/*

From e9765e09637988bfe352415d6eea0d32432709f8 Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Thu, 26 May 2022 19:59:27 +0200
Subject: [PATCH 02/10] Add gallery.

---
 examples/conf.py  | 63 ++---------------------------------------------
 examples/index.md | 12 +++++++++
 2 files changed, 14 insertions(+), 61 deletions(-)

diff --git a/examples/conf.py b/examples/conf.py
index d056dd706..f6ba1397b 100644
--- a/examples/conf.py
+++ b/examples/conf.py
@@ -1,6 +1,5 @@
 import os
 from sphinx.application import Sphinx
-
 # -- Project information -----------------------------------------------------
 project = "PyMC"
 copyright = "2022, PyMC Community"
@@ -22,7 +21,7 @@
     "sphinxcontrib.bibtex",
     "sphinx_codeautolink",
     "notfound.extension",
-    "sphinx_gallery.load_style",
+    'sphinx_gallery.load_style',
 ]
 
 # List of patterns, relative to source directory, that match files and
@@ -38,7 +37,6 @@
     "page_footer.md",
 ]
 
-
 def hack_nbsphinx(app: Sphinx) -> None:
     from nbsphinx import (
         depart_gallery_html,
@@ -46,23 +44,9 @@ def hack_nbsphinx(app: Sphinx) -> None:
         GalleryNode,
         NbGallery,
         patched_toctree_resolve,
-        NotebookParser,
-        NbInput,
-        NbOutput,
-        NbInfo,
-        NbWarning,
-        CodeAreaNode,
-        depart_codearea_html,
-        visit_codearea_latex,
-        depart_codearea_latex,
-        GetSizeFromImages,
     )
     from sphinx.environment.adapters import toctree
 
-    nbsphinx_thumbnails = {
-        "case_studies/stochastic_volatility": "_static/stochastic_volatility.png",
-    }
-
     def builder_inited(app: Sphinx):
         if not hasattr(app.env, "nbsphinx_thumbnails"):
             app.env.nbsphinx_thumbnails = {}
@@ -70,47 +54,7 @@ def builder_inited(app: Sphinx):
     def do_nothing(*node):
         pass
 
-    app.add_source_parser(NotebookParser)
-    # app.add_config_value('nbsphinx_execute', 'auto', rebuild='env')
-    app.add_config_value("nbsphinx_kernel_name", "", rebuild="env")
-    app.add_config_value("nbsphinx_execute_arguments", [], rebuild="env")
-    app.add_config_value("nbsphinx_allow_errors", False, rebuild="")
-    app.add_config_value("nbsphinx_timeout", None, rebuild="")
-    app.add_config_value("nbsphinx_codecell_lexer", "none", rebuild="env")
-    app.add_config_value("nbsphinx_prompt_width", "4.5ex", rebuild="html")
-    app.add_config_value("nbsphinx_responsive_width", "540px", rebuild="html")
-    app.add_config_value("nbsphinx_prolog", None, rebuild="env")
-    app.add_config_value("nbsphinx_epilog", None, rebuild="env")
-    app.add_config_value("nbsphinx_input_prompt", "[%s]:", rebuild="env")
-    app.add_config_value("nbsphinx_output_prompt", "[%s]:", rebuild="env")
-    app.add_config_value("nbsphinx_custom_formats", {}, rebuild="env")
-    # Default value is set in config_inited():
-    app.add_config_value("nbsphinx_requirejs_path", None, rebuild="html")
-    # Default value is set in config_inited():
-    app.add_config_value("nbsphinx_requirejs_options", None, rebuild="html")
-    # This will be updated in env_updated():
-    app.add_config_value("nbsphinx_widgets_path", None, rebuild="html")
-    app.add_config_value("nbsphinx_widgets_options", {}, rebuild="html")
-    # app.add_config_value('nbsphinx_thumbnails', {}, rebuild='html')
-    app.add_config_value("nbsphinx_assume_equations", True, rebuild="env")
-
-    app.add_directive("nbinput", NbInput)
-    app.add_directive("nboutput", NbOutput)
-    app.add_directive("nbinfo", NbInfo)
-    app.add_directive("nbwarning", NbWarning)
-    app.add_directive("nbgallery", NbGallery)
-    app.add_node(
-        CodeAreaNode,
-        html=(do_nothing, depart_codearea_html),
-        latex=(visit_codearea_latex, depart_codearea_latex),
-        text=(do_nothing, do_nothing),
-    )
-    app.connect("builder-inited", builder_inited)
-    app.connect("doctree-resolved", doctree_resolved)
-    app.add_post_transform(GetSizeFromImages)
-
-    app.add_config_value("nbsphinx_execute", "auto", rebuild="env")
-    app.add_config_value("nbsphinx_thumbnails", nbsphinx_thumbnails, rebuild="html")
+    app.add_config_value("nbsphinx_thumbnails", {}, rebuild="html")
     app.add_directive("nbgallery", NbGallery)
     app.add_node(
         GalleryNode,
@@ -124,11 +68,8 @@ def do_nothing(*node):
     # Monkey-patch Sphinx TocTree adapter
     toctree.TocTree.resolve = patched_toctree_resolve
 
-
 def setup(app: Sphinx):
     hack_nbsphinx(app)
-
-
 # -- Options for HTML output -------------------------------------------------
 
 # The theme to use for HTML and HTML Help pages.  See the documentation for
diff --git a/examples/index.md b/examples/index.md
index 05a56b431..aca00ad6e 100644
--- a/examples/index.md
+++ b/examples/index.md
@@ -71,6 +71,18 @@ The left sidebar shows all tags at all times. Like categories, they can be click
 on to reach the page listing all notebooks that contain the tag. If a notebook
 has tags in its metadata they are listed on the right sidebar after the {fas}`tags` icon.
 
+## NB Gallery
+
+```nbgallery
+---
+caption: This is a thumbnail gallery:
+name: rst-gallery
+glob:
+reversed:
+---
+case_studies/*
+```
+
 :::{toctree}
 :maxdepth: 1
 :hidden:

From 324877c356d9577b26cb0a535343fc4c9c1c7ecc Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Thu, 26 May 2022 20:08:01 +0200
Subject: [PATCH 03/10] Run black.

---
 examples/conf.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/examples/conf.py b/examples/conf.py
index f6ba1397b..fc3bb87b8 100644
--- a/examples/conf.py
+++ b/examples/conf.py
@@ -1,5 +1,6 @@
 import os
 from sphinx.application import Sphinx
+
 # -- Project information -----------------------------------------------------
 project = "PyMC"
 copyright = "2022, PyMC Community"
@@ -21,7 +22,7 @@
     "sphinxcontrib.bibtex",
     "sphinx_codeautolink",
     "notfound.extension",
-    'sphinx_gallery.load_style',
+    "sphinx_gallery.load_style",
 ]
 
 # List of patterns, relative to source directory, that match files and
@@ -37,6 +38,7 @@
     "page_footer.md",
 ]
 
+
 def hack_nbsphinx(app: Sphinx) -> None:
     from nbsphinx import (
         depart_gallery_html,
@@ -68,8 +70,11 @@ def do_nothing(*node):
     # Monkey-patch Sphinx TocTree adapter
     toctree.TocTree.resolve = patched_toctree_resolve
 
+
 def setup(app: Sphinx):
     hack_nbsphinx(app)
+
+
 # -- Options for HTML output -------------------------------------------------
 
 # The theme to use for HTML and HTML Help pages.  See the documentation for

From fab5ebb26cdc852ec21f9d4c157c0da588261983 Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Mon, 30 May 2022 15:19:23 +0200
Subject: [PATCH 04/10] Add all dirs to gallery.

---
 examples/gallery.rst | 24 +++++++++++++++++++-----
 examples/index.md    | 12 ------------
 2 files changed, 19 insertions(+), 17 deletions(-)

diff --git a/examples/gallery.rst b/examples/gallery.rst
index d72ff950d..2623d6f8a 100644
--- a/examples/gallery.rst
+++ b/examples/gallery.rst
@@ -1,7 +1,20 @@
 Gallery
 =======
 
-Case studies
+
+(Generalized) Linear and Hierarchical Linear Models
+-------------------------
+
+.. nbgallery::
+    :caption: This is a thumbnail gallery:
+    :name: generalized_linear_models
+    :glob:
+    :reversed:
+
+    generalized_linear_models/*
+
+
+Case Studies
 ------------
 
 .. nbgallery::
@@ -12,16 +25,17 @@ Case studies
 
     case_studies/*
 
-Generalized Linear Models
--------------------------
+Diagnostics and Model Criticism
+-------------------------------
 
 .. nbgallery::
     :caption: This is a thumbnail gallery:
-    :name: generalized_linear_models
+    :name: diagnostics_and_criticism
     :glob:
     :reversed:
 
-    generalized_linear_models/*
+    diagnostics_and_criticism/*
+
 
 Gaussian Processes
 ------------------
diff --git a/examples/index.md b/examples/index.md
index aca00ad6e..05a56b431 100644
--- a/examples/index.md
+++ b/examples/index.md
@@ -71,18 +71,6 @@ The left sidebar shows all tags at all times. Like categories, they can be click
 on to reach the page listing all notebooks that contain the tag. If a notebook
 has tags in its metadata they are listed on the right sidebar after the {fas}`tags` icon.
 
-## NB Gallery
-
-```nbgallery
----
-caption: This is a thumbnail gallery:
-name: rst-gallery
-glob:
-reversed:
----
-case_studies/*
-```
-
 :::{toctree}
 :maxdepth: 1
 :hidden:

From c5c7fecbfe7d7981eb174d3bb3cc8943fd3b64f4 Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Tue, 31 May 2022 16:59:58 +0200
Subject: [PATCH 05/10] WIP: Trying to get thumbnails to work.

---
 examples/conf.py | 61 ++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 59 insertions(+), 2 deletions(-)

diff --git a/examples/conf.py b/examples/conf.py
index fc3bb87b8..95db275c5 100644
--- a/examples/conf.py
+++ b/examples/conf.py
@@ -1,4 +1,4 @@
-import os
+import os, sys
 from sphinx.application import Sphinx
 
 # -- Project information -----------------------------------------------------
@@ -8,6 +8,8 @@
 
 # -- General configuration ---------------------------------------------------
 
+sys.path.insert(0, os.path.abspath("../sphinxext"))
+
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
@@ -23,6 +25,7 @@
     "sphinx_codeautolink",
     "notfound.extension",
     "sphinx_gallery.load_style",
+    "gallery_generator",
 ]
 
 # List of patterns, relative to source directory, that match files and
@@ -46,9 +49,23 @@ def hack_nbsphinx(app: Sphinx) -> None:
         GalleryNode,
         NbGallery,
         patched_toctree_resolve,
+        NotebookParser,
+        NbInput,
+        NbOutput,
+        NbInfo,
+        NbWarning,
+        CodeAreaNode,
+        depart_codearea_html,
+        visit_codearea_latex,
+        depart_codearea_latex,
+        GetSizeFromImages,
     )
     from sphinx.environment.adapters import toctree
 
+    nbsphinx_thumbnails = {
+        "case_studies/stochastic_volatility": "_static/stochastic_volatility.png",
+    }
+
     def builder_inited(app: Sphinx):
         if not hasattr(app.env, "nbsphinx_thumbnails"):
             app.env.nbsphinx_thumbnails = {}
@@ -56,7 +73,47 @@ def builder_inited(app: Sphinx):
     def do_nothing(*node):
         pass
 
-    app.add_config_value("nbsphinx_thumbnails", {}, rebuild="html")
+    app.add_source_parser(NotebookParser)
+    # app.add_config_value('nbsphinx_execute', 'auto', rebuild='env')
+    app.add_config_value("nbsphinx_kernel_name", "", rebuild="env")
+    app.add_config_value("nbsphinx_execute_arguments", [], rebuild="env")
+    app.add_config_value("nbsphinx_allow_errors", False, rebuild="")
+    app.add_config_value("nbsphinx_timeout", None, rebuild="")
+    app.add_config_value("nbsphinx_codecell_lexer", "none", rebuild="env")
+    app.add_config_value("nbsphinx_prompt_width", "4.5ex", rebuild="html")
+    app.add_config_value("nbsphinx_responsive_width", "540px", rebuild="html")
+    app.add_config_value("nbsphinx_prolog", None, rebuild="env")
+    app.add_config_value("nbsphinx_epilog", None, rebuild="env")
+    app.add_config_value("nbsphinx_input_prompt", "[%s]:", rebuild="env")
+    app.add_config_value("nbsphinx_output_prompt", "[%s]:", rebuild="env")
+    app.add_config_value("nbsphinx_custom_formats", {}, rebuild="env")
+    # Default value is set in config_inited():
+    app.add_config_value("nbsphinx_requirejs_path", None, rebuild="html")
+    # Default value is set in config_inited():
+    app.add_config_value("nbsphinx_requirejs_options", None, rebuild="html")
+    # This will be updated in env_updated():
+    app.add_config_value("nbsphinx_widgets_path", None, rebuild="html")
+    app.add_config_value("nbsphinx_widgets_options", {}, rebuild="html")
+    # app.add_config_value('nbsphinx_thumbnails', {}, rebuild='html')
+    app.add_config_value("nbsphinx_assume_equations", True, rebuild="env")
+
+    app.add_directive("nbinput", NbInput)
+    app.add_directive("nboutput", NbOutput)
+    app.add_directive("nbinfo", NbInfo)
+    app.add_directive("nbwarning", NbWarning)
+    app.add_directive("nbgallery", NbGallery)
+    app.add_node(
+        CodeAreaNode,
+        html=(do_nothing, depart_codearea_html),
+        latex=(visit_codearea_latex, depart_codearea_latex),
+        text=(do_nothing, do_nothing),
+    )
+    app.connect("builder-inited", builder_inited)
+    app.connect("doctree-resolved", doctree_resolved)
+    app.add_post_transform(GetSizeFromImages)
+
+    app.add_config_value("nbsphinx_execute", "auto", rebuild="env")
+    app.add_config_value("nbsphinx_thumbnails", nbsphinx_thumbnails, rebuild="html")
     app.add_directive("nbgallery", NbGallery)
     app.add_node(
         GalleryNode,

From 82c2c8cf64cbce262ee745c1af7a2fe645e7854e Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Tue, 31 May 2022 17:07:15 +0200
Subject: [PATCH 06/10] Remove sphinxext stuff

---
 examples/conf.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/examples/conf.py b/examples/conf.py
index 95db275c5..d056dd706 100644
--- a/examples/conf.py
+++ b/examples/conf.py
@@ -1,4 +1,4 @@
-import os, sys
+import os
 from sphinx.application import Sphinx
 
 # -- Project information -----------------------------------------------------
@@ -8,8 +8,6 @@
 
 # -- General configuration ---------------------------------------------------
 
-sys.path.insert(0, os.path.abspath("../sphinxext"))
-
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
@@ -25,7 +23,6 @@
     "sphinx_codeautolink",
     "notfound.extension",
     "sphinx_gallery.load_style",
-    "gallery_generator",
 ]
 
 # List of patterns, relative to source directory, that match files and

From c2729f0a805267d1db9d50bc60715318f2a043f1 Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Tue, 31 May 2022 18:18:41 +0200
Subject: [PATCH 07/10] Remove print and debug stuff.

---
 examples/conf.py               |  70 +++++----------------
 sphinxext/gallery_generator.py | 107 +++++++++++++++++++++++++++++++++
 2 files changed, 122 insertions(+), 55 deletions(-)
 create mode 100644 sphinxext/gallery_generator.py

diff --git a/examples/conf.py b/examples/conf.py
index d056dd706..ac57e9a4f 100644
--- a/examples/conf.py
+++ b/examples/conf.py
@@ -1,4 +1,4 @@
-import os
+import os, sys
 from sphinx.application import Sphinx
 
 # -- Project information -----------------------------------------------------
@@ -8,6 +8,8 @@
 
 # -- General configuration ---------------------------------------------------
 
+sys.path.insert(0, os.path.abspath("../sphinxext"))
+
 # Add any Sphinx extension module names here, as strings. They can be
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
@@ -23,6 +25,7 @@
     "sphinx_codeautolink",
     "notfound.extension",
     "sphinx_gallery.load_style",
+    "gallery_generator",
 ]
 
 # List of patterns, relative to source directory, that match files and
@@ -46,22 +49,19 @@ def hack_nbsphinx(app: Sphinx) -> None:
         GalleryNode,
         NbGallery,
         patched_toctree_resolve,
-        NotebookParser,
-        NbInput,
-        NbOutput,
-        NbInfo,
-        NbWarning,
-        CodeAreaNode,
-        depart_codearea_html,
-        visit_codearea_latex,
-        depart_codearea_latex,
-        GetSizeFromImages,
     )
     from sphinx.environment.adapters import toctree
 
-    nbsphinx_thumbnails = {
-        "case_studies/stochastic_volatility": "_static/stochastic_volatility.png",
-    }
+    from glob import glob
+
+    nb_paths = glob("examples/*/*.ipynb")
+    nbsphinx_thumbnails = {}
+    for nb_path in nb_paths:
+        png_file = os.path.join("_static", os.path.splitext(os.path.split(nb_path)[-1])[0] + ".png")
+        nb_path_rel = os.path.splitext(
+            os.path.join(*os.path.normpath(nb_path).split(os.path.sep)[1:])
+        )[0]
+        nbsphinx_thumbnails[nb_path_rel] = png_file
 
     def builder_inited(app: Sphinx):
         if not hasattr(app.env, "nbsphinx_thumbnails"):
@@ -70,46 +70,6 @@ def builder_inited(app: Sphinx):
     def do_nothing(*node):
         pass
 
-    app.add_source_parser(NotebookParser)
-    # app.add_config_value('nbsphinx_execute', 'auto', rebuild='env')
-    app.add_config_value("nbsphinx_kernel_name", "", rebuild="env")
-    app.add_config_value("nbsphinx_execute_arguments", [], rebuild="env")
-    app.add_config_value("nbsphinx_allow_errors", False, rebuild="")
-    app.add_config_value("nbsphinx_timeout", None, rebuild="")
-    app.add_config_value("nbsphinx_codecell_lexer", "none", rebuild="env")
-    app.add_config_value("nbsphinx_prompt_width", "4.5ex", rebuild="html")
-    app.add_config_value("nbsphinx_responsive_width", "540px", rebuild="html")
-    app.add_config_value("nbsphinx_prolog", None, rebuild="env")
-    app.add_config_value("nbsphinx_epilog", None, rebuild="env")
-    app.add_config_value("nbsphinx_input_prompt", "[%s]:", rebuild="env")
-    app.add_config_value("nbsphinx_output_prompt", "[%s]:", rebuild="env")
-    app.add_config_value("nbsphinx_custom_formats", {}, rebuild="env")
-    # Default value is set in config_inited():
-    app.add_config_value("nbsphinx_requirejs_path", None, rebuild="html")
-    # Default value is set in config_inited():
-    app.add_config_value("nbsphinx_requirejs_options", None, rebuild="html")
-    # This will be updated in env_updated():
-    app.add_config_value("nbsphinx_widgets_path", None, rebuild="html")
-    app.add_config_value("nbsphinx_widgets_options", {}, rebuild="html")
-    # app.add_config_value('nbsphinx_thumbnails', {}, rebuild='html')
-    app.add_config_value("nbsphinx_assume_equations", True, rebuild="env")
-
-    app.add_directive("nbinput", NbInput)
-    app.add_directive("nboutput", NbOutput)
-    app.add_directive("nbinfo", NbInfo)
-    app.add_directive("nbwarning", NbWarning)
-    app.add_directive("nbgallery", NbGallery)
-    app.add_node(
-        CodeAreaNode,
-        html=(do_nothing, depart_codearea_html),
-        latex=(visit_codearea_latex, depart_codearea_latex),
-        text=(do_nothing, do_nothing),
-    )
-    app.connect("builder-inited", builder_inited)
-    app.connect("doctree-resolved", doctree_resolved)
-    app.add_post_transform(GetSizeFromImages)
-
-    app.add_config_value("nbsphinx_execute", "auto", rebuild="env")
     app.add_config_value("nbsphinx_thumbnails", nbsphinx_thumbnails, rebuild="html")
     app.add_directive("nbgallery", NbGallery)
     app.add_node(
@@ -185,7 +145,7 @@ def setup(app: Sphinx):
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,
 # so a file named "default.css" will overwrite the builtin "default.css".
-html_static_path = ["../_static", "../thumbnails"]
+html_static_path = ["../_static", "../_images", "../_templates"]
 html_css_files = ["custom.css"]
 templates_path = ["../_templates"]
 html_sidebars = {
diff --git a/sphinxext/gallery_generator.py b/sphinxext/gallery_generator.py
new file mode 100644
index 000000000..34d4445a6
--- /dev/null
+++ b/sphinxext/gallery_generator.py
@@ -0,0 +1,107 @@
+"""
+Sphinx plugin to run generate a gallery for notebooks
+
+Modified from the seaborn project, which modified the mpld3 project.
+"""
+import base64
+import json
+import os
+import runpy
+import shutil
+
+from pathlib import Path
+
+import matplotlib
+
+matplotlib.use("Agg")
+import matplotlib.pyplot as plt
+
+from matplotlib import image
+
+DOC_SRC = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+DEFAULT_IMG_LOC = os.path.join(os.path.dirname(DOC_SRC), "pymc-examples/_static", "PyMC.png")
+
+
+def create_thumbnail(infile, width=275, height=275, cx=0.5, cy=0.5, border=4):
+    """Overwrites `infile` with a new file of the given size"""
+    im = image.imread(infile)
+    rows, cols = im.shape[:2]
+    size = min(rows, cols)
+    if size == cols:
+        xslice = slice(0, size)
+        ymin = min(max(0, int(cx * rows - size // 2)), rows - size)
+        yslice = slice(ymin, ymin + size)
+    else:
+        yslice = slice(0, size)
+        xmin = min(max(0, int(cx * cols - size // 2)), cols - size)
+        xslice = slice(xmin, xmin + size)
+    thumb = im[yslice, xslice]
+    thumb[:border, :, :3] = thumb[-border:, :, :3] = 0
+    thumb[:, :border, :3] = thumb[:, -border:, :3] = 0
+
+    dpi = 100
+    fig = plt.figure(figsize=(width / dpi, height / dpi), dpi=dpi)
+
+    ax = fig.add_axes([0, 0, 1, 1], aspect="auto", frameon=False, xticks=[], yticks=[])
+    ax.imshow(thumb, aspect="auto", resample=True, interpolation="bilinear")
+    fig.savefig(infile, dpi=dpi)
+    plt.close(fig)
+    return fig
+
+
+class NotebookGenerator:
+    """Tools for generating an example page from a file"""
+
+    def __init__(self, filename, target_dir):
+        self.basename = os.path.basename(filename)
+        stripped_name = os.path.splitext(self.basename)[0]
+        self.image_dir = os.path.join(target_dir, "_static")
+        self.png_path = os.path.join(self.image_dir, f"{stripped_name}.png")
+        with open(filename) as fid:
+            self.json_source = json.load(fid)
+        self.default_image_loc = DEFAULT_IMG_LOC
+
+    def extract_preview_pic(self):
+        """By default, just uses the last image in the notebook."""
+        pic = None
+        for cell in self.json_source["cells"]:
+            for output in cell.get("outputs", []):
+                if "image/png" in output.get("data", []):
+                    pic = output["data"]["image/png"]
+        if pic is not None:
+            return base64.b64decode(pic)
+        return None
+
+    def find_last_pic_cell(self):
+        """By default, just uses the last image in the notebook."""
+        pic = None
+        for cell in self.json_source["cells"]:
+            for output in cell.get("outputs", []):
+                if "image/png" in output.get("data", []):
+                    cell["metadata"]["tags"] = ["nbsphinx-thumbnail"]
+                    pic = cell
+        return pic
+
+    def gen_previews(self):
+        preview = self.extract_preview_pic()
+        if preview is not None:
+            with open(self.png_path, "wb") as buff:
+                buff.write(preview)
+        else:
+            print(f"didn't find for {self.png_path}")
+            shutil.copy(self.default_image_loc, self.png_path)
+        create_thumbnail(self.png_path)
+
+
+def main(app):
+    from glob import glob
+
+    nb_paths = glob("examples/*/*.ipynb")
+
+    for nb_path in nb_paths:
+        nbg = NotebookGenerator(nb_path, "")
+        nbg.gen_previews()
+
+
+def setup(app):
+    app.connect("builder-inited", main)

From 60924aa2ba7ec2ef4f53a9d38836e77d6a307012 Mon Sep 17 00:00:00 2001
From: Thomas Wiecki <thomas.wiecki@gmail.com>
Date: Tue, 31 May 2022 18:47:00 +0200
Subject: [PATCH 08/10] Remove unused method.

---
 sphinxext/gallery_generator.py | 10 ----------
 1 file changed, 10 deletions(-)

diff --git a/sphinxext/gallery_generator.py b/sphinxext/gallery_generator.py
index 34d4445a6..360046be1 100644
--- a/sphinxext/gallery_generator.py
+++ b/sphinxext/gallery_generator.py
@@ -72,16 +72,6 @@ def extract_preview_pic(self):
             return base64.b64decode(pic)
         return None
 
-    def find_last_pic_cell(self):
-        """By default, just uses the last image in the notebook."""
-        pic = None
-        for cell in self.json_source["cells"]:
-            for output in cell.get("outputs", []):
-                if "image/png" in output.get("data", []):
-                    cell["metadata"]["tags"] = ["nbsphinx-thumbnail"]
-                    pic = cell
-        return pic
-
     def gen_previews(self):
         preview = self.extract_preview_pic()
         if preview is not None:

From 695da62d837215001e96153525ffbbed3997bc54 Mon Sep 17 00:00:00 2001
From: "Oriol (ZBook)" <oriol.abril.pla@gmail.com>
Date: Tue, 31 May 2022 20:04:41 +0300
Subject: [PATCH 09/10] fix readthedocs config

---
 .readthedocs.yml | 2 --
 1 file changed, 2 deletions(-)

diff --git a/.readthedocs.yml b/.readthedocs.yml
index bf5f09be6..8f89cb494 100644
--- a/.readthedocs.yml
+++ b/.readthedocs.yml
@@ -7,5 +7,3 @@ python:
    version: "3.8"
    install:
    - requirements: requirements-docs.txt
-   - method: pip
-     path: .

From d5e25ca283ab015c8eded9f82c3df5f68de84314 Mon Sep 17 00:00:00 2001
From: "Oriol (ZBook)" <oriol.abril.pla@gmail.com>
Date: Tue, 31 May 2022 20:12:12 +0300
Subject: [PATCH 10/10] sort inputs and requirements

---
 requirements-docs.txt          | 1 +
 sphinxext/gallery_generator.py | 3 ---
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/requirements-docs.txt b/requirements-docs.txt
index 0f9d0b660..35e5b1cc1 100644
--- a/requirements-docs.txt
+++ b/requirements-docs.txt
@@ -10,3 +10,4 @@ sphinx-codeautolink
 sphinx-notfound-page
 nbsphinx
 sphinx-gallery
+matplotlib
diff --git a/sphinxext/gallery_generator.py b/sphinxext/gallery_generator.py
index 360046be1..3e45c2425 100644
--- a/sphinxext/gallery_generator.py
+++ b/sphinxext/gallery_generator.py
@@ -6,11 +6,8 @@
 import base64
 import json
 import os
-import runpy
 import shutil
 
-from pathlib import Path
-
 import matplotlib
 
 matplotlib.use("Agg")