THEODOROS commited on
Commit
9adedb7
1 Parent(s): 6b8fb62

add placeholders for gradio

Browse files
Files changed (4) hide show
  1. app.py +73 -4
  2. empty.png +0 -0
  3. legend.png +0 -0
  4. thumbnail_gradio.PNG +0 -0
app.py CHANGED
@@ -86,7 +86,7 @@ def prompt_to_layout(user_prompt, fpath=None):
86
 
87
  _, im = draw_polygons(geom, colors, fpath=fpath)
88
 
89
- legend = Image.open(r"C:\\Users\\user\\Desktop\\legend3.png")
90
 
91
  im = np.array(im)
92
  im[:40, :] = np.array(legend)
@@ -111,7 +111,7 @@ def mut_txt2layout(mut_output):
111
  colors = [housegan_colors[housegan_labels[space]] for space in spaces]
112
  _, im = draw_polygons(geom, colors, fpath=None)
113
 
114
- legend = Image.open(r"C:\\Users\\user\\Desktop\\legend3.png")
115
 
116
  im = np.array(im)
117
  im[:40, :] = np.array(legend)
@@ -138,13 +138,81 @@ def prompt_with_mutation(user_prompt, fpath=None):
138
 
139
  # Gradio App
140
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
141
  def gen_and_mutate(user_prompt, mutate=False):
142
  if(mutate):
143
  im, mut_im = None, None
144
  while (mut_im is None):
145
  im, mut_im, output, mut_output = prompt_with_mutation(user_prompt)
146
  else:
147
- mut_im=Image.open(r"C:\\Users\\user\\Desktop\\empty.png")
148
  im, _ = prompt_to_layout(user_prompt)
149
 
150
  return im, mut_im
@@ -156,7 +224,8 @@ generated = gr.outputs.Image(label='Generated Layout')
156
  mutated = gr.outputs.Image(label='Mutated Layout')
157
 
158
  iface = gr.Interface(fn=gen_and_mutate, inputs=[textbox, checkbox], outputs=[generated, mutated],
159
- thumbnail=r"E:\Datasets\MyFloorplans\text2text\thumbnail_gradio.PNG",
 
160
  description='Demo of Semantic Generation of Residential Layouts \n',
161
  article='''<div>
162
  <p> This app allows users the use of natural language prompts for appartment layout generation, using a variety of semantic information:</p>
 
86
 
87
  _, im = draw_polygons(geom, colors, fpath=fpath)
88
 
89
+ legend = Image.open("legend.png")
90
 
91
  im = np.array(im)
92
  im[:40, :] = np.array(legend)
 
111
  colors = [housegan_colors[housegan_labels[space]] for space in spaces]
112
  _, im = draw_polygons(geom, colors, fpath=None)
113
 
114
+ legend = Image.open("legend.png")
115
 
116
  im = np.array(im)
117
  im[:40, :] = np.array(legend)
 
138
 
139
  # Gradio App
140
 
141
+ custom_css="""
142
+ @import url("https://use.typekit.net/nid3pfr.css");
143
+ .gradio_page {
144
+ display: flex;
145
+ width: 100vw;
146
+ height: 100vh;
147
+ flex-direction: column;
148
+ justify-content: center;
149
+ align-items: center;
150
+ margin: 0px;
151
+ max-width: 100vw;
152
+ background: #FFFFFF;
153
+ }
154
+ .gradio_interface {
155
+ width: 100vw;
156
+ max-width: 1500px;
157
+ }
158
+ .gradio_interface[theme=default] .panel_buttons {
159
+ justify-content: flex-end;
160
+ }
161
+ .gradio_interface[theme=default] .panel_button {
162
+ flex: 0 0 0;
163
+ min-width: 150px;
164
+ }
165
+ .gradio_interface[theme=default] .panel_button.submit {
166
+ background: #11213A;
167
+ border-radius: 5px;
168
+ color: #FFFFFF;
169
+ text-transform: uppercase;
170
+ min-width: 150px;
171
+ height: 4em;
172
+ letter-spacing: 0.15em;
173
+ flex: 0 0 0;
174
+ }
175
+ .gradio_interface[theme=default] .panel_button.submit:hover {
176
+ background: #000000;
177
+ }
178
+ .input_text {
179
+ font: 200 50px garamond-premier-pro-display, serif;
180
+ line-height: 115%;
181
+ color: #11213A;
182
+ border-radius: 0px;
183
+ border: 3px solid #11213A;
184
+ }
185
+ .input_text:focus {
186
+ border-color: #FA7880;
187
+ }
188
+ .gradio_interface[theme=default] .input_text input,
189
+ .gradio_interface[theme=default] .input_text textarea {
190
+ padding: 30px;
191
+ }
192
+ .input_text textarea:focus-visible {
193
+ outline: none;
194
+ }
195
+ .panel:nth-child(1) {
196
+ margin-left: 50px;
197
+ margin-right: 50px;
198
+ margin-top: 80px;
199
+ margin-bottom: 80px;
200
+ max-width: 750px;
201
+ }
202
+ .panel:nth-child(2) {
203
+ background: #D3ECF5;
204
+ }
205
+ .gradio_interface[theme=default] .component_set {
206
+ background: transparent;
207
+ }"""
208
+
209
  def gen_and_mutate(user_prompt, mutate=False):
210
  if(mutate):
211
  im, mut_im = None, None
212
  while (mut_im is None):
213
  im, mut_im, output, mut_output = prompt_with_mutation(user_prompt)
214
  else:
215
+ mut_im=Image.open("empty.png")
216
  im, _ = prompt_to_layout(user_prompt)
217
 
218
  return im, mut_im
 
224
  mutated = gr.outputs.Image(label='Mutated Layout')
225
 
226
  iface = gr.Interface(fn=gen_and_mutate, inputs=[textbox, checkbox], outputs=[generated, mutated],
227
+ css=custom_css,
228
+ thumbnail="thumbnail_gradio.PNG",
229
  description='Demo of Semantic Generation of Residential Layouts \n',
230
  article='''<div>
231
  <p> This app allows users the use of natural language prompts for appartment layout generation, using a variety of semantic information:</p>
empty.png ADDED
legend.png ADDED
thumbnail_gradio.PNG ADDED