fix gradio UI

app.py

@@ -20,8 +20,8 @@ tokenizer = AutoTokenizer.from_pretrained('gpt2')
 housegan_labels = {"living_room": 1, "kitchen": 2, "bedroom": 3, "bathroom": 4, "missing": 5, "closet": 6, 
                          "balcony": 7, "corridor": 8, "dining_room": 9, "laundry_room": 10}
 
-housegan_colors = [[0, 0, 0], [197, 203, 159], [169, 89, 142], [0, 132, 66], [190, 0, 198], [255, 255, 255], 
-                   [6, 53, 17], [2, 54, 192], [132, 151, 246], [197, 203, 159], [6, 53, 17],]
+architext_colors = [[0, 0, 0], [249, 222, 182], [195, 209, 217], [250, 120, 128], [126, 202, 234], [190, 0, 198], [255, 255, 255], 
+                   [6, 53, 17], [17, 33, 58], [132, 151, 246], [197, 203, 159], [6, 53, 17],]
 
 regex = re.compile(".*?\((.*?)\)")
 
@@ -82,7 +82,7 @@ def prompt_to_layout(user_prompt, fpath=None):
     for poly in polygons:
         geom.append(Polygon(np.array(poly, dtype=int)))
         
-    colors = [housegan_colors[housegan_labels[space]] for space in spaces]
+    colors = [architext_colors[housegan_labels[space]] for space in spaces]
     
     _, im = draw_polygons(geom, colors, fpath=fpath)
     
@@ -108,7 +108,7 @@ def mut_txt2layout(mut_output):
     for poly in polygons:
         geom.append(Polygon(np.array(poly, dtype=int)))
 
-    colors = [housegan_colors[housegan_labels[space]] for space in spaces]
+    colors = [architext_colors[housegan_labels[space]] for space in spaces]
     _, im = draw_polygons(geom, colors, fpath=None)
 
     legend = Image.open("legend.png")
@@ -143,7 +143,7 @@ custom_css="""
 .gradio_page {
   display: flex;
   width: 100vw;
-  height: 100vh;
+  min-height: 100vh;
   flex-direction: column;
   justify-content: center;
   align-items: center;
@@ -175,6 +175,7 @@ custom_css="""
 .gradio_interface[theme=default] .panel_button.submit:hover { 
   background: #000000;
 }
+
 .input_text {
   font: 200 50px garamond-premier-pro-display, serif;
   line-height: 115%;
@@ -202,8 +203,12 @@ custom_css="""
 .panel:nth-child(2) {
   background: #D3ECF5;
 }
+.gradio_interface[theme=default] .output_image .image_preview_holder {
+  background: #D3ECF5;
+}
 .gradio_interface[theme=default] .component_set  {
   background: transparent;
+  opacity: 1 !important;
 }"""
 
 def gen_and_mutate(user_prompt, mutate=False):