Fixed the image bug

Author: rishikanthc

markopolis/app.py

@@ -3,7 +3,7 @@
 from fastapi import FastAPI, HTTPException, Depends, Header, Path, Request, Query
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-from fastapi.responses import JSONResponse, HTMLResponse
+from fastapi.responses import JSONResponse, HTMLResponse, FileResponse
 from fastapi.middleware.cors import CORSMiddleware
 import markopolis.dantic as D
 import markopolis.md as M
@@ -189,6 +189,17 @@ async def get_note_html(
 
 @app.get("/{path:path}", response_class=HTMLResponse)
 async def load_page(request: Request, path: str):
+    _cond = (
+        path.endswith(".png")
+        or path.endswith(".jpg")
+        or path.endswith(".jpeg")
+        or path.endswith(".gif")
+        or path.endswith(".svg")
+        or path.endswith(".webp")
+    )
+    if _cond:
+        img_pth = os.path.join(settings.md_path, path)
+        return FileResponse(img_pth)
     try:
         frontmatter = M.get_frontmatter(path)
         html_content = M.get_note_html(path)

markopolis/md.py

@@ -6,6 +6,7 @@
 import sh
 import markdown
 from .md_extensions import (
+    ImageExtension,
     CalloutExtension,
     MermaidExtension,
     StrikethroughExtension,
@@ -210,6 +211,7 @@ def get_note_html(note_path):
         extensions=[
             "fenced_code",
             "codehilite",
+            ImageExtension(),
             "mdx_wikilink_plus",
             # WikiLinkExtension(base_url="/", end_url=""),
             "markdown_checklist.extension",

markopolis/md_extensions.py

@@ -7,6 +7,27 @@
 import re
 
 
+class ImageExtension(markdown.extensions.Extension):
+    def extendMarkdown(self, md):
+        # Register the preprocessor
+        md.preprocessors.register(ImagePreprocessor(md), "image_preprocessor", 25)
+
+
+class ImagePreprocessor(Preprocessor):
+    # Pattern for image extensions (webp, jpg, jpeg, png, svg)
+    IMAGE_PATTERN = re.compile(
+        r"!\[\[(.*?\.(?:webp|jpg|jpeg|png|svg))\]\]"
+    )  # Custom ![[image.ext]]
+
+    def run(self, lines):
+        new_lines = []
+        for line in lines:
+            # Substitute custom image syntax ![[image.ext]] with standard Markdown image syntax ![](image.ext)
+            new_line = re.sub(self.IMAGE_PATTERN, r"![](\1)", line)
+            new_lines.append(new_line)
+        return new_lines
+
+
 # Custom inline pattern for obsidian style images
 class ObsidianImageInlineProcessor(InlineProcessor):
     def handleMatch(self, m, data):