checking if the image path is inside a subdir of dir

Author: AakashGfude

jupyter_sphinx/ast.py

@@ -274,6 +274,14 @@ def cell_output_to_nodes(cell, data_priority, write_stderr, dir, thebe_config):
                 # directory, so make a relative path, which Sphinx treats
                 # as being relative to the current working directory.
                 filename = os.path.basename(output.metadata["filenames"][mime_type])
+
+                # checks if file dir path is inside a subpath of dir
+                filedir = os.path.dirname(output.metadata["filenames"][mime_type])
+                subpaths = filedir.split(dir)
+                if subpaths and len(subpaths) > 1:
+                    subpath = subpaths[1]
+                    dir += subpath
+
                 uri = os.path.join(dir, filename)
                 to_add.append(docutils.nodes.image(uri=uri))
             elif mime_type == "text/html":