Skip to content

Commit

Permalink
Quick formatting update #5
Browse files Browse the repository at this point in the history
  • Loading branch information
@pramitchoudhary
pramitchoudhary committed Nov 7, 2022
1 parent 9aaca54 commit 3ddb9b3
Show file tree
Hide file tree
Showing 5 changed files with 353 additions and 237 deletions.
4 changes: 2 additions & 2 deletions img_styler/ui/app.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,8 @@ async def serve(q: Q):

log_args(q.args)

hash = q.args['#']
if hash == 'home':
hash = q.args["#"]
if hash == "home":
await home(q)

# Initialization
Expand Down
23 changes: 13 additions & 10 deletions img_styler/ui/capture.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -69,28 +69,31 @@
async def capture_img(q: Q):
events_dict = expando_to_dict(q.events)
_img = None
if 'the_plot' in events_dict.keys():
if "the_plot" in events_dict.keys():
if q.events.the_plot.selected:
q.page['capture_img'].content = f'You selected {q.events.the_plot.selected}'
if 'video' in events_dict.keys():
q.page["capture_img"].content = f"You selected {q.events.the_plot.selected}"
if "video" in events_dict.keys():
_img = q.events.video.click if q.events.video.click else None
if 'exit' in events_dict.keys():
if "exit" in events_dict.keys():
q.args.exit_camera = True
return _img


def draw_boundary(img, x, y, w, h, text: str, image_scale=1):
_img = Image.fromarray(img)
draw = ImageDraw.Draw(_img)
pt_1 = (int(x * image_scale), int(y * image_scale),
int((x + w) * image_scale), int((y + h) * image_scale))
draw.rounded_rectangle(pt_1, outline="yellow",
width=3, radius=7)
pt_1 = (
int(x * image_scale),
int(y * image_scale),
int((x + w) * image_scale),
int((y + h) * image_scale),
)
draw.rounded_rectangle(pt_1, outline="yellow", width=3, radius=7)

font = font_manager.FontProperties(family='sans-serif', weight='bold')
font = font_manager.FontProperties(family="sans-serif", weight="bold")
file = font_manager.findfont(font)
font = ImageFont.truetype(file, 24)
constant = 5
draw.text((x+w+constant, y), text, font=font, fill='blue')
draw.text((x + w + constant, y), text, font=font, fill="blue")

return np.array(_img)
163 changes: 109 additions & 54 deletions img_styler/ui/common.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,33 +15,42 @@
get_processed_face_card,
get_source_face_card,
get_style_face_card,
get_user_title
get_user_title,
)


async def update_faces(q: Q, save=False):
del q.page['prompt_form']
del q.page["prompt_form"]
if not q.client.source_face or not os.path.exists(q.client.source_face):
q.client.source_face = random.choice(q.app.source_faces)

if q.client.task_choice == 'B':
q.page['source_face'] = get_source_face_card(
img2buf(q.client.source_face), type='jpg'
if q.client.task_choice == "B":
q.page["source_face"] = get_source_face_card(
img2buf(q.client.source_face), type="jpg"
)
else:
txt_val = q.client.prompt_textbox if q.client.prompt_textbox else ''
q.page['source_face'] = get_source_face_card(
img2buf(q.client.source_face), type='jpg', height='520px', width='500px'
txt_val = q.client.prompt_textbox if q.client.prompt_textbox else ""
q.page["source_face"] = get_source_face_card(
img2buf(q.client.source_face), type="jpg", height="520px", width="500px"
)
if q.client.task_choice == 'D':
q.page['prompt_form'] = ui.form_card(ui.box('main', order=1, height='200px', width='900px'), items=[
ui.copyable_text(name='prompt_textbox', label='Prompt (Express your creativity)', multiline=True, value=txt_val),
ui.button(name='prompt_apply', label='Apply')])

del q.page['style_face']
if q.client.task_choice == 'B':
q.page['style_face'] = get_style_face_card(
img2buf(q.client.style_face), type='jpg'
if q.client.task_choice == "D":
q.page["prompt_form"] = ui.form_card(
ui.box("main", order=1, height="200px", width="900px"),
items=[
ui.copyable_text(
name="prompt_textbox",
label="Prompt (Express your creativity)",
multiline=True,
value=txt_val,
),
ui.button(name="prompt_apply", label="Apply"),
],
)

del q.page["style_face"]
if q.client.task_choice == "B":
q.page["style_face"] = get_style_face_card(
img2buf(q.client.style_face), type="jpg"
)
if save:
await q.page.save()
Expand All @@ -51,58 +60,104 @@ async def update_processed_face(q: Q, save=False):
img_buf = img2buf(q.client.processedimg) if q.client.processedimg else None
# Delete pages not needed.
# It's cheap to create them.
del q.page['processed_face']
del q.page['prompt_form']
del q.page["processed_face"]
del q.page["prompt_form"]

if q.client.task_choice == 'B':
q.page['processed_face'] = get_processed_face_card(img_buf, type='jpg')
if q.client.task_choice == "B":
q.page["processed_face"] = get_processed_face_card(img_buf, type="jpg")
else:
q.page['processed_face'] = get_processed_face_card(
img_buf, title="Fixed Image", type='jpg', layout_pos='middle_right', order=2
q.page["processed_face"] = get_processed_face_card(
img_buf, title="Fixed Image", type="jpg", layout_pos="middle_right", order=2
)
if q.client.task_choice == 'D':
q.page['prompt_form'] = ui.form_card(ui.box('main', order=1, height='320px', width='980px'), items=[
ui.inline(items=[
ui.checkbox(name='prompt_use_source_img', label='Use source image',
value=q.client.prompt_use_source_img, tooltip='Image-to-Image text-guided diffusion is applied by default.\
If un-checked, default Text-to-Image diffusion is used.'),
ui.button(name='prompt_apply', label='Apply')
]),
ui.textbox(name='prompt_textbox', label='Prompt', multiline=True, value=q.client.prompt_textbox),
ui.expander(name='expander', label='Settings', items=[
ui.slider(name='diffusion_n_steps', label='Steps', min=10, max=150, value=q.client.diffusion_n_steps,
tooltip='No of steps for image synthesis.'),
ui.slider(name='prompt_guidance_scale', label='Guidance scale', min=3, max=20, step=0.5, value=q.client.prompt_guidance_scale,
tooltip='No of steps for image synthesis.')]),
])
if q.client.task_choice == "D":
q.page["prompt_form"] = ui.form_card(
ui.box("main", order=1, height="320px", width="980px"),
items=[
ui.inline(
items=[
ui.checkbox(
name="prompt_use_source_img",
label="Use source image",
value=q.client.prompt_use_source_img,
tooltip="Image-to-Image text-guided diffusion is applied by default.\
If un-checked, default Text-to-Image diffusion is used.",
),
ui.button(name="prompt_apply", label="Apply"),
]
),
ui.textbox(
name="prompt_textbox",
label="Prompt",
multiline=True,
value=q.client.prompt_textbox,
),
ui.expander(
name="expander",
label="Settings",
items=[
ui.dropdown(
name="df_sampling_dropdown",
label="Samplers",
value="K-LMS",
choices=[
ui.choice(
name="K-LMS", label="K-LMS(Katherine Crowson)"
),
ui.choice(name="DDIM", label="DDIM"),
],
),
ui.slider(
name="diffusion_n_steps",
label="Steps",
min=10,
max=150,
value=q.client.diffusion_n_steps,
tooltip="No of steps for image synthesis.",
),
ui.slider(
name="prompt_guidance_scale",
label="Guidance scale",
min=3,
max=20,
step=0.5,
value=q.client.prompt_guidance_scale,
tooltip="No of steps for image synthesis.",
),
],
),
],
)
if save:
await q.page.save()


async def update_controls(q: Q, save=False):
q.page['controls'] = get_controls(q)
q.page["controls"] = get_controls(q)
if save:
await q.page.save()


async def update_gif(q: Q):
del q.page['processed_gif']
if q.client.task_choice == 'C':
del q.page["processed_gif"]
if q.client.task_choice == "C":
if q.client.gif_path:
with open(q.client.gif_path, "rb") as gif_file:
img_buf = base64.b64encode(gif_file.read()).decode("utf-8")
q.page['processed_gif'] = await get_gif_generated_card(q, img_buf)
q.page["processed_gif"] = await get_gif_generated_card(q, img_buf)
else:
q.page['processed_gif'] = get_processed_face_card(
None, title="Generated GIF", type='jpg', layout_pos='main',
q.page["processed_gif"] = get_processed_face_card(
None,
title="Generated GIF",
type="jpg",
layout_pos="main",
order=2,
empty_msg="After applying the edits, click 'Generate GIF' to generate a gif of the transformation!"
empty_msg="After applying the edits, click 'Generate GIF' to generate a gif of the transformation!",
)
await q.page.save()


async def progress_generate_gif(q: Q):
q.page['processed_gif'] = get_generate_gif_progress_card(title="")
q.page["processed_gif"] = get_generate_gif_progress_card(title="")
await q.page.save()


Expand All @@ -112,29 +167,29 @@ async def make_base_ui(q: Q, save=False):
q.app.logo_path = (
await q.run(q.site.upload, ["./img_styler/ui/assets/h2o_logo.svg"])
)[0]
q.page['meta'] = get_meta(q)
q.page['header'] = get_header(q)
q.page['user_title'] = get_user_title(q)
q.page["meta"] = get_meta(q)
q.page["header"] = get_header(q)
q.page["user_title"] = get_user_title(q)
await update_controls(q)
await update_faces(q)
await update_processed_face(q)
await update_gif(q)
q.page['footer'] = get_footer()
q.page["footer"] = get_footer()
if save:
await q.page.save()


def set_username(q: Q):
first = last = ''
first = last = ""
email = q.auth.username
names = email.split('@')[0].split('.')
names = email.split("@")[0].split(".")
if len(names) > 1:
first, *_, last = names
elif names:
first = names[0]
last = ''
last = ""

q.user.email = q.auth.username
q.user.first_name = first.strip().title()
q.user.last_name = last.strip().title()
q.user.full_name = f'{q.user.first_name} {q.user.last_name}'.strip().title()
q.user.full_name = f"{q.user.first_name} {q.user.last_name}".strip().title()
Loading

0 comments on commit 3ddb9b3

Please sign in to comment.