ro/stable-diffusion-telegram-bot
1
0
Fork 0
forked from yair/stable-diffusion-telegram-bot
Code Pull Requests Activity

added seed

Browse Source
This commit is contained in:
yair 2023-05-06 21:09:46 +03:00
parent cc4a637720
commit 1a70b2fd5b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -68,7 +68,6 @@ def parse_input(input_string):
# Initialize an empty payload with the 'prompt' key # Initialize an empty payload with the 'prompt' key
payload = {"prompt": ""} payload = {"prompt": ""}
# Check if the input_string starts with "/draw "
prompt = [] prompt = []
# Find all occurrences of keys (words ending with a colon) # Find all occurrences of keys (words ending with a colon)
@ -112,9 +111,9 @@ def parse_input(input_string):
# Join the prompt words and store it in the payload # Join the prompt words and store it in the payload
payload["prompt"] = " ".join(prompt) payload["prompt"] = " ".join(prompt)
# If the prompt is empty, return an empty dictionary # If the prompt is empty, set the input string as the prompt
if not payload["prompt"]: if not payload["prompt"]:
return {} payload["prompt"] = input_string.strip()
# Return the final payload # Return the final payload
return payload return payload
@ -153,6 +152,9 @@ def draw(client, message):
pnginfo.add_text("parameters", response2.json().get("info")) pnginfo.add_text("parameters", response2.json().get("info"))
image.save(f"{word}.png", pnginfo=pnginfo) image.save(f"{word}.png", pnginfo=pnginfo)
# Add a flag to check if the user provided a seed value
user_provided_seed = "seed" in payload
info_dict = response2.json() info_dict = response2.json()
seed_value = info_dict['info'].split(", Seed: ")[1].split(",")[0] seed_value = info_dict['info'].split(", Seed: ")[1].split(",")[0]
# print(seed_value) # print(seed_value)