yizhangliu commited on
Commit
5863d22
·
1 Parent(s): d3c2ede

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +40 -49
app.py CHANGED
@@ -12,27 +12,8 @@ def getTextTrans(text, source='zh', target='en'):
12
  except Exception as e:
13
  return text
14
 
15
- extend_prompt_pipe = pipeline('text-generation', model='yizhangliu/prompt-extend', max_length=77)
16
 
17
- def extend_prompt(prompt):
18
- prompt_en = getTextTrans(prompt, source='zh', target='en')
19
- extend_prompt_en = extend_prompt_pipe(prompt_en+',', num_return_sequences=1)[0]["generated_text"]
20
- if (prompt != prompt_en):
21
- logger.info(f"extend_prompt__1__")
22
- extend_prompt_zh = getTextTrans(extend_prompt_en, source='en', target='zh')
23
- extend_prompt_out = f'{extend_prompt_zh} 【{extend_prompt_en}】'
24
- else:
25
- logger.info(f"extend_prompt__2__")
26
- extend_prompt_out = extend_prompt_en
27
-
28
- return prompt_en, extend_prompt_en, extend_prompt_out
29
-
30
- def change_prompt_radio(choice):
31
- if choice == "Original prompt":
32
- return gr.update(visible=False)
33
- else:
34
- return gr.update(visible=True)
35
-
36
  examples = [
37
  ['elon musk as thor'],
38
  ["giant dragon flying in the sky"],
@@ -125,10 +106,8 @@ start_work = """async() => {
125
  tabitems[i].childNodes[0].children[0].style.display='none';
126
  tabitems[i].childNodes[0].children[1].style.display='none';
127
  tabitems[i].childNodes[0].children[2].children[0].style.display='none';
128
- tabitems[i].childNodes[0].children[3].style.display='none';
129
-
130
- }
131
-
132
  }
133
 
134
  tab_demo = window['gradioEl'].querySelectorAll('#tab_demo')[0];
@@ -136,24 +115,15 @@ start_work = """async() => {
136
  tab_demo.setAttribute('style', 'height: 100%;');
137
  const page1 = window['gradioEl'].querySelectorAll('#page_1')[0];
138
  const page2 = window['gradioEl'].querySelectorAll('#page_2')[0];
139
- window['gradioEl'].querySelectorAll('.gr-radio')[0].disabled = "";
140
- window['gradioEl'].querySelectorAll('.gr-radio')[1].disabled = "";
141
-
142
  page1.style.display = "none";
143
  page2.style.display = "block";
144
  window['prevPrompt'] = '';
145
  window['doCheckPrompt'] = 0;
146
  window['checkPrompt'] = function checkPrompt() {
147
  try {
148
- texts = window['gradioEl'].querySelectorAll('textarea');
149
- text0 = texts[0];
150
- text1 = texts[1];
151
- text2 = texts[2];
152
- if (window['gradioEl'].querySelectorAll('.gr-radio')[0].checked) {
153
- text_value = text1.value;
154
- } else {
155
- text_value = text2.value;
156
- }
157
  progress_bar = window['gradioEl'].querySelectorAll('.progress-bar');
158
  if (window['doCheckPrompt'] === 0 && window['prevPrompt'] !== text_value && progress_bar.length == 0) {
159
  console.log('_____new prompt___[' + text_value + ']_');
@@ -189,6 +159,26 @@ start_work = """async() => {
189
  return false;
190
  }"""
191
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
192
  with gr.Blocks(title='Text-to-Image') as demo:
193
  with gr.Group(elem_id="page_1", visible=True) as page_1:
194
  with gr.Box():
@@ -199,24 +189,25 @@ with gr.Blocks(title='Text-to-Image') as demo:
199
  with gr.Group(elem_id="page_2", visible=False) as page_2:
200
  with gr.Row(elem_id="prompt_row0"):
201
  with gr.Column(id="input_col1"):
202
- prompt_input0 = gr.Textbox(lines=1, label="Original prompt", visible=True)
203
- prompt_input0_en = gr.Textbox(lines=1, label="Original prompt", visible=False)
204
- prompt_radio = gr.Radio(["Original prompt", "Extend prompt"], elem_id="prompt_radio",value="Extend prompt", show_label=False)
205
- # with gr.Row(elem_id="prompt_row1"):
 
 
 
206
  with gr.Column(id="input_col2"):
207
- prompt_input1 = gr.Textbox(lines=2, label="Extend prompt", visible=False)
208
- prompt_input2 = gr.Textbox(lines=2, label="Extend prompt", visible=True)
209
- with gr.Row():
210
- submit_btn = gr.Button(value = "submit",elem_id="submit-btn").style(
211
- margin=True,
212
- rounded=(True, True, True, True),
213
- )
214
  with gr.Row(elem_id='tab_demo', visible=True).style(height=200):
215
  tab_demo = gr.TabbedInterface(tab_actions, tab_titles)
216
  with gr.Row():
217
  gr.HTML(f"<p>{thanks_info}</p>")
218
 
219
- # prompt_radio.change(fn=change_prompt_radio, inputs=[prompt_radio], outputs=[prompt_input2])
220
- submit_btn.click(fn=extend_prompt, inputs=[prompt_input0], outputs=[prompt_input0_en, prompt_input1, prompt_input2])
221
 
222
  demo.launch()
 
12
  except Exception as e:
13
  return text
14
 
15
+ extend_prompt_pipe = pipeline('text-generation', model='yizhangliu/prompt-extend', max_length=77, pad_token_id=0)
16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  examples = [
18
  ['elon musk as thor'],
19
  ["giant dragon flying in the sky"],
 
106
  tabitems[i].childNodes[0].children[0].style.display='none';
107
  tabitems[i].childNodes[0].children[1].style.display='none';
108
  tabitems[i].childNodes[0].children[2].children[0].style.display='none';
109
+ tabitems[i].childNodes[0].children[3].style.display='none';
110
+ }
 
 
111
  }
112
 
113
  tab_demo = window['gradioEl'].querySelectorAll('#tab_demo')[0];
 
115
  tab_demo.setAttribute('style', 'height: 100%;');
116
  const page1 = window['gradioEl'].querySelectorAll('#page_1')[0];
117
  const page2 = window['gradioEl'].querySelectorAll('#page_2')[0];
118
+ window['gradioEl'].querySelector('#input_col1_row2').children[0].setAttribute('style', 'min-width:0px;width:50%;');
119
+ window['gradioEl'].querySelector('#input_col1_row2').children[1].setAttribute('style', 'min-width:0px;width:50%;');
 
120
  page1.style.display = "none";
121
  page2.style.display = "block";
122
  window['prevPrompt'] = '';
123
  window['doCheckPrompt'] = 0;
124
  window['checkPrompt'] = function checkPrompt() {
125
  try {
126
+ text_value = window['gradioEl'].querySelectorAll('#prompt_work')[0].querySelectorAll('textarea')[0].value
 
 
 
 
 
 
 
 
127
  progress_bar = window['gradioEl'].querySelectorAll('.progress-bar');
128
  if (window['doCheckPrompt'] === 0 && window['prevPrompt'] !== text_value && progress_bar.length == 0) {
129
  console.log('_____new prompt___[' + text_value + ']_');
 
159
  return false;
160
  }"""
161
 
162
+ def prompt_extend(prompt):
163
+ prompt_en = getTextTrans(prompt, source='zh', target='en')
164
+ extend_prompt_en = extend_prompt_pipe(prompt_en+',', num_return_sequences=1)[0]["generated_text"]
165
+ if (prompt != prompt_en):
166
+ logger.info(f"extend_prompt__1__")
167
+ extend_prompt_out = getTextTrans(extend_prompt_en, source='en', target='zh')
168
+ else:
169
+ logger.info(f"extend_prompt__2__")
170
+ extend_prompt_out = extend_prompt_en
171
+
172
+ return extend_prompt_out
173
+
174
+ def prompt_draw(prompt):
175
+ prompt_en = getTextTrans(prompt, source='zh', target='en')
176
+ if (prompt != prompt_en):
177
+ logger.info(f"draw_prompt______1__")
178
+ else:
179
+ logger.info(f"draw_prompt______2__")
180
+ return prompt_en
181
+
182
  with gr.Blocks(title='Text-to-Image') as demo:
183
  with gr.Group(elem_id="page_1", visible=True) as page_1:
184
  with gr.Box():
 
189
  with gr.Group(elem_id="page_2", visible=False) as page_2:
190
  with gr.Row(elem_id="prompt_row0"):
191
  with gr.Column(id="input_col1"):
192
+ with gr.Row(elem_id="input_col1_row1"):
193
+ prompt_input0 = gr.Textbox(lines=2, label="Original prompt", visible=True)
194
+ with gr.Row(elem_id="input_col1_row2"):
195
+ with gr.Column(elem_id="input_col1_row2_col1"):
196
+ draw_btn_0 = gr.Button(value = "Generate(original)", elem_id="draw-btn-0")
197
+ with gr.Column(elem_id="input_col1_row2_col2"):
198
+ extend_btn = gr.Button(value = "Extend prompt",elem_id="extend-btn")
199
  with gr.Column(id="input_col2"):
200
+ prompt_input1 = gr.Textbox(lines=2, label="Extend prompt", visible=True)
201
+ draw_btn_1 = gr.Button(value = "Generate(extend)", elem_id="draw-btn-1")
202
+ prompt_work = gr.Textbox(lines=1, label="prompt_work", elem_id="prompt_work", visible=False)
203
+ extend_btn.click(fn=prompt_extend, inputs=[prompt_input0], outputs=[prompt_input1])
204
+ draw_btn_0.click(fn=prompt_draw, inputs=[prompt_input0], outputs=[prompt_work])
205
+ draw_btn_1.click(fn=prompt_draw, inputs=[prompt_input1], outputs=[prompt_work])
 
206
  with gr.Row(elem_id='tab_demo', visible=True).style(height=200):
207
  tab_demo = gr.TabbedInterface(tab_actions, tab_titles)
208
  with gr.Row():
209
  gr.HTML(f"<p>{thanks_info}</p>")
210
 
211
+ # submit_btn.click(fn=extend_prompt, inputs=[prompt_input0], outputs=[prompt_input0_en, prompt_input1, prompt_input2])
 
212
 
213
  demo.launch()