halib 0.1.91__py3-none-any.whl → 0.1.93__py3-none-any.whl

halib/research/plot.py

@@ -1,24 +1,28 @@
+import ast
 import os
+import json
+import time
+import click
+import base64
 import pandas as pd
+
+from PIL import Image
+from io import BytesIO
+
 import plotly.express as px
-from rich.console import Console
-from ..common import now_str, norm_str, ConsoleLog
+from ..common import now_str
 from ..filetype import csvfile
+import plotly.graph_objects as go
 from ..system import filesys as fs
-import click
-import time
 
-import pandas as pd
-import plotly.graph_objects as go
-from PIL import Image
-import base64
-from io import BytesIO
+from rich.console import Console
 from typing import Callable, Optional, Tuple, List, Union
 
 
 console = Console()
 desktop_path = os.path.expanduser("~/Desktop")
 
+
 class PlotHelper:
     def _verify_csv(self, csv_file):
         """Read a CSV and normalize column names (lowercase)."""
@@ -179,273 +183,752 @@ class PlotHelper:
                 console.log("Stopped live updates.")
         else:
             run_once()
+
     @staticmethod
-    def plot_image_grid(csv_path, sep=";", max_width=300, max_height=300):
+    def get_img_grid_df(input_dir, log=False):
+        """
+        Use images in input_dir to create a dataframe for plot_image_grid.
+
+        Directory structures supported:
+
+        A. Row/Col structure:
+            input_dir/
+                ├── row0/
+                │   ├── col0/
+                │   │   ├── 0.png
+                │   │   ├── 1.png
+                │   └── col1/
+                │       ├── 0.png
+                │       ├── 1.png
+                ├── row1/
+                │   ├── col0/
+                │   │   ├── 0.png
+                │   │   ├── 1.png
+                │   └── col1/
+                │       ├── 0.png
+                │       ├── 1.png
+
+        B. Row-only structure (no cols):
+            input_dir/
+                ├── row0/
+                │   ├── 0.png
+                │   ├── 1.png
+                ├── row1/
+                │   ├── 0.png
+                │   ├── 1.png
+
+        Returns:
+            pd.DataFrame: DataFrame suitable for plot_image_grid.
+                        Each cell contains a list of image paths.
         """
-        Plot a grid of images using Plotly from a CSV file.
+        # --- Collect row dirs ---
+        rows = sorted([r for r in fs.list_dirs(input_dir) if r.startswith("row")])
+        if not rows:
+            raise ValueError(f"No 'row*' directories found in {input_dir}")
+
+        first_row_path = os.path.join(input_dir, rows[0])
+        subdirs = fs.list_dirs(first_row_path)
+
+        if subdirs:  # --- Case A: row/col structure ---
+            cols_ref = sorted(subdirs)
+
+            # Ensure column consistency
+            meta_dict = {row: sorted(fs.list_dirs(os.path.join(input_dir, row))) for row in rows}
+            for row, cols in meta_dict.items():
+                if cols != cols_ref:
+                    raise ValueError(f"Row {row} has mismatched columns: {cols} vs {cols_ref}")
+
+            # Collect image paths
+            meta_with_paths = {
+                row: {
+                    col: fs.filter_files_by_extension(os.path.join(input_dir, row, col), ["png", "jpg", "jpeg"])
+                    for col in cols_ref
+                }
+                for row in rows
+            }
+
+            # Validate equal number of images per (row, col)
+            n_imgs = len(meta_with_paths[rows[0]][cols_ref[0]])
+            for row, cols in meta_with_paths.items():
+                for col, paths in cols.items():
+                    if len(paths) != n_imgs:
+                        raise ValueError(
+                            f"Inconsistent file counts in {row}/{col}: {len(paths)} vs expected {n_imgs}"
+                        )
+
+            # Flatten long format
+            data = {"row": [row for row in rows for _ in range(n_imgs)]}
+            for col in cols_ref:
+                data[col] = [meta_with_paths[row][col][i] for row in rows for i in range(n_imgs)]
+
+        else:  # --- Case B: row-only structure ---
+            meta_with_paths = {
+                row: fs.filter_files_by_extension(os.path.join(input_dir, row), ["png", "jpg", "jpeg"])
+                for row in rows
+            }
+
+            # Validate equal number of images per row
+            n_imgs = len(next(iter(meta_with_paths.values())))
+            for row, paths in meta_with_paths.items():
+                if len(paths) != n_imgs:
+                    raise ValueError(f"Inconsistent file counts in {row}: {len(paths)} vs expected {n_imgs}")
+
+            # Flatten long format (images indexed as img0,img1,...)
+            data = {"row": rows}
+            for i in range(n_imgs):
+                data[f"img{i}"] = [meta_with_paths[row][i] for row in rows]
+
+        # --- Convert to wide "multi-list" format ---
+        df = pd.DataFrame(data)
+        row_col = df.columns[0]       # first col = row labels
+        col_cols = df.columns[1:]     # the rest = groupable cols
+
+        df = (
+            df.melt(id_vars=[row_col], var_name="col", value_name="path")
+            .groupby([row_col, "col"])["path"]
+            .apply(list)
+            .unstack("col")
+            .reset_index()
+        )
+
+        if log:
+            csvfile.fn_display_df(df)
 
-        Args:
-            csv_path (str): Path to CSV file.
-            max_width (int): Maximum width of each image in pixels.
-            max_height (int): Maximum height of each image in pixels.
+        return df
+
+    @staticmethod
+    def _parse_cell_to_list(cell) -> List[str]:
+        """Parse a DataFrame cell that may already be a list, a Python-list string, JSON list string,
+        or a single path. Returns list[str]."""
+        if cell is None:
+            return []
+        # pandas NA
+        try:
+            if pd.isna(cell):
+                return []
+        except Exception:
+            pass
+
+        if isinstance(cell, list):
+            return [str(x) for x in cell]
+
+        if isinstance(cell, (tuple, set)):
+            return [str(x) for x in cell]
+
+        if isinstance(cell, str):
+            s = cell.strip()
+            if not s:
+                return []
+
+            # Try Python literal (e.g. "['a','b']")
+            try:
+                val = ast.literal_eval(s)
+                if isinstance(val, (list, tuple)):
+                    return [str(x) for x in val]
+                if isinstance(val, str):
+                    return [val]
+            except Exception:
+                pass
+
+            # Try JSON
+            try:
+                val = json.loads(s)
+                if isinstance(val, list):
+                    return [str(x) for x in val]
+                if isinstance(val, str):
+                    return [val]
+            except Exception:
+                pass
+
+            # Fallback: split on common separators
+            for sep in [";;", ";", "|", ", "]:
+                if sep in s:
+                    parts = [p.strip() for p in s.split(sep) if p.strip()]
+                    if parts:
+                        return parts
+
+            # Single path string
+            return [s]
+
+        # anything else -> coerce to string
+        return [str(cell)]
+
+    @staticmethod
+    def plot_image_grid(
+        indir_or_csvf_or_df: Union[str, pd.DataFrame],
+        img_width: int = 300,
+        img_height: int = 300,
+        img_stack_direction: str = "horizontal",  # "horizontal" or "vertical"
+        img_stack_padding_px: int = 5,
+        img_scale_mode: str = "fit",  # "fit" or "fill"
+        format_row_label_func: Optional[Callable[[str], str]] = None,
+        format_col_label_func: Optional[Callable[[str], str]] = None,
+        title: str = "",
+        tickfont=dict(size=16, family="Arial", color="black"),  # <-- bigger labels
+        fig_margin: dict = dict(l=50, r=50, t=50, b=50),
+        outline_color: str = "",
+        outline_size: int = 1,
+        cell_margin_px: int = 10,  # spacing between cells
+        row_line_size: int = 0,  # if >0, draw horizontal dotted lines
+        col_line_size: int = 0   # if >0, draw vertical dotted lines
+    ):
         """
-        # Load CSV
-        df = csvfile.read_auto_sep(csv_path, sep=sep)
-
-        # Column names for headers
-        col_names = df.columns.tolist()
-
-        # Function to convert image to base64
-        def pil_to_base64(img_path):
-            with Image.open(img_path) as im:
-                im.thumbnail((max_width, max_height))
-                buffer = BytesIO()
-                im.save(buffer, format="PNG")
-                encoded = base64.b64encode(buffer.getvalue()).decode()
-                return "data:image/png;base64," + encoded
-
-        # Initialize figure
+            Plot a grid of images using Plotly.
+
+            - Accepts DataFrame where each cell is either:
+                * a Python list object,
+                * a string representation of a Python list (e.g. "['a','b']"),
+                * a JSON list string, or
+                * a single path string.
+            - For each cell, stack the images into a single composite that exactly fits
+            (img_width, img_height) is the target size for each individual image in the stack.
+            The final cell size will depend on the number of images and stacking direction.
+        """
+
+        def process_image_for_slot(path: str, target_size: Tuple[int, int], scale_mode: str, outline: str, outline_size: int) -> Image.Image:
+            try:
+                img = Image.open(path).convert("RGB")
+            except Exception:
+                return Image.new("RGB", target_size, (255, 255, 255))
+
+            if scale_mode == "fit":
+                img_ratio = img.width / img.height
+                target_ratio = target_size[0] / target_size[1]
+
+                if img_ratio > target_ratio:
+                    new_height = target_size[1]
+                    new_width = max(1, int(new_height * img_ratio))
+                else:
+                    new_width = target_size[0]
+                    new_height = max(1, int(new_width / img_ratio))
+
+                img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
+                left = (new_width - target_size[0]) // 2
+                top = (new_height - target_size[1]) // 2
+                right = left + target_size[0]
+                bottom = top + target_size[1]
+
+                if len(outline) == 7 and outline.startswith("#"):
+                    border_px = outline_size
+                    bordered = Image.new("RGB", (target_size[0] + 2*border_px, target_size[1] + 2*border_px), outline)
+                    bordered.paste(img.crop((left, top, right, bottom)), (border_px, border_px))
+                    return bordered
+                return img.crop((left, top, right, bottom))
+
+            elif scale_mode == "fill":
+                if len(outline) == 7 and outline.startswith("#"):
+                    border_px = outline_size
+                    bordered = Image.new("RGB", (target_size[0] + 2*border_px, target_size[1] + 2*border_px), outline)
+                    img = img.resize(target_size, Image.Resampling.LANCZOS)
+                    bordered.paste(img, (border_px, border_px))
+                    return bordered
+                return img.resize(target_size, Image.Resampling.LANCZOS)
+            else:
+                raise ValueError("img_scale_mode must be 'fit' or 'fill'.")
+
+        def stack_images_base64(image_paths: List[str], direction: str, single_img_size: Tuple[int,int], outline: str, outline_size: int, padding: int) -> Tuple[str, Tuple[int,int]]:
+            image_paths = [p for p in image_paths if p is not None and str(p).strip() != ""]
+            n = len(image_paths)
+            if n == 0:
+                blank = Image.new("RGB", single_img_size, (255,255,255))
+                buf = BytesIO()
+                blank.save(buf, format="PNG")
+                return "data:image/png;base64," + base64.b64encode(buf.getvalue()).decode(), single_img_size
+
+            processed = [process_image_for_slot(p, single_img_size, img_scale_mode, outline, outline_size) for p in image_paths]
+            pad_total = padding * (n-1)
+
+            if direction == "horizontal":
+                total_w = sum(im.width for im in processed) + pad_total
+                total_h = max(im.height for im in processed)
+                stacked = Image.new("RGB", (total_w, total_h), (255,255,255))
+                x = 0
+                for im in processed:
+                    stacked.paste(im, (x,0))
+                    x += im.width + padding
+            elif direction == "vertical":
+                total_w = max(im.width for im in processed)
+                total_h = sum(im.height for im in processed) + pad_total
+                stacked = Image.new("RGB", (total_w, total_h), (255,255,255))
+                y = 0
+                for im in processed:
+                    stacked.paste(im, (0,y))
+                    y += im.height + padding
+            else:
+                raise ValueError("img_stack_direction must be 'horizontal' or 'vertical'.")
+
+            buf = BytesIO()
+            stacked.save(buf, format="PNG")
+            encoded = base64.b64encode(buf.getvalue()).decode()
+            return f"data:image/png;base64,{encoded}", (total_w, total_h)
+
+        # --- Load DataFrame ---
+        if isinstance(indir_or_csvf_or_df, str):
+            fname, ext = os.path.splitext(indir_or_csvf_or_df)
+            if ext.lower() == ".csv":
+                df = pd.read_csv(indir_or_csvf_or_df)
+            elif os.path.isdir(indir_or_csvf_or_df):
+                df = PlotHelper.img_grid_indir_1(indir_or_csvf_or_df, log=False)
+            else:
+                raise ValueError("Input string must be a valid CSV file or directory path")
+        elif isinstance(indir_or_csvf_or_df, pd.DataFrame):
+            df = indir_or_csvf_or_df.copy()
+        else:
+            raise ValueError("Input must be CSV file path, DataFrame, or directory path")
+
+        rows = df.iloc[:,0].astype(str).tolist()
+        columns = list(df.columns[1:])
+        n_rows, n_cols = len(rows), len(columns)
+
         fig = go.Figure()
+        col_widths = [0]*n_cols
+        row_heights = [0]*n_rows
 
-        n_rows = len(df)
-        n_cols = len(df.columns) - 1  # skip label column
+        cell_imgs = [[None]*n_cols for _ in range(n_rows)]
+        for i in range(n_rows):
+            for j, col_label in enumerate(columns):
+                raw_cell = df.iloc[i, j+1]
+                image_paths = PlotHelper._parse_cell_to_list(raw_cell)
+                image_paths = [str(p).strip() for p in image_paths if str(p).strip() != ""]
+
+                img_src, (cell_w_actual, cell_h_actual) = stack_images_base64(
+                    image_paths, img_stack_direction, (img_width, img_height),
+                    outline=outline_color, outline_size=outline_size,
+                    padding=img_stack_padding_px
+                )
 
-        # Add images
-        for i, row in df.iterrows():
-            for j, col in enumerate(df.columns[1:]):
-                img_path = row[col]
-                img_src = pil_to_base64(img_path)
+                col_widths[j] = max(col_widths[j], cell_w_actual)
+                row_heights[i] = max(row_heights[i], cell_h_actual)
+                cell_imgs[i][j] = img_src
+
+        # Compute x/y positions including cell_margin
+        x_positions = []
+        cum_w = 0
+        for w in col_widths:
+            x_positions.append(cum_w)
+            cum_w += w + cell_margin_px
+
+        y_positions = []
+        cum_h = 0
+        for h in row_heights:
+            y_positions.append(-cum_h)
+            cum_h += h + cell_margin_px
+
+        # Add images to figure
+        for i in range(n_rows):
+            for j in range(n_cols):
                 fig.add_layout_image(
                     dict(
-                        source=img_src,
-                        x=j,
-                        y=-i,  # negative to have row 0 on top
+                        source=cell_imgs[i][j],
+                        x=x_positions[j],
+                        y=y_positions[i],
                         xref="x",
                         yref="y",
-                        sizex=1,
-                        sizey=1,
+                        sizex=col_widths[j],
+                        sizey=row_heights[i],
                         xanchor="left",
                         yanchor="top",
-                        layer="above"
+                        layer="above",
                     )
                 )
+        # ! Optional grid lines
+        # Add horizontal grid lines if row_line_size > 0
+        if row_line_size > 0:
+            for i in range(1, n_rows):
+                # Place line in the middle of the gap between rows
+                y = (
+                    y_positions[i - 1] - row_heights[i - 1] - y_positions[i]
+                ) / 2 + y_positions[i]
+                fig.add_shape(
+                    type="line",
+                    x0=-cell_margin_px,
+                    x1=cum_w - cell_margin_px,
+                    y0=y,
+                    y1=y,
+                    line=dict(width=row_line_size, color="black", dash="dot"),
+                )
+
+        # Add vertical grid lines if col_line_size > 0
+        if col_line_size > 0:
+            for j in range(1, n_cols):
+                x = x_positions[j] - cell_margin_px / 2
+                fig.add_shape(
+                    type="line",
+                    x0=x,
+                    x1=x,
+                    y0=cell_margin_px,
+                    y1=-cum_h + cell_margin_px,
+                    line=dict(width=col_line_size, color="black", dash="dot"),
+                )
+        # Axis labels
+        col_labels = [format_col_label_func(c) if format_col_label_func else c for c in columns]
+        row_labels = [format_row_label_func(r) if format_row_label_func else r for r in rows]
 
-        # Set axes for grid layout
         fig.update_xaxes(
-            tickvals=list(range(n_cols)),
-            ticktext=list(df.columns[1:]),
-            range=[-0.5, n_cols-0.5],
+            tickvals=[x_positions[j] + col_widths[j]/2 for j in range(n_cols)],
+            ticktext=col_labels,
+            range=[-cell_margin_px, cum_w - cell_margin_px],
             showgrid=False,
-            zeroline=False
+            zeroline=False,
+            tickfont=tickfont  # <-- apply bigger font here
         )
         fig.update_yaxes(
-            tickvals=[-i for i in range(n_rows)],
-            ticktext=df[df.columns[0]],
-            range=[-n_rows + 0.5, 0.5],
+            tickvals=[y_positions[i] - row_heights[i]/2 for i in range(n_rows)],
+            ticktext=row_labels,
+            range=[-cum_h + cell_margin_px, cell_margin_px],
             showgrid=False,
-            zeroline=False
+            zeroline=False,
+            tickfont=tickfont  # <-- apply bigger font here
         )
 
         fig.update_layout(
-            width=max_width*n_cols,
-            height=max_height*n_rows,
-            margin=dict(l=100, r=20, t=50, b=50)
+            width=cum_w + 100,
+            height=cum_h + 100,
+            title=title,
+            title_x=0.5,
+            margin=fig_margin,
         )
 
         fig.show()
 
     @staticmethod
-    # this plot_df contains the data to be plotted (row, column)
-    def img_grid_df(input_dir, log=False):
-        rows = fs.list_dirs(input_dir)
-        rows = [r for r in rows if r.startswith("row")]
-        meta_dict = {}
-        cols_of_row = None
-        for row in rows:
-            row_path = os.path.join(input_dir, row)
-            cols = sorted(fs.list_dirs(row_path))
-            if cols_of_row is None:
-                cols_of_row = cols
-            else:
-                if cols_of_row != cols:
-                    raise ValueError(
-                        f"Row {row} has different columns than previous rows: {cols_of_row} vs {cols}"
-                    )
-            meta_dict[row] = cols
-
-        meta_dict_with_paths = {}
-        for row, cols in meta_dict.items():
-            meta_dict_with_paths[row] = {
-                col: fs.filter_files_by_extension(
-                    os.path.join(input_dir, row, col), ["png", "jpg", "jpeg"]
-                )
-                for col in cols
-            }
-        first_row = list(meta_dict_with_paths.keys())[0]
-        first_col = list(meta_dict_with_paths[first_row].keys())[0]
-        len_first_col = len(meta_dict_with_paths[first_row][first_col])
-        for row, cols in meta_dict_with_paths.items():
-            for col, paths in cols.items():
-                if len(paths) != len_first_col:
-                    raise ValueError(
-                        f"Row {row}, Column {col} has different number of files: {len(paths)} vs {len_first_col}"
-                    )
-        cols = sorted(meta_dict_with_paths[first_row].keys())
-        rows_set = sorted(meta_dict_with_paths.keys())
-        row_per_col = len(meta_dict_with_paths[first_row][first_col])
-        rows = [item for item in rows_set for _ in range(row_per_col)]
-        data_dict = {}
-        data_dict["row"] = rows
-        col_data = {col: [] for col in cols}
-        for row_base in rows_set:
-            for col in cols:
-                for i in range(row_per_col):
-                    col_data[col].append(meta_dict_with_paths[row_base][col][i])
-        data_dict.update(col_data)
-        df = pd.DataFrame(data_dict)
-        if log:
-            csvfile.fn_display_df(df)
-        return df
-
-    @staticmethod
-    def plot_image_grid(
-        csv_file_or_df: Union[str, pd.DataFrame],
-        max_width: int = 300,
-        max_height: int = 300,
-        img_stack_direction: str = "horizontal",
-        img_stack_padding_px: int = 10,
+    def plot_image_grid1(
+        indir_or_csvf_or_df: Union[str, pd.DataFrame],
+        img_width: int = 300,
+        img_height: int = 300,
+        img_stack_direction: str = "horizontal",  # "horizontal" or "vertical"
+        img_stack_padding_px: int = 5,
+        img_scale_mode: str = "fit",  # "fit" or "fill"
         format_row_label_func: Optional[Callable[[str], str]] = None,
-        format_col_label_func: Optional[Callable[[str, str], str]] = None,
+        format_col_label_func: Optional[Callable[[str], str]] = None,
         title: str = "",
+        tickfont=dict(size=16, family="Arial", color="black"),  # <-- bigger labels
+        fig_margin: dict = dict(l=50, r=50, t=50, b=50),
+        outline_color: str = "",
+        outline_size: int = 1,
+        cell_margin_px: int = 10,  # padding (top, left, right, bottom) inside each cell
+        row_line_size: int = 0,  # if >0, draw horizontal dotted lines
+        col_line_size: int = 0,  # if >0, draw vertical dotted lines
     ):
         """
-        Plot a grid of images using Plotly from a DataFrame.
-
-        Args:
-            df (pd.DataFrame): DataFrame with first column as row labels, remaining columns as image paths.
-            max_width (int): Maximum width of stacked images per cell in pixels.
-            max_height (int): Maximum height of stacked images per cell in pixels.
-            img_stack_direction (str): "horizontal" or "vertical" stacking.
-            img_stack_padding_px (int): Padding between stacked images in pixels.
-            format_row_label_func (Callable): Function to format row labels.
-            format_col_label_func (Callable): Function to format column labels.
-            title (str): Figure title.
+        Plot a grid of images using Plotly.
+
+        - Accepts DataFrame where each cell is either:
+            * a Python list object,
+            * a string representation of a Python list (e.g. "['a','b']"),
+            * a JSON list string, or
+            * a single path string.
+        - For each cell, stack the images into a single composite that exactly fits
+        (img_width, img_height) is the target size for each individual image in the stack.
+        The final cell size will depend on the number of images and stacking direction.
         """
 
+        def process_image_for_slot(
+            path: str,
+            target_size: Tuple[int, int],
+            scale_mode: str,
+            outline: str,
+            outline_size: int,
+        ) -> Image.Image:
+            try:
+                img = Image.open(path).convert("RGB")
+            except Exception:
+                return Image.new("RGB", target_size, (255, 255, 255))
+
+            if scale_mode == "fit":
+                img_ratio = img.width / img.height
+                target_ratio = target_size[0] / target_size[1]
+
+                if img_ratio > target_ratio:
+                    new_height = target_size[1]
+                    new_width = max(1, int(new_height * img_ratio))
+                else:
+                    new_width = target_size[0]
+                    new_height = max(1, int(new_width / img_ratio))
+
+                img = img.resize((new_width, new_height), Image.Resampling.LANCZOS)
+                left = (new_width - target_size[0]) // 2
+                top = (new_height - target_size[1]) // 2
+                right = left + target_size[0]
+                bottom = top + target_size[1]
+
+                if len(outline) == 7 and outline.startswith("#"):
+                    border_px = outline_size
+                    bordered = Image.new(
+                        "RGB",
+                        (target_size[0] + 2 * border_px, target_size[1] + 2 * border_px),
+                        outline,
+                    )
+                    bordered.paste(
+                        img.crop((left, top, right, bottom)), (border_px, border_px)
+                    )
+                    return bordered
+                return img.crop((left, top, right, bottom))
+
+            elif scale_mode == "fill":
+                if len(outline) == 7 and outline.startswith("#"):
+                    border_px = outline_size
+                    bordered = Image.new(
+                        "RGB",
+                        (target_size[0] + 2 * border_px, target_size[1] + 2 * border_px),
+                        outline,
+                    )
+                    img = img.resize(target_size, Image.Resampling.LANCZOS)
+                    bordered.paste(img, (border_px, border_px))
+                    return bordered
+                return img.resize(target_size, Image.Resampling.LANCZOS)
+            else:
+                raise ValueError("img_scale_mode must be 'fit' or 'fill'.")
+
         def stack_images_base64(
-            image_paths: List[str], direction: str, target_size: Tuple[int, int]
-        ) -> str:
-            """Stack images and return base64-encoded PNG."""
-            if not image_paths:
-                return ""
-
-            processed_images = []
-            for path in image_paths:
-                try:
-                    img = Image.open(path).convert("RGB")
-                    img.thumbnail(target_size, Image.Resampling.LANCZOS)
-                    processed_images.append(img)
-                except:
-                    # blank image if error
-                    processed_images.append(Image.new("RGB", target_size, (255, 255, 255)))
-
-            # Stack
-            widths, heights = zip(*(img.size for img in processed_images))
-            if direction == "horizontal":
-                total_width = sum(widths) + img_stack_padding_px * (
-                    len(processed_images) - 1
-                )
-                total_height = max(heights)
-                stacked = Image.new("RGB", (total_width, total_height), (255, 255, 255))
-                x_offset = 0
-                for im in processed_images:
-                    stacked.paste(im, (x_offset, 0))
-                    x_offset += im.width + img_stack_padding_px
-            elif direction == "vertical":
-                total_width = max(widths)
-                total_height = sum(heights) + img_stack_padding_px * (
-                    len(processed_images) - 1
+            image_paths: List[str],
+            direction: str,
+            single_img_size: Tuple[int, int],
+            outline: str,
+            outline_size: int,
+            padding: int,
+        ) -> Tuple[str, Tuple[int, int]]:
+            image_paths = [p for p in image_paths if p is not None and str(p).strip() != ""]
+            n = len(image_paths)
+            if n == 0:
+                blank = Image.new("RGB", single_img_size, (255, 255, 255))
+                buf = BytesIO()
+                blank.save(buf, format="PNG")
+                return (
+                    "data:image/png;base64," + base64.b64encode(buf.getvalue()).decode(),
+                    single_img_size,
                 )
-                stacked = Image.new("RGB", (total_width, total_height), (255, 255, 255))
-                y_offset = 0
-                for im in processed_images:
-                    stacked.paste(im, (0, y_offset))
-                    y_offset += im.height + img_stack_padding_px
-            else:
-                raise ValueError("img_stack_direction must be 'horizontal' or 'vertical'")
 
-            # Encode as base64
-            buffer = BytesIO()
-            stacked.save(buffer, format="PNG")
-            encoded = base64.b64encode(buffer.getvalue()).decode()
-            return "data:image/png;base64," + encoded
+            processed = [
+                process_image_for_slot(
+                    p, single_img_size, img_scale_mode, outline, outline_size
+                )
+                for p in image_paths
+            ]
+            pad_total = padding * (n - 1)
 
-        # Load DataFrame if a file path is provided
-        if isinstance(csv_file_or_df, str):
-            df = csvfile.read_auto_sep(csv_file_or_df)
+            if direction == "horizontal":
+                total_w = sum(im.width for im in processed) + pad_total
+                total_h = max(im.height for im in processed)
+                stacked = Image.new("RGB", (total_w, total_h), (255, 255, 255))
+                x = 0
+                for im in processed:
+                    stacked.paste(im, (x, 0))
+                    x += im.width + padding
+            elif direction == "vertical":
+                total_w = max(im.width for im in processed)
+                total_h = sum(im.height for im in processed) + pad_total
+                stacked = Image.new("RGB", (total_w, total_h), (255, 255, 255))
+                y = 0
+                for im in processed:
+                    stacked.paste(im, (0, y))
+                    y += im.height + padding
+            else:
+                raise ValueError("img_stack_direction must be 'horizontal' or 'vertical'.")
+
+            buf = BytesIO()
+            stacked.save(buf, format="PNG")
+            encoded = base64.b64encode(buf.getvalue()).decode()
+            return f"data:image/png;base64,{encoded}", (total_w, total_h)
+
+        def compute_stacked_size(
+            image_paths: List[str],
+            direction: str,
+            single_w: int,
+            single_h: int,
+            padding: int,
+            outline: str,
+            outline_size: int,
+        ) -> Tuple[int, int]:
+            image_paths = [p for p in image_paths if p is not None and str(p).strip() != ""]
+            n = len(image_paths)
+            if n == 0:
+                return single_w, single_h
+            has_outline = len(outline) == 7 and outline.startswith("#")
+            border = 2 * outline_size if has_outline else 0
+            unit_w = single_w + border
+            unit_h = single_h + border
+            if direction == "horizontal":
+                total_w = n * unit_w + (n - 1) * padding
+                total_h = unit_h
+            elif direction == "vertical":
+                total_w = unit_w
+                total_h = n * unit_h + (n - 1) * padding
+            else:
+                raise ValueError("img_stack_direction must be 'horizontal' or 'vertical'.")
+            return total_w, total_h
+
+        # --- Load DataFrame ---
+        if isinstance(indir_or_csvf_or_df, str):
+            fname, ext = os.path.splitext(indir_or_csvf_or_df)
+            if ext.lower() == ".csv":
+                df = pd.read_csv(indir_or_csvf_or_df)
+            elif os.path.isdir(indir_or_csvf_or_df):
+                df = PlotHelper.img_grid_indir_1(indir_or_csvf_or_df, log=False)
+            else:
+                raise ValueError("Input string must be a valid CSV file or directory path")
+        elif isinstance(indir_or_csvf_or_df, pd.DataFrame):
+            df = indir_or_csvf_or_df.copy()
         else:
-            df = csv_file_or_df
-        assert isinstance(df, pd.DataFrame), "Input must be a DataFrame or valid CSV file path"
+            raise ValueError("Input must be CSV file path, DataFrame, or directory path")
 
-        rows = df[df.columns[0]].tolist()
-        columns = df.columns[1:].tolist()
+        rows = df.iloc[:, 0].astype(str).tolist()
+        columns = list(df.columns[1:])
         n_rows, n_cols = len(rows), len(columns)
 
         fig = go.Figure()
 
-        for i, row_label in enumerate(rows):
-            for j, col_label in enumerate(columns):
-                image_paths = df.loc[i, col_label]
-                if isinstance(image_paths, str):
-                    image_paths = [image_paths]
-                img_src = stack_images_base64(
-                    image_paths, img_stack_direction, (max_width, max_height)
+        # First pass: compute content sizes
+        content_col_max = [0] * n_cols
+        content_row_max = [0] * n_rows
+        cell_paths = [[None] * n_cols for _ in range(n_rows)]
+        for i in range(n_rows):
+            for j in range(n_cols):
+                raw_cell = df.iloc[i, j + 1]
+                paths = PlotHelper._parse_cell_to_list(raw_cell)
+                image_paths = [str(p).strip() for p in paths if str(p).strip() != ""]
+                cell_paths[i][j] = image_paths
+                cw, ch = compute_stacked_size(
+                    image_paths,
+                    img_stack_direction,
+                    img_width,
+                    img_height,
+                    img_stack_padding_px,
+                    outline_color,
+                    outline_size,
                 )
-
+                content_col_max[j] = max(content_col_max[j], cw)
+                content_row_max[i] = max(content_row_max[i], ch)
+
+        # Compute display sizes (content max + padding)
+        display_col_w = [content_col_max[j] + 2 * cell_margin_px for j in range(n_cols)]
+        display_row_h = [content_row_max[i] + 2 * cell_margin_px for i in range(n_rows)]
+
+        # Compute positions (cells adjacent)
+        x_positions = []
+        cum_w = 0
+        for dw in display_col_w:
+            x_positions.append(cum_w)
+            cum_w += dw
+
+        y_positions = []
+        cum_h = 0
+        for dh in display_row_h:
+            y_positions.append(-cum_h)
+            cum_h += dh
+
+        # Second pass: create padded images (centered content)
+        cell_imgs = [[None] * n_cols for _ in range(n_rows)]
+        p = cell_margin_px
+        for i in range(n_rows):
+            for j in range(n_cols):
+                image_paths = cell_paths[i][j]
+                content_src, (cw, ch) = stack_images_base64(
+                    image_paths,
+                    img_stack_direction,
+                    (img_width, img_height),
+                    outline_color,
+                    outline_size,
+                    img_stack_padding_px,
+                )
+                if cw == 0 or ch == 0:
+                    # Skip empty, but create white padded
+                    pad_w = display_col_w[j]
+                    pad_h = display_row_h[i]
+                    padded = Image.new("RGB", (pad_w, pad_h), (255, 255, 255))
+                else:
+                    content_img = Image.open(
+                        BytesIO(base64.b64decode(content_src.split(",")[1]))
+                    )
+                    ca_w = content_col_max[j]
+                    ca_h = content_row_max[i]
+                    left_offset = (ca_w - cw) // 2
+                    top_offset = (ca_h - ch) // 2
+                    pad_w = display_col_w[j]
+                    pad_h = display_row_h[i]
+                    padded = Image.new("RGB", (pad_w, pad_h), (255, 255, 255))
+                    paste_x = p + left_offset
+                    paste_y = p + top_offset
+                    padded.paste(content_img, (paste_x, paste_y))
+                buf = BytesIO()
+                padded.save(buf, format="PNG")
+                encoded = base64.b64encode(buf.getvalue()).decode()
+                cell_imgs[i][j] = f"data:image/png;base64,{encoded}"
+
+        # Add images to figure
+        for i in range(n_rows):
+            for j in range(n_cols):
                 fig.add_layout_image(
                     dict(
-                        source=img_src,
-                        x=j,
-                        y=-i,  # negative so row 0 on top
+                        source=cell_imgs[i][j],
+                        x=x_positions[j],
+                        y=y_positions[i],
                         xref="x",
                         yref="y",
-                        sizex=1,
-                        sizey=1,
+                        sizex=display_col_w[j],
+                        sizey=display_row_h[i],
                         xanchor="left",
                         yanchor="top",
                         layer="above",
                     )
                 )
 
-        # Format axis labels
+        # Optional grid lines (at cell boundaries, adjusted for inter-content spaces)
+        if row_line_size > 0:
+            for i in range(1, n_rows):
+                y = (y_positions[i - 1] - display_row_h[i - 1] + y_positions[i]) / 2
+                fig.add_shape(
+                    type="line",
+                    x0=-p,
+                    x1=cum_w,
+                    y0=y,
+                    y1=y,
+                    line=dict(width=row_line_size, color="black", dash="dot"),
+                )
+
+        if col_line_size > 0:
+            for j in range(1, n_cols):
+                x = x_positions[j]
+                fig.add_shape(
+                    type="line",
+                    x0=x,
+                    x1=x,
+                    y0=p,
+                    y1=-cum_h,
+                    line=dict(width=col_line_size, color="black", dash="dot"),
+                )
+
+        # Axis labels
         col_labels = [
-            format_col_label_func(c, pattern="___") if format_col_label_func else c
-            for c in columns
+            format_col_label_func(c) if format_col_label_func else c for c in columns
         ]
         row_labels = [
             format_row_label_func(r) if format_row_label_func else r for r in rows
         ]
 
         fig.update_xaxes(
-            tickvals=list(range(n_cols)),
+            tickvals=[x_positions[j] + display_col_w[j] / 2 for j in range(n_cols)],
             ticktext=col_labels,
-            range=[-0.5, n_cols - 0.5],
+            range=[-p, cum_w],
             showgrid=False,
             zeroline=False,
+            tickfont=tickfont,  # <-- apply bigger font here
         )
         fig.update_yaxes(
-            tickvals=[-i for i in range(n_rows)],
+            tickvals=[y_positions[i] - display_row_h[i] / 2 for i in range(n_rows)],
             ticktext=row_labels,
-            range=[-n_rows + 0.5, 0.5],
+            range=[-cum_h, p],
             showgrid=False,
             zeroline=False,
+            tickfont=tickfont,  # <-- apply bigger font here
         )
 
         fig.update_layout(
-            width=max_width * n_cols + 200,  # extra for labels
-            height=max_height * n_rows + 100,
+            width=cum_w + 100,
+            height=cum_h + 100,
             title=title,
-            margin=dict(l=100, r=20, t=50, b=50),
+            title_x=0.5,
+            margin=fig_margin,
         )
 
         fig.show()

halib/utils/video.py

@@ -23,11 +23,17 @@ class VideoUtils:
         # Get the FPS
         fps = cap.get(cv2.CAP_PROP_FPS)
 
+        # get frame size
+        width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+        height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+
         # Release the video capture object
         cap.release()
 
         meta_dict = {
             "video_path": video_path,
+            "width": width,
+            "height": height,
             "frame_count": frame_count,
             "fps": fps
         }

{halib-0.1.91.dist-info → halib-0.1.93.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: halib
-Version: 0.1.91
+Version: 0.1.93
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -50,7 +50,11 @@ Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-Helper package for coding and automation
+# Helper package for coding and automation
+
+**Version 0.1.93**
++ `research/plot': add `PlotHelper` class to plot train history + plot grid of images (e.g., image samples from dataset or model outputs)
+
 
 **Version 0.1.91**
 + `research/param_gen`: add `ParamGen` class to generate parameter list from yaml file for hyperparameter search (grid search, random search, etc.)

{halib-0.1.91.dist-info → halib-0.1.93.dist-info}/RECORD

@@ -37,7 +37,7 @@ halib/research/mics.py,sha256=uX17AGrBGER-OFMqUULE_A9YPPbn1RpQ4o5-omrmqZ8,377
 halib/research/params_gen.py,sha256=GcTMlniL0iE3HalJY-gVRiYa8Qy8u6nX4LkKZeMkct8,4262
 halib/research/perfcalc.py,sha256=qDa0sqfpWrwGZVJtjuUVFK7JX6j8xyXP9OnnfYmdamg,15898
 halib/research/perftb.py,sha256=FWg0b8wSgy4UwuvHSXwEqvTq1Rhi-z-HtAKuQg1lWc4,30989
-halib/research/plot.py,sha256=A3di1HZhIHIKf7d9b-I68yu_cm4u2LpHoPKlirCaNOI,17956
+halib/research/plot.py,sha256=4xMGJuP1lGN1wF27XFM5eMFb73Gu9qB582VZhTdcCSA,38418
 halib/research/profiler.py,sha256=GRAewTo0jGkOputjmRwtYVfJYBze_ivsOnrW9exWkPQ,11772
 halib/research/torchloader.py,sha256=yqUjcSiME6H5W210363HyRUrOi3ISpUFAFkTr1w4DCw,6503
 halib/research/wandb_op.py,sha256=YzLEqME5kIRxi3VvjFkW83wnFrsn92oYeqYuNwtYRkY,4188
@@ -53,9 +53,9 @@ halib/utils/dict_op.py,sha256=wYE6Iw-_CnCWdMg9tpJ2Y2-e2ESkW9FxmdBkZkbUh80,299
 halib/utils/gpu_mon.py,sha256=vD41_ZnmPLKguuq9X44SB_vwd9JrblO4BDzHLXZhhFY,2233
 halib/utils/listop.py,sha256=Vpa8_2fI0wySpB2-8sfTBkyi_A4FhoFVVvFiuvW8N64,339
 halib/utils/tele_noti.py,sha256=-4WXZelCA4W9BroapkRyIdUu9cUVrcJJhegnMs_WpGU,5928
-halib/utils/video.py,sha256=ZqzNVPgc1RZr_T0OlHvZ6SzyBpL7O27LtB86JMbBuR0,3059
-halib-0.1.91.dist-info/licenses/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
-halib-0.1.91.dist-info/METADATA,sha256=Fx00uk-QuxbR2R_Z2fJhnfHXSNEa3fHD0SdUFs-IiEs,6035
-halib-0.1.91.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-halib-0.1.91.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
-halib-0.1.91.dist-info/RECORD,,
+halib/utils/video.py,sha256=zLoj5EHk4SmP9OnoHjO8mLbzPdtq6gQPzTQisOEDdO8,3261
+halib-0.1.93.dist-info/licenses/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
+halib-0.1.93.dist-info/METADATA,sha256=F3yohPt7k7wer8ur5j0yIi51-oZd81PKdg0ixUmNLFo,6200
+halib-0.1.93.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+halib-0.1.93.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
+halib-0.1.93.dist-info/RECORD,,