From 81f55952f9eee08e90b0d8d99a359017dd76e3cf Mon Sep 17 00:00:00 2001 From: Christian Heller Date: Mon, 26 Aug 2024 08:14:29 +0200 Subject: [PATCH] Add hints where inside a work batch we currently are. --- browser.py | 7 +++++-- stable.py | 6 ++++-- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/browser.py b/browser.py index bd16e0e..9768443 100755 --- a/browser.py +++ b/browser.py @@ -202,8 +202,11 @@ class Window(Gtk.ApplicationWindow): if self.item_img: params_strs = [f'{k}: ' + str(getattr(self.item_img, k.lower())) for k in GEN_PARAMS] - self.metadata.props.label = '\n'.join([self.item_img.full_path] - + params_strs) + offset = len(self.unsorted_dirs) + 1 + position = self.selection.props.selected + 1 - offset + total = len(self.unsorted_files) + title = f'{self.item_img.full_path} ({position} of {total})' + self.metadata.props.label = '\n'.join([title] + params_strs) pic = Gtk.Picture.new_for_filename(self.item_img.full_path) pic.props.halign = Gtk.Align.START self.viewer.append(pic) diff --git a/stable.py b/stable.py index e601f00..29e1918 100755 --- a/stable.py +++ b/stable.py @@ -65,7 +65,7 @@ def parse_args(): temp_gen_paramses = [None] if parsed_args.defaults_from_stdin: temp_gen_paramses = [] - for line in [l for l in stdin.readlines() if len(l.rstrip())]: + for line in [line for line in stdin.readlines() if len(line.rstrip())]: temp_gen_paramses += [GenParams.from_str(line)] parsed_args.models = [] parsed_args.gen_paramses = [] @@ -119,6 +119,7 @@ def run(): from stable.core import ImageMaker # noqa: E402 old_model_path = '' maker = None + total = len(args.models) * args.quantity for i, model_name in enumerate(args.models): new_model_path = path_join(args.models_dir, model_name) if new_model_path != old_model_path: @@ -144,7 +145,8 @@ def run(): gen_params.seed = start_seed + n + seed_corrector path = save_path(i*args.quantity + n) maker.set_gen_params(gen_params) - print(f'GENERATING: {path}; {gen_params.to_str}') + index = i*args.quantity + n + print(f'GENERATING {index}/{total}: {path}; {gen_params.to_str}') maker.gen_image_to(path) -- 2.30.2