Update app.py

app.py

@@ -71,10 +71,26 @@ def Piper(_do, _dont):
 
 def infer(prompt1,prompt2,prompt3,prompt4):
     name = generate_random_string(12)+".png"
-    prompt1 = " and ".join([translate(v,"english").upper() for v in prompt1])
-    prompt2 = " and ".join([translate(v,"english").upper() for v in prompt2])
-    prompt3 = " and ".join([translate(v,"english").upper() for v in prompt3])
-    prompt4 = " or ".join([translate(v,"english").upper() for v in prompt4])
+    if prompt1 == None:
+        prompt1 = "any"
+    else:
+        prompt1 = " and ".join([translate(v,"english").upper() for v in prompt1])
+    
+    if prompt2 == None:
+        prompt2 = "any"
+    else:
+        prompt2 = " and ".join([translate(v,"english").upper() for v in prompt2])
+    
+    if prompt3 == None:
+        prompt3 = "any"
+    else:
+        prompt3 = " and ".join([translate(v,"english").upper() for v in prompt3])
+    
+    if prompt4 == None:
+        prompt4 = "none"
+    else:
+        prompt4 = " or ".join([translate(v,"english").upper() for v in prompt4])
+        
     _do = f'Show an authentic {prompt3} scene, while focusing on the details, of {prompt1}, as the main elements, and, showing {prompt2} in the background.'
     _dont = f'ANY usage of {prompt4}...'
     image = Piper(_do, _dont).images[0].save(name)
@@ -83,7 +99,7 @@ def infer(prompt1,prompt2,prompt3,prompt4):
 css="""
 #col-container {
     margin: 0 auto;
-    max-width: 15cm;
+    max-width: 13cm;
 }
 #image-container {
     aspect-ratio: 1 / 1;
@@ -95,7 +111,7 @@ if torch.cuda.is_available():
 else:
     power_device = "CPU"
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(theme=gr.themes.Glass(primary_hue="white", secondary_hue="black",spacing_size="xl",radius_size="xl"),css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
             # Image Generator
@@ -109,8 +125,7 @@ with gr.Blocks(css=css) as demo:
                     max_choices=3,
                     label="Foreground Elements",
                     show_label=True,
-                    container=False,
-                    value="any"
+                    container=True
                 )
             with gr.Row():
                 prompt2 = gr.Dropdown(
@@ -119,8 +134,7 @@ with gr.Blocks(css=css) as demo:
                     max_choices=4,
                     label="Background Elements",
                     show_label=True,
-                    container=False,
-                    value="any"
+                    container=True
                 )
             with gr.Row():
                 prompt3 = gr.Dropdown(
@@ -129,21 +143,19 @@ with gr.Blocks(css=css) as demo:
                     max_choices=2,
                     label="Background Events",
                     show_label=True,
-                    container=False,
-                    value="any"
+                    container=True
                 )
             with gr.Row():
                 prompt4 = gr.Dropdown(
                     multiselect=True,
                     allow_custom_value=True,
-                    max_choices=6,
+                    max_choices=5,
                     label="Forbidden Elements/Events",
                     show_label=True,
-                    container=False,
-                    value="None"
+                    container=True
                 )
             with gr.Row():
-                run_button = gr.Button("Run", scale=0)
+                run_button = gr.Button("Run")
         result = gr.Image(elem_id="image-container", label="Result", show_label=False, type='filepath')
     run_button.click(
         fn = infer,