aboutsummaryrefslogtreecommitdiff
path: root/modules
diff options
context:
space:
mode:
Diffstat (limited to 'modules')
-rw-r--r--modules/generation_parameters_copypaste.py9
-rw-r--r--modules/ui.py7
2 files changed, 9 insertions, 7 deletions
diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index 2b80737a..224a17ea 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -1,3 +1,5 @@
+import base64
+import io
import os
import re
import gradio as gr
@@ -14,6 +16,7 @@ type_of_gr_update = type(gr.update())
paste_fields = {}
bind_list = []
+
def quote(text):
if ',' not in str(text):
return text
@@ -23,6 +26,7 @@ def quote(text):
text = text.replace('"', '\\"')
return f'"{text}"'
+
def image_from_url_text(filedata):
if type(filedata) == dict and filedata["is_file"]:
filename = filedata["name"]
@@ -45,19 +49,23 @@ def image_from_url_text(filedata):
image = Image.open(io.BytesIO(filedata))
return image
+
def add_paste_fields(tabname, init_img, fields):
paste_fields[tabname] = {"init_img":init_img, "fields": fields}
+
def create_buttons(tabs_list):
buttons = {}
for tab in tabs_list:
buttons[tab] = gr.Button(f"Send to {tab}")
return buttons
+
#if send_generate_info is a tab name, mean generate_info comes from the params fields of the tab
def bind_buttons(buttons, send_image, send_generate_info):
bind_list.append([buttons, send_image, send_generate_info])
+
def run_bind():
for buttons, send_image, send_generate_info in bind_list:
for tab in buttons:
@@ -98,6 +106,7 @@ def run_bind():
outputs=None,
)
+
def parse_generation_parameters(x: str):
"""parses generation parameters string, the one you see in text field under the picture in UI:
```
diff --git a/modules/ui.py b/modules/ui.py
index 46657dd4..280910d0 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1,6 +1,4 @@
-import base64
import html
-import io
import json
import math
import mimetypes
@@ -18,13 +16,8 @@ import gradio as gr
import gradio.routes
import gradio.utils
import numpy as np
-import piexif
-import torch
from PIL import Image, PngImagePlugin
-import gradio as gr
-import gradio.utils
-import gradio.routes
from modules import sd_hijack, sd_models, localization, script_callbacks
from modules.paths import script_path