formatting

gradio_dualvision/app_template.py

@@ -27,6 +27,14 @@ import os
 import re
 
 import gradio as gr
+from .version import __version__
+
+if __version__ != gr.__version__:
+    raise gr.Error(
+        f"gradio version ({gr.__version__}) must match gradio-dualvision version ({__version__}). "
+        f"Check the metadata of the README.md in your demo (sdk_version field)."
+    )
+
 import spaces
 from PIL import Image as PILImage
 from gradio import Component, ImageSlider
@@ -35,7 +43,6 @@ from .gradio_patches.examples import Examples
 from .gradio_patches.gallery import Gallery
 from .gradio_patches.image import Image
 from .gradio_patches.radio import Radio
-from .version import __version__
 
 
 class DualVisionApp(gr.Blocks):
@@ -82,12 +89,6 @@ class DualVisionApp(gr.Blocks):
             gallery_thumb_min_size: Min size of the gallery thumbnail (Default: `96px`).
             **kwargs: Any other arguments that Gradio Blocks class can take.
         """
-        if __version__ != gr.__version__:
-            raise gr.Error(
-                f"gradio version ({gr.__version__}) must match gradio-dualvision version ({__version__}). "
-                f"Check the metadata of the README.md in your demo (sdk_version field)."
-            )
-
         squeeze_viewport_height_pct = int(squeeze_viewport_height_pct)
         if not 50 <= squeeze_viewport_height_pct <= 100:
             raise gr.Error(
@@ -113,8 +114,9 @@ class DualVisionApp(gr.Blocks):
             self.process_components = spaces.GPU(
                 self.process_components, duration=spaces_zero_gpu_duration
             )
-        self.head = ""
-        self.head += """
+        # fmt: off
+        self.head = (
+            """
             <script>
                 let observerFooterButtons = new MutationObserver((mutationsList, observer) => {
                     const origButtonShowAPI = document.querySelector(".show-api");
@@ -158,6 +160,8 @@ class DualVisionApp(gr.Blocks):
                 observerFooterButtons.observe(document.body, { childList: true, subtree: true });
             </script>
         """
+        )
+        # fmt: on
         if kwargs.get("analytics_enabled") is not False:
             self.head += f"""
                 <script async src="https://www.googletagmanager.com/gtag/js?id=G-1FWSVCGZTG"></script>

gradio_dualvision/gradio_patches/gallery.py

@@ -29,7 +29,13 @@ from pathlib import Path
 from urllib.parse import quote, urlparse
 
 from gradio import FileData, image_utils, processing_utils, utils, wasm_utils
-from gradio.components.gallery import GalleryData, GalleryImage, GalleryMediaType, CaptionedGalleryMediaType, GalleryVideo
+from gradio.components.gallery import (
+    GalleryData,
+    GalleryImage,
+    GalleryMediaType,
+    CaptionedGalleryMediaType,
+    GalleryVideo,
+)
 from gradio_client import utils as client_utils
 from gradio_client.utils import is_http_url_like
 from gradio.data_classes import ImageData
@@ -68,7 +74,9 @@ class Gallery(gradio.Gallery):
                 )
                 file_path = str(utils.abspath(file))
             elif isinstance(img, PIL.Image.Image):
-                format = "png" if img.mode == "I;16" else self.format  # Patch 1: change format based on the inbound dtype
+                format = (
+                    "png" if img.mode == "I;16" else self.format
+                )  # Patch 1: change format based on the inbound dtype
                 file = processing_utils.save_pil_to_cache(
                     img, cache_dir=self.GRADIO_CACHE, format=format
                 )

gradio_dualvision/gradio_patches/image.py

@@ -39,7 +39,16 @@ class Image(gradio.Image):
         self,
         value,
     ) -> ImageData:
-        fn_format_selector = lambda x: "png" if (isinstance(x, np.ndarray) and x.dtype == np.uint16 and x.squeeze().ndim == 2) or (isinstance(x, PILImage.Image) and x.mode == "I;16") else self.format
+        fn_format_selector = lambda x: (
+            "png"
+            if (
+                isinstance(x, np.ndarray)
+                and x.dtype == np.uint16
+                and x.squeeze().ndim == 2
+            )
+            or (isinstance(x, PILImage.Image) and x.mode == "I;16")
+            else self.format
+        )
         format = fn_format_selector(value)
 
         out = image_utils.postprocess_image(
@@ -57,7 +66,9 @@ class Image(gradio.Image):
 
         return out
 
-    def preprocess(self, payload: ImageData) ->  str | PIL.Image.Image | np.ndarray | None:
+    def preprocess(
+        self, payload: ImageData
+    ) -> str | PIL.Image.Image | np.ndarray | None:
         out = super().preprocess(payload)
 
         if "settings" in payload.meta:
@@ -78,11 +89,11 @@ class Image(gradio.Image):
             right = left + min_side
             bottom = top + min_side
             img = img.crop((left, top, right, bottom))
-        
+
         img.thumbnail((max_dim, max_dim))
         temp_file = tempfile.NamedTemporaryFile(suffix=".webp", delete=False)
         img.save(temp_file.name, "WEBP")
-        
+
         return temp_file.name
 
     def process_example(