From c6bea3ac33a9b48db9174ef2e53ed8735eda931f Mon Sep 17 00:00:00 2001
From: Christian Heller <c.heller@plomlompom.de>
Date: Mon, 19 Aug 2024 07:36:37 +0200
Subject: [PATCH] Improve parameter parsing.

---
 stable.py      | 8 ++++++--
 stable/core.py | 2 ++
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/stable.py b/stable.py
index be5d135..460c0ee 100755
--- a/stable.py
+++ b/stable.py
@@ -40,14 +40,18 @@ def parse_args():
                         help=f'default: {DEFAULT_SCHEDULER}')
     parsed_args = parser.parse_args()
     if parsed_args.copy_params:
-        for section in parsed_args.copy_params.split('; '):
+        first_split = parsed_args.copy_params.split('; PROMPT: ', maxsplit=1)
+        if 2 == len(first_split):
+            parser.set_defaults(prompt=first_split[1])
+        for section in first_split[0].split('; '):
             key, val = section.split(': ', maxsplit=1)
             key = key.lower()
             if key in {'seed', 'height', 'width', 'n_steps'}:
                 val = int(val)
             elif key in {'guidance'}:
                 val = float(val)
-            setattr(parsed_args, key, val)
+            parser.set_defaults(**{key: val})
+    parsed_args = parser.parse_args()
     prefix = f'{argv[0]}: error: '
     if parsed_args.list_schedulers:
         required = {'model': 'm'}
diff --git a/stable/core.py b/stable/core.py
index d89d4a9..a5f7b56 100644
--- a/stable/core.py
+++ b/stable/core.py
@@ -23,6 +23,8 @@ class ImageMaker:
                 return self.target not in record.getMessage()
 
         self.model_filename = basename(model_path)
+        if '; ' in self.model_filename:
+            raise Exception('illegal filename (must not contain "; ")')
         self.seed = None
         self.prompt = None
         self.guidance = None
-- 
2.30.2