Update app.py

app.py

@@ -160,141 +160,162 @@ with gr.Blocks(css=css) as demo:
                 </h3>
                 """
             )
-    with gr.Group():
-        with gr.Row():
-            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.Gallery(label="Result", elem_id="gallery", show_label=False)
-    with gr.Accordion("Advanced options", open=False):
-        with gr.Row():
-            use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
-            negative_prompt = gr.Text(
-                label="Negative prompt",
-                max_lines=1,
-                value = "deformed, distorted, disfigured, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, mutated hands and fingers, disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
-                visible=True,
+    
+    with gr.Tabs():
+        with gr.TabItem("Text to Image"):
+            with gr.Group():
+                with gr.Row():
+                    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.Gallery(label="Result", elem_id="gallery", show_label=False)
+            with gr.Accordion("Advanced options", open=False):
+                with gr.Row():
+                    use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
+                    negative_prompt = gr.Text(
+                        label="Negative prompt",
+                        max_lines=1,
+                        value = "deformed, distorted, disfigured, poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, mutated hands and fingers, disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, NSFW",
+                        visible=True,
+                    )
+                seed = gr.Slider(
+                    label="Seed",
+                    min=0,
+                    maximum=MAX_SEED,
+                    step=1,
+                    value=0,
+                )
+
+                steps = gr.Slider(
+                    label="Steps",
+                    min=0,
+                    maximum=60,
+                    step=1,
+                    value=25,
+                )
+                number_image = gr.Slider(
+                    label="Number of Images",
+                    min=1,
+                    maximum=4,
+                    step=1,
+                    value=1,
+                )
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                with gr.Row(visible=True):
+                    width = gr.Slider(
+                        label="Width",
+                        min=256,
+                        maximum=MAX_IMAGE_SIZE,
+                        step=32,
+                        value=1024,
+                    )
+                    height = gr.Slider(
+                        label="Height",
+                        minimum=256,
+                        maximum=MAX_IMAGE_SIZE,
+                        step=32,
+                        value=1024,
+                    )
+                with gr.Row():
+                    guidance_scale = gr.Slider(
+                        label="Guidance Scale",
+                        min=0.1,
+                        maximum=10,
+                        step=0.1,
+                        value=7.0,
+                    )
+            
+            gr.Examples(
+                examples=examples,
+                inputs=prompt,
+                outputs=[result],
+                fn=generate,
+                cache_examples=CACHE_EXAMPLES,
             )
-        seed = gr.Slider(
-            label="Seed",
-            minimum=0,
-            maximum=MAX_SEED,
-            step=1,
-            value=0,
-        )
 
-        steps = gr.Slider(
-            label="Steps",
-            minimum=0,
-            maximum=60,
-            step=1,
-            value=25,
-        )
-        number_image = gr.Slider(
-            label="Number of Images",
-            minimum=1,
-            maximum=4,
-            step=1,
-            value=1,
-        )
-        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-        with gr.Row(visible=True):
-            width = gr.Slider(
-                label="Width",
-                minimum=256,
-                maximum=MAX_IMAGE_SIZE,
-                step=32,
-                value=1024,
+            use_negative_prompt.change(
+                fn=lambda x: gr.update(visible=x),
+                inputs=use_negative_prompt,
+                outputs=negative_prompt,
+                api_name=False,
             )
-            height = gr.Slider(
-                label="Height",
-                minimum=256,
-                maximum=MAX_IMAGE_SIZE,
-                step=32,
-                value=1024,
+
+            gr.on(
+                triggers=[
+                    prompt.submit,
+                    negative_prompt.submit,
+                    run_button.click,
+                ],
+                fn=generate,
+                inputs=[
+                    prompt,
+                    negative_prompt,
+                    use_negative_prompt,
+                    seed,
+                    width,
+                    height,
+                    guidance_scale,
+                    randomize_seed,
+                    steps,
+                    number_image,
+                ],
+                outputs=[result],
+                api_name="run",
             )
-        with gr.Row():
-            guidance_scale = gr.Slider(
-                label="Guidance Scale",
-                minimum=0.1,
-                maximum=10,
-                step=0.1,
-                value=7.0,
+        
+        with gr.TabItem("Image to Image"):
+            with gr.Group():
+                with gr.Row():
+                    with gr.Column():
+                        init_image = gr.Image(label="Input Image", type="pil")
+                    with gr.Column():
+                        img2img_output = gr.Gallery(label="Result", elem_id="gallery", show_label=False)
+                strength = gr.Slider(label="Img2Img Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.8)
+            
+            gr.on(
+                triggers=[
+                    prompt.submit,
+                    negative_prompt.submit,
+                    run_button.click,
+                ],
+                fn=img2img_generate,
+                inputs=[
+                    prompt,
+                    init_image,
+                    negative_prompt,
+                    use_negative_prompt,
+                    seed,
+                    guidance_scale,
+                    randomize_seed,
+                    steps,
+                    strength,
+                    number_image,
+                ],
+                outputs=[img2img_output],
+                api_name="img2img_run",
             )
-
+    
     with gr.Group():
-        with gr.Row():
-            with gr.Column():
-                init_image = gr.Image(label="Input Image", type="pil")
-            with gr.Column():
-                img2img_output = gr.Gallery(label="Result", elem_id="gallery", show_label=False)
-        strength = gr.Slider(label="Img2Img Strength", minimum=0.0, maximum=1.0, step=0.01, value=0.8)
-
-    gr.Examples(
-        examples=examples,
-        inputs=prompt,
-        outputs=[result],
-        fn=generate,
-        cache_examples=CACHE_EXAMPLES,
-    )
-
-    use_negative_prompt.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=use_negative_prompt,
-        outputs=negative_prompt,
-        api_name=False,
-    )
-
-    gr.on(
-        triggers=[
-            prompt.submit,
-            negative_prompt.submit,
-            run_button.click,
-        ],
-        fn=generate,
-        inputs=[
-            prompt,
-            negative_prompt,
-            use_negative_prompt,
-            seed,
-            width,
-            height,
-            guidance_scale,
-            randomize_seed,
-            steps,
-            number_image,
-        ],
-        outputs=[result],
-        api_name="run",
-    )
-
-    gr.on(
-        triggers=[
-            prompt.submit,
-            negative_prompt.submit,
-            run_button.click,
-        ],
-        fn=img2img_generate,
-        inputs=[
-            prompt,
-            init_image,
-            negative_prompt,
-            use_negative_prompt,
-            seed,
-            guidance_scale,
-            randomize_seed,
-            steps,
-            strength,
-            number_image,
-        ],
-        outputs=[img2img_output],
-        api_name="img2img_run",
-    )
+        gr.HTML(
+            """
+            <p style='text-align: center'>
+            Generated results will be displayed here.
+            </p>
+            """
+        )
+        generated_table = gr.Table(
+            headers=["Prompt", "Seed", "Image"],
+            row_count=1,
+            col_count=(3, "fixed"),
+            type="gallery",
+            show_label=False,
+            elem_id="generated-table",
+        )
 
 if __name__ == "__main__":
     demo.queue().launch()