interactive-pipe 0.7.0__tar.gz → 0.7.1__tar.gz

{interactive_pipe-0.7.0/src/interactive_pipe.egg-info → interactive_pipe-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: interactive_pipe
-Version: 0.7.0
+Version: 0.7.1
 Summary: Library to create flexible interactive image processing pipelines and automatically add a graphical user interface without knowing anything about GUI coding!
 Author-email: Balthazar Neveu <balthazarneveu@gmail.com>
 Project-URL: Homepage, https://github.com/balthazarneveu/interactive_pipe
@@ -95,7 +95,7 @@ pip install -e ".[full]"
 
 ## :scroll:  Features
 
-**Version 0.7.0**
+**Version 0.7.1**
 - Modular multi-image processing filters
 - Declarative: Easily make graphical user interface without having to learn anything about pyQt or matplotlib
 - Support in jupyter notebooks

{interactive_pipe-0.7.0 → interactive_pipe-0.7.1}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "interactive_pipe"
-version = "0.7.0"
+version = "0.7.1"
 authors = [
   { name="Balthazar Neveu", email="balthazarneveu@gmail.com" },
 ]

{interactive_pipe-0.7.0 → interactive_pipe-0.7.1}/readme.md

@@ -51,7 +51,7 @@ pip install -e ".[full]"
 
 ## :scroll:  Features
 
-**Version 0.7.0**
+**Version 0.7.1**
 - Modular multi-image processing filters
 - Declarative: Easily make graphical user interface without having to learn anything about pyQt or matplotlib
 - Support in jupyter notebooks

interactive_pipe-0.7.1/src/interactive_pipe/data_objects/audio.py

@@ -0,0 +1,32 @@
+from pathlib import Path
+import numpy as np
+from io import BytesIO
+import base64
+import logging
+from typing import Union, Tuple
+WAVIO_AVAILABLE = False
+try:
+    WAVIO_AVAILABLE = True
+    import wavio
+except ImportError:
+    logging.info("Cannot import wavio")
+
+
+def audio_to_html(audio: Union[None, str, Path, Tuple[int, np.ndarray]]) -> str:
+    if audio is None:
+        logging.debug("No audio to display")
+        return ""
+    if isinstance(audio, str) or isinstance(audio, Path):
+        audio_base64 = base64.b64encode(open(audio, "rb").read()).decode("utf-8")
+    elif isinstance(audio, tuple):
+        assert WAVIO_AVAILABLE, "wavio is not available"
+        assert len(audio) == 2, "audio tuple should have 2 elements: (rate, data)"
+        assert isinstance(audio[0], int), "audio[0] should be an integer"
+        assert isinstance(audio[1], np.ndarray), "audio[1] should be a numpy array"
+        audio_bytes = BytesIO()
+        wavio.write(audio_bytes, audio[1].astype(np.float32), audio[0], sampwidth=4)
+        audio_bytes.seek(0)
+        audio_base64 = base64.b64encode(audio_bytes.read()).decode("utf-8")
+    audio_player = f'<audio src="data:audio/mpeg;base64,{audio_base64}" controls autoplay></audio>'
+
+    return audio_player

{interactive_pipe-0.7.0 → interactive_pipe-0.7.1}/src/interactive_pipe/graphical/gradio_control.py

@@ -49,7 +49,12 @@ class IntSliderControl(BaseControl):
 
     def create(self) -> gr.Slider:
         self.control_widget = gr.Slider(
-            value=self.ctrl.value_default, minimum=self.ctrl.value_range[0], maximum=self.ctrl.value_range[1], label=self.name, step=1)
+            value=self.ctrl.value_default,
+            minimum=self.ctrl.value_range[0],
+            maximum=self.ctrl.value_range[1],
+            label=self.name,
+            step=1
+        )
         return self.control_widget
 
 
@@ -60,7 +65,10 @@ class FloatSliderControl(BaseControl):
     def create(self) -> gr.Slider:
         self.control_widget = gr.Slider(
             value=self.ctrl.value_default,
-            minimum=self.ctrl.value_range[0], maximum=self.ctrl.value_range[1], label=self.name)
+            minimum=self.ctrl.value_range[0],
+            maximum=self.ctrl.value_range[1],
+            label=self.name,
+        )
 
         return self.control_widget
 

{interactive_pipe-0.7.0 → interactive_pipe-0.7.1}/src/interactive_pipe/graphical/gradio_gui.py

@@ -1,11 +1,13 @@
 import gradio as gr
 import numpy as np
+import math
 from typing import List
 from interactive_pipe.headless.pipeline import HeadlessPipeline
 from interactive_pipe.graphical.gradio_control import ControlFactory
 from interactive_pipe.graphical.window import InteractivePipeWindow
 from interactive_pipe.graphical.gui import InteractivePipeGUI
 from interactive_pipe.headless.control import Control
+from interactive_pipe.data_objects.audio import audio_to_html
 import logging
 PYQTVERSION = None
 MPL_SUPPORT = False
@@ -21,23 +23,48 @@ except ImportError:
 class InteractivePipeGradio(InteractivePipeGUI):
     def init_app(self, **kwargs):
         self.window = MainWindow(controls=self.controls, name=self.name,
-                                 pipeline=self.pipeline, size=self.size, main_gui=self, **kwargs)
+                                 pipeline=self.pipeline, size=self.size, main_gui=self, audio=self.audio, **kwargs)
         self.pipeline.global_params["__pipeline"] = self.pipeline
-        # self.set_default_key_bindings()
+        if self.audio:
+            self.pipeline.global_params["__set_audio"] = self.__set_audio
+            self.pipeline.global_params["__play"] = self.__play
+            self.pipeline.global_params["__stop"] = self.__stop
+            self.pipeline.global_params["__pause"] = self.__pause
+            self.__set_audio(None)
+
+    def __set_audio(self, audio_content):
+        self.window.audio_content = audio_content
+
+    def __play(self):
+        pass
+
+    def __pause(self):
+        self.__set_audio(None)
+
+    def __stop(self):
+        self.__set_audio(None)
 
     def run(self) -> list:
         assert self.pipeline._PipelineCore__initialized_inputs, "Did you forget to initialize the pipeline inputs?"
         out_list = self.window.process_inputs_fn(*self.window.default_values)
+        self.window.refresh_display(out_list)
         out_list_gradio_containers = []
         for idx in range(len(out_list)):
             for idy in range(len(out_list[idx])):
+                title = self.window.image_canvas[idx][idy].get("title", f"{idx} {idy}")
+                title = title.replace("_", " ")
                 if isinstance(out_list[idx][idy], np.ndarray):
-                    out_list_gradio_containers.append(gr.Image())
+                    out_list_gradio_containers.append(gr.Image(label=title))
                 elif MPL_SUPPORT and isinstance(out_list[idx][idy], Curve):
-                    out_list_gradio_containers.append(gr.Plot())
+                    # if out_list[idx][idy].title is not None:
+                    #     title = out_list[idx][idy].title
+                    out_list_gradio_containers.append(gr.Plot(label=title))
                 else:
                     raise NotImplementedError(
                         f"output type {type(out_list[idx][idy])} not supported")
+        if self.window.audio:
+            self.window.audio = gr.HTML()
+            self.pipeline.global_params["__audio"] = self.window.audio
         self.window.instantiate_gradio_interface(out_list_gradio_containers)
         self.window.refresh()
         self.custom_end()
@@ -48,14 +75,18 @@ class InteractivePipeGradio(InteractivePipeGUI):
 
 
 class MainWindow(InteractivePipeWindow):
-    def __init__(self, *args, controls=[], name="", pipeline: HeadlessPipeline = None, size=None, share_gradio_app=False, **kwargs):
+    def __init__(self, *args, controls=[], name="", pipeline: HeadlessPipeline = None, size=None, share_gradio_app=False, markdown_description=None, sliders_layout=None, sliders_per_row_layout=None, audio=False, **kwargs):
         InteractivePipeWindow.__init__(
             self, name=name, pipeline=pipeline, size=size)
+        self.markdown_description = markdown_description
+        self.sliders_layout = sliders_layout
+        self.sliders_per_row_layout = sliders_per_row_layout
         self.init_sliders(controls)
         self.size = size
         self.full_screen_flag = False
         self.pipeline = pipeline
         self.share_gradio_app = share_gradio_app
+        self.audio = audio
         # Define the functions that will be called when the input changes for gradio. => gr.Interface(fn=process_fn)
 
         def process_outputs_fn(out) -> tuple:
@@ -92,6 +123,9 @@ class MainWindow(InteractivePipeWindow):
         def run_fn(*args) -> tuple:
             out = process_inputs_fn(*args)
             out_tuple = process_outputs_fn(out)
+            if self.audio:
+                html_audio = audio_to_html(self.audio_content)
+                return *out_tuple, html_audio
             return out_tuple
         self.default_values = [self.ctrl[ctrl_key].value for ctrl_key in self.ctrl.keys()]
         self.process_inputs_fn = process_inputs_fn
@@ -116,15 +150,68 @@ class MainWindow(InteractivePipeWindow):
             # Gradio Blocks mode
             # https://www.gradio.app/guides/blocks-and-event-listeners
             with gr.Blocks() as io:
+                if self.audio:
+                    outputs.append(self.audio)
+                with gr.Row(variant="compact"):
+                    gr.Markdown("### " + self.name.replace("_", " "))
                 with gr.Row():
                     for elem in outputs:
                         elem.render()
-                with gr.Row():
-                    for elem in self.widget_list:
-                        with gr.Row():
-                            elem.render()
-                with gr.Row():
-                    gr.Examples([self.default_values], inputs=self.widget_list)
+
+                if self.sliders_layout is None:
+                    self.sliders_layout = "collapsible"
+                if self.sliders_per_row_layout is None:
+                    self.sliders_per_row_layout = 1
+                assert self.sliders_layout in ["compact", "vertical", "collapsible", "smart"]
+                ctrl_dict_by_type = {"all": list(range(len(self.ctrl)))}
+                categories = ["all"]
+                if self.sliders_layout == "compact":
+                    selected_mode = gr.Row()
+                elif self.sliders_layout == "vertical":
+                    # Use Column to stack elements vertically
+                    selected_mode = gr.Column()
+                elif self.sliders_layout == "collapsible":
+                    selected_mode = gr.Accordion("Parameters", open=True)
+                elif self.sliders_layout == "smart":
+                    # Group sliders by type
+                    ctrl_dict_by_type = {}
+                    for ctrl_index, ctrl_key in enumerate(self.ctrl.keys()):
+                        ctrl_type = self.ctrl[ctrl_key]._type
+                        ctrl_type = str(ctrl_type).split("<class '")[1].replace("'>", "")
+                        if ctrl_type == "int":
+                            ctrl_type = "float"
+                        if ctrl_type not in ctrl_dict_by_type:
+                            ctrl_dict_by_type[ctrl_type] = []
+                        ctrl_dict_by_type[ctrl_type].append(ctrl_index)
+                    categories = ["str", "bool", "float"]
+                    selected_mode = gr.Column()
+                else:
+                    raise NotImplementedError(f"Sliders layout {self.sliders_layout} not supported")
+                for ctrl_type in categories:
+                    ctrl_indices = ctrl_dict_by_type.get(ctrl_type, [])
+                    if len(ctrl_indices) == 0:
+                        continue
+                    if self.sliders_per_row_layout == 1:
+                        with selected_mode:
+                            for idx in ctrl_indices:
+                                elem = self.widget_list[idx]
+                                elem.render()
+                    else:
+                        with selected_mode:
+                            for split_num in range(math.ceil(len(ctrl_indices)/self.sliders_per_row_layout)):
+                                with gr.Row():
+                                    start = split_num*self.sliders_per_row_layout
+                                    end = min((split_num+1)*self.sliders_per_row_layout, len(ctrl_indices))
+                                    for idx in ctrl_indices[start:end]:
+                                        elem = self.widget_list[idx]
+                                        elem.render()
+
+                with gr.Accordion("Reset to default values", open=False):
+                    gr.Examples([self.default_values], inputs=self.widget_list, label="Presets")
+
+                if self.markdown_description is not None:
+                    with gr.Row():
+                        gr.Markdown(self.markdown_description)
                 io.load(fn=self.run_fn, inputs=self.widget_list, outputs=outputs)
                 for idx in range(len(self.widget_list)):
                     self.widget_list[idx].change(
@@ -167,7 +254,10 @@ class MainWindow(InteractivePipeWindow):
 
     def add_image_placeholder(self, row, col):
         ax_placeholder = None
-        text_label = None
+        text_label = self.get_current_style(row, col).get("title", "")
         image_label = None
         self.image_canvas[row][col] = {
             "image": image_label, "title": text_label, "ax_placeholder": ax_placeholder}
+
+    def update_image(self, content, row, col):
+        pass

{interactive_pipe-0.7.0 → interactive_pipe-0.7.1/src/interactive_pipe.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: interactive_pipe
-Version: 0.7.0
+Version: 0.7.1
 Summary: Library to create flexible interactive image processing pipelines and automatically add a graphical user interface without knowing anything about GUI coding!
 Author-email: Balthazar Neveu <balthazarneveu@gmail.com>
 Project-URL: Homepage, https://github.com/balthazarneveu/interactive_pipe
@@ -95,7 +95,7 @@ pip install -e ".[full]"
 
 ## :scroll:  Features
 
-**Version 0.7.0**
+**Version 0.7.1**
 - Modular multi-image processing filters
 - Declarative: Easily make graphical user interface without having to learn anything about pyQt or matplotlib
 - Support in jupyter notebooks

{interactive_pipe-0.7.0 → interactive_pipe-0.7.1}/src/interactive_pipe.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ src/interactive_pipe/core/graph.py
 src/interactive_pipe/core/pipeline.py
 src/interactive_pipe/core/signature.py
 src/interactive_pipe/data_objects/__init__.py
+src/interactive_pipe/data_objects/audio.py
 src/interactive_pipe/data_objects/curves.py
 src/interactive_pipe/data_objects/data.py
 src/interactive_pipe/data_objects/image.py