22

app.py

@@ -88,94 +88,92 @@ with gr.Blocks() as demo:
     with gr.Column(elem_id="col-container"):
         with gr.Column():
             gr.HTML(read_file("static/header.html"))
-        # with gr.Row():
-        #     with gr.Column():
-        #         prompt = gr.Textbox(
-        #             label="Prompt",
-        #             show_label=False,
-        #             lines=2,
-        #             placeholder="Enter your prompt",
-        #             # container=False,
-        #         )
-        #         is_polish_prompt = gr.Checkbox(label="Polish prompt", value=False)
-        #         run_button = gr.Button("Generate", variant="primary")
-        #         with gr.Accordion("Advanced Settings", open=False):
+        with gr.Row():
+            with gr.Column():
+                prompt = gr.Textbox(
+                    label="Prompt",
+                    show_label=False,
+                    lines=2,
+                    placeholder="Enter your prompt",
+                    # container=False,
+                )
+                is_polish_prompt = gr.Checkbox(label="Polish prompt", value=False)
+                run_button = gr.Button("Generate", variant="primary")
+                with gr.Accordion("Advanced Settings", open=False):
                     
-        #             negative_prompt = gr.Textbox(
-        #                 label="Negative prompt",
-        #                 lines=2,
-        #                 container=False,
-        #                 placeholder="Enter your negative prompt",
-        #                 value="blurry ugly bad"
-        #             )
-        #             num_inference_steps = gr.Slider(
-        #                 label="Steps",
-        #                 minimum=1,
-        #                 maximum=50,
-        #                 step=1,
-        #                 value=20,
-        #             )
-        #             with gr.Row():
-        #                 width = gr.Slider(
-        #                     label="Width",
-        #                     minimum=512,
-        #                     maximum=1280,
-        #                     step=32,
-        #                     value=768, 
-        #                 )
-
-        #                 height = gr.Slider(
-        #                     label="Height",
-        #                     minimum=512,
-        #                     maximum=1280,
-        #                     step=32,
-        #                     value=1024,
-        #                 )
-        #             with gr.Row():
-        #                 seed = gr.Slider(
-        #                     label="Seed",
-        #                     minimum=0,
-        #                     maximum=MAX_SEED,
-        #                     step=1,
-        #                     value=42,
-        #                 )
-        #                 guidance_scale = gr.Slider(
-        #                     label="Guidance scale",
-        #                     minimum=0.0,
-        #                     maximum=10.0,
-        #                     step=0.1,
-        #                     value=1.0,
-        #                 )
+                    negative_prompt = gr.Textbox(
+                        label="Negative prompt",
+                        lines=2,
+                        container=False,
+                        placeholder="Enter your negative prompt",
+                        value="blurry ugly bad"
+                    )
+                    num_inference_steps = gr.Slider(
+                        label="Steps",
+                        minimum=1,
+                        maximum=50,
+                        step=1,
+                        value=20,
+                    )
+                    with gr.Row():
+                        width = gr.Slider(
+                            label="Width",
+                            minimum=512,
+                            maximum=1280,
+                            step=32,
+                            value=768, 
+                        )
+
+                        height = gr.Slider(
+                            label="Height",
+                            minimum=512,
+                            maximum=1280,
+                            step=32,
+                            value=1024,
+                        )
+                    with gr.Row():
+                        seed = gr.Slider(
+                            label="Seed",
+                            minimum=0,
+                            maximum=MAX_SEED,
+                            step=1,
+                            value=42,
+                        )
+                        guidance_scale = gr.Slider(
+                            label="Guidance scale",
+                            minimum=0.0,
+                            maximum=10.0,
+                            step=0.1,
+                            value=1.0,
+                        )
 
                     
-        #             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                    randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
-        #     with gr.Column():
-        #         output_image = gr.Image(label="Generated image", show_label=False)
-        #         polished_prompt = gr.Textbox(label="Final prompt",lines=2, interactive=False)
+            with gr.Column():
+                output_image = gr.Image(label="Generated image", show_label=False)
+                polished_prompt = gr.Textbox(label="Final prompt",lines=2, interactive=False)
 
                     
-        # gr.Examples(examples=examples, inputs=[prompt])
-        # gr.Markdown(read_file("static/footer.md"))
-
-    # run_button.click(
-    #     fn=prepare,
-    #     inputs=[prompt, is_polish_prompt],
-    #     outputs=[polished_prompt, is_polish_prompt]
-    # ).then(
-    #     fn=inference,
-    #     inputs=[
-    #         polished_prompt,
-    #         negative_prompt,
-    #         width,
-    #         height,
-    #         seed,
-    #         randomize_seed,
-    #         guidance_scale,
-    #         num_inference_steps,
-    #     ],
-    #     outputs=[output_image, seed],
-    # )
+        gr.Examples(examples=examples, inputs=[prompt])
+        gr.Markdown(read_file("static/footer.md"))
+
+    run_button.click(
+        fn=inference,
+        inputs=[
+            prompt,
+            negative_prompt,
+            width,
+            height,
+            seed,
+            randomize_seed,
+            guidance_scale,
+            num_inference_steps,
+        ],
+        outputs=[output_image, seed],
+    ).then(
+        
+    )
 
 
 if __name__ == "__main__":