Update app.py

app.py

@@ -44,16 +44,12 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
 
 def generate(prompt: str,
              negative_prompt: str = '',
-             prompt_2: str = '',
-             negative_prompt_2: str = '',
              use_negative_prompt: bool = False,
-             use_prompt_2: bool = False,
-             use_negative_prompt_2: bool = False,
              seed: int = 0,
              width: int = 1024,
              height: int = 1024,
              guidance_scale: float = 1.0,
-             num_inference_steps: int = 4,
+             num_inference_steps: int = 6,
              secret_token: str = '') -> PIL.Image.Image:
     if secret_token != SECRET_TOKEN:
         raise gr.Error(
@@ -63,15 +59,9 @@ def generate(prompt: str,
 
     if not use_negative_prompt:
         negative_prompt = None  # type: ignore
-    if not use_prompt_2:
-        prompt_2 = None  # type: ignore
-    if not use_negative_prompt_2:
-        negative_prompt_2 = None  # type: ignore
 
     return pipe(prompt=prompt,
                 negative_prompt=negative_prompt,
-                prompt_2=prompt_2,
-                negative_prompt_2=negative_prompt_2,
                 width=width,
                 height=height,
                 guidance_scale=guidance_scale,
@@ -79,108 +69,74 @@ def generate(prompt: str,
                 generator=generator,
                 output_type='pil').images[0]
 
-with gr.Blocks(css='style.css') as demo:
+with gr.Blocks() as demo:
     gr.Markdown(DESCRIPTION)
-    with gr.Box():
-        with gr.Row():
-            secret_token = gr.Text(
-                label='Secret Token',
-                max_lines=1,
-                placeholder='Enter your secret token',
-            )
-            prompt = gr.Text(
-                label='Prompt',
-                show_label=False,
-                max_lines=1,
-                placeholder='Enter your prompt',
-                container=False,
-            )
-            run_button = gr.Button('Run', scale=0)
-        result = gr.Image(label='Result', show_label=False)
-        with gr.Accordion('Advanced options', open=False):
-            with gr.Row():
-                use_negative_prompt = gr.Checkbox(label='Use negative prompt',
-                                                  value=False)
-                use_prompt_2 = gr.Checkbox(label='Use prompt 2', value=False)
-                use_negative_prompt_2 = gr.Checkbox(
-                    label='Use negative prompt 2', value=False)
-            negative_prompt = gr.Text(
-                label='Negative prompt',
-                max_lines=1,
-                placeholder='Enter a negative prompt',
-                visible=False,
-            )
-            prompt_2 = gr.Text(
-                label='Prompt 2',
-                max_lines=1,
-                placeholder='Enter your prompt',
-                visible=False,
-            )
-            negative_prompt_2 = gr.Text(
-                label='Negative prompt 2',
-                max_lines=1,
-                placeholder='Enter a negative prompt',
-                visible=False,
-            )
+    secret_token = gr.Text(
+        label='Secret Token',
+        max_lines=1,
+        placeholder='Enter your secret token',
+    )
+    prompt = gr.Text(
+        label='Prompt',
+        show_label=False,
+        max_lines=1,
+        placeholder='Enter your prompt',
+        container=False,
+    )
+    run_button = gr.Button('Run', scale=0)
+    result = gr.Image(label='Result', show_label=False)
 
-            seed = gr.Slider(label='Seed',
-                             minimum=0,
-                             maximum=MAX_SEED,
-                             step=1,
-                             value=0)
-            randomize_seed = gr.Checkbox(label='Randomize seed', value=True)
-            with gr.Row():
-                width = gr.Slider(
-                    label='Width',
-                    minimum=256,
-                    maximum=MAX_IMAGE_SIZE,
-                    step=32,
-                    value=1024,
-                )
-                height = gr.Slider(
-                    label='Height',
-                    minimum=256,
-                    maximum=MAX_IMAGE_SIZE,
-                    step=32,
-                    value=1024,
-                )
-            guidance_scale = gr.Slider(
-                label='Guidance scale',
-                minimum=1,
-                maximum=20,
-                step=0.1,
-                value=1.0)
-            num_inference_steps = gr.Slider(
-                label='Number of inference steps',
-                minimum=2,
-                maximum=8,
-                step=1,
-                value=4)
+    use_negative_prompt = gr.Checkbox(label='Use negative prompt', value=False)
+    negative_prompt = gr.Text(
+        label='Negative prompt',
+        max_lines=1,
+        placeholder='Enter a negative prompt',
+        visible=False,
+    )
+    seed = gr.Slider(label='Seed',
+                    minimum=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=0)
+    randomize_seed = gr.Checkbox(label='Randomize seed', value=True)
 
+    width = gr.Slider(
+        label='Width',
+        minimum=256,
+        maximum=MAX_IMAGE_SIZE,
+        step=32,
+        value=1024,
+    )
+    height = gr.Slider(
+        label='Height',
+        minimum=256,
+        maximum=MAX_IMAGE_SIZE,
+        step=32,
+        value=1024,
+    )
+    guidance_scale = gr.Slider(
+        label='Guidance scale',
+        minimum=1,
+        maximum=20,
+        step=0.1,
+        value=1.0)
+    num_inference_steps = gr.Slider(
+        label='Number of inference steps',
+        minimum=2,
+        maximum=8,
+        step=1,
+        value=4)
+    
     use_negative_prompt.change(
         fn=lambda x: gr.update(visible=x),
         inputs=use_negative_prompt,
         outputs=negative_prompt
     )
-    use_prompt_2.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=use_prompt_2,
-        outputs=prompt_2
-    )
-    use_negative_prompt_2.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=use_negative_prompt_2,
-        outputs=negative_prompt_2
-    )
 
     inputs = [
         prompt,
         negative_prompt,
-        prompt_2,
-        negative_prompt_2,
         use_negative_prompt,
-        use_prompt_2,
-        use_negative_prompt_2,
         seed,
         width,
         height,