Update app.py

app.py

@@ -97,32 +97,65 @@ header = """
 
 
 def create_app():
+    # with gr.Blocks() as app:
+    #     gr.Markdown(header, elem_id="header")
+    #     # with gr.Tabs():
+    #     #     with gr.Tab("Subject-driven"):
+    #     gr.Interface(
+    #         fn=process_image_and_text,
+    #         inputs=[
+    #             gr.Image(type="pil", label="Condition Image", width=300, elem_id="input"),
+    #             gr.Radio(
+    #                 [("512", 512), ("1024(beta)", 1024)],
+    #                 label="Resolution",
+    #                 value=512,
+    #                 elem_id="resolution",
+    #             ),
+    #             # gr.Slider(4, 16, 4, step=4, label="Inference Steps"),
+    #             gr.Textbox(lines=2, label="Text Prompt", elem_id="text"),
+    #         ],
+    #         outputs=gr.Image(type="pil", elem_id="output"),
+    #         examples=get_samples(),
+    #     )
+    #         # with gr.Tab("Fill"):
+    #         #     gr.Markdown("Coming soon")
+    #         # with gr.Tab("Canny"):
+    #         #     gr.Markdown("Coming soon")
+    #         # with gr.Tab("Depth"):
+    #         #     gr.Markdown("Coming soon")
+    
     with gr.Blocks() as app:
         gr.Markdown(header, elem_id="header")
-        # with gr.Tabs():
-        #     with gr.Tab("Subject-driven"):
-        gr.Interface(
-            fn=process_image_and_text,
-            inputs=[
-                gr.Image(type="pil", label="Condition Image", width=300, elem_id="input"),
-                gr.Radio(
+        with gr.Row(equal_height=False):
+            with gr.Column(variant="panel", elem_classes="inputPanel"):
+                original_image = gr.Image(
+                    type="pil", label="Condition Image", width=300, elem_id="input"
+                )
+                resolution = gr.Radio(
                     [("512", 512), ("1024(beta)", 1024)],
                     label="Resolution",
                     value=512,
                     elem_id="resolution",
-                ),
-                # gr.Slider(4, 16, 4, step=4, label="Inference Steps"),
-                gr.Textbox(lines=2, label="Text Prompt", elem_id="text"),
-            ],
-            outputs=gr.Image(type="pil", elem_id="output"),
-            examples=get_samples(),
+                )
+                text = gr.Textbox(lines=2, label="Text Prompt", elem_id="text")
+                submit_btn = gr.Button("Run", elem_id="submit_btn")
+
+            with gr.Column(variant="panel", elem_classes="outputPanel"):
+                output_image = gr.Image(type="pil", elem_id="output")
+
+        with gr.Row():
+            examples = gr.Examples(
+                examples=get_samples(),
+                inputs=[original_image, resolution, text],
+                label="Examples",
+            )
+
+        submit_btn.click(
+            fn=process_image_and_text,
+            inputs=[original_image, resolution, text],
+            outputs=output_image,
         )
-            # with gr.Tab("Fill"):
-            #     gr.Markdown("Coming soon")
-            # with gr.Tab("Canny"):
-            #     gr.Markdown("Coming soon")
-            # with gr.Tab("Depth"):
-            #     gr.Markdown("Coming soon")
+        
     return app