yyk19 commited on
Commit
dd51ab2
1 Parent(s): c3e2dbb

modify the interface.

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  import os
7
  import torch
8
  import time
9
-
10
  ALLOW_RUN_GENERATION = False
11
  def process_multi_wrapper(rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3,
12
  shared_prompt,
@@ -22,7 +22,7 @@ def process_multi_wrapper(rendered_txt_0, rendered_txt_1, rendered_txt_2, render
22
  shared_eta, shared_a_prompt, shared_n_prompt):
23
  global ALLOW_RUN_GENERATION
24
  if not ALLOW_RUN_GENERATION:
25
- return ["please get the glyph image first by clicking the 'Only Rendered' button"]
26
 
27
  rendered_txt_values = [rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3]
28
  width_values = [width_0, width_1, width_2, width_3]
@@ -32,7 +32,7 @@ def process_multi_wrapper(rendered_txt_0, rendered_txt_1, rendered_txt_2, render
32
  yaw_values = [yaw_0, yaw_1, yaw_2, yaw_3]
33
  num_rows_values = [num_rows_0, num_rows_1, num_rows_2, num_rows_3]
34
  ALLOW_RUN_GENERATION = False
35
- return render_tool.process_multi(rendered_txt_values, shared_prompt,
36
  width_values, ratio_values,
37
  top_left_x_values, top_left_y_values,
38
  yaw_values, num_rows_values,
@@ -64,7 +64,7 @@ def process_multi_wrapper_only_show_rendered(rendered_txt_0, rendered_txt_1, ren
64
  num_rows_values = [num_rows_0, num_rows_1, num_rows_2, num_rows_3]
65
  ALLOW_RUN_GENERATION = True
66
 
67
- return render_tool.process_multi(rendered_txt_values, shared_prompt,
68
  width_values, ratio_values,
69
  top_left_x_values, top_left_y_values,
70
  yaw_values, num_rows_values,
@@ -167,9 +167,11 @@ with block:
167
  shared_n_prompt = gr.Textbox(label="Negative Prompt",
168
  value='longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality')
169
 
170
- with gr.Row():
171
- # message = gr.outputs.Textbox()
172
- result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery").style(grid=2, height='auto')
 
 
173
 
174
  run_button.click(fn=process_multi_wrapper,
175
  inputs=[rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3,
@@ -184,7 +186,7 @@ with block:
184
  shared_ddim_steps, shared_guess_mode,
185
  shared_strength, shared_scale, shared_seed,
186
  shared_eta, shared_a_prompt, shared_n_prompt],
187
- outputs=[result_gallery])
188
 
189
  show_render_button.click(fn=process_multi_wrapper_only_show_rendered,
190
  inputs=[rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3,
@@ -199,11 +201,11 @@ with block:
199
  shared_ddim_steps, shared_guess_mode,
200
  shared_strength, shared_scale, shared_seed,
201
  shared_eta, shared_a_prompt, shared_n_prompt],
202
- outputs=[result_gallery])
203
 
204
  load_button.click(fn = load_ckpt,
205
  inputs = [model_ckpt],
206
- outputs = [result_gallery]
207
  )
208
 
209
  block.launch()
 
6
  import os
7
  import torch
8
  import time
9
+ from PIL import Image
10
  ALLOW_RUN_GENERATION = False
11
  def process_multi_wrapper(rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3,
12
  shared_prompt,
 
22
  shared_eta, shared_a_prompt, shared_n_prompt):
23
  global ALLOW_RUN_GENERATION
24
  if not ALLOW_RUN_GENERATION:
25
+ return "Please get the glyph image first by clicking the 'Only Rendered' button", None
26
 
27
  rendered_txt_values = [rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3]
28
  width_values = [width_0, width_1, width_2, width_3]
 
32
  yaw_values = [yaw_0, yaw_1, yaw_2, yaw_3]
33
  num_rows_values = [num_rows_0, num_rows_1, num_rows_2, num_rows_3]
34
  ALLOW_RUN_GENERATION = False
35
+ return "Image generation processes finished!", render_tool.process_multi(rendered_txt_values, shared_prompt,
36
  width_values, ratio_values,
37
  top_left_x_values, top_left_y_values,
38
  yaw_values, num_rows_values,
 
64
  num_rows_values = [num_rows_0, num_rows_1, num_rows_2, num_rows_3]
65
  ALLOW_RUN_GENERATION = True
66
 
67
+ return "Glyph images are generated!", render_tool.process_multi(rendered_txt_values, shared_prompt,
68
  width_values, ratio_values,
69
  top_left_x_values, top_left_y_values,
70
  yaw_values, num_rows_values,
 
167
  shared_n_prompt = gr.Textbox(label="Negative Prompt",
168
  value='longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality')
169
 
170
+ with gr.Accordion("Output", open=True):
171
+ with gr.Row():
172
+ message = gr.Text(interactive=False, label = "Message")
173
+ with gr.Row():
174
+ result_gallery = gr.Gallery(label='Images', show_label=False, elem_id="gallery").style(grid=2, height='auto')
175
 
176
  run_button.click(fn=process_multi_wrapper,
177
  inputs=[rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3,
 
186
  shared_ddim_steps, shared_guess_mode,
187
  shared_strength, shared_scale, shared_seed,
188
  shared_eta, shared_a_prompt, shared_n_prompt],
189
+ outputs=[message, result_gallery])
190
 
191
  show_render_button.click(fn=process_multi_wrapper_only_show_rendered,
192
  inputs=[rendered_txt_0, rendered_txt_1, rendered_txt_2, rendered_txt_3,
 
201
  shared_ddim_steps, shared_guess_mode,
202
  shared_strength, shared_scale, shared_seed,
203
  shared_eta, shared_a_prompt, shared_n_prompt],
204
+ outputs=[message, result_gallery])
205
 
206
  load_button.click(fn = load_ckpt,
207
  inputs = [model_ckpt],
208
+ outputs = [message, result_gallery]
209
  )
210
 
211
  block.launch()