diff --git a/.gitignore b/.gitignore index 6274c34..baaee34 100644 --- a/.gitignore +++ b/.gitignore @@ -1,5 +1,8 @@ *.png .env -.session +*.session vscode/ venv/ +*.session-journal +logs/stable_diff_telegram_bot.log +*.session diff --git a/main.py b/main.py index 2c42960..16e3460 100644 --- a/main.py +++ b/main.py @@ -65,23 +65,15 @@ def parse_input(input_string): if last_index != match.start(): prompt.append(input_string[last_index: match.start()].strip()) last_index = value_start_index - + if key == "ds": + key = "denoising_strength" + if key == "ng": + key = "negative_prompt" + if key in default_payload: value_end_index = re.search(r"(?=\s+\w+:|$)", input_string[value_start_index:]).start() value = input_string[value_start_index: value_start_index + value_end_index].strip() - - if key == "denoising_strength": - try: - ds_value = float(value) - if 0.0 <= ds_value <= 1.0: - payload[key] = ds_value - else: - payload[key] = default_payload[key] - except ValueError: - payload[key] = default_payload[key] - else: - payload[key] = value - + payload[key] = value last_index += value_end_index else: prompt.append(f"{key}:") diff --git a/requirements.txt b/requirements.txt index c5098fd..f82ac2b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -pyrogram==1.4.16 +pyrogram requests tgcrypto==1.2.2 Pillow