add health check

app_base.py

@@ -139,6 +139,9 @@ def resize_image(image, target_size = 1024):
 
 def create_demo() -> gr.Blocks:
 
+    def health_check():
+        return "Healthy"
+
     with gr.Blocks() as demo:
         cropper = gr.State()
         with gr.Row():
@@ -160,6 +163,8 @@ def create_demo() -> gr.Blocks:
                 start_step = gr.Slider(minimum=1, maximum=100, value=15, step=1, label="Start Step")
                 filter = gr.Dropdown(choices=filter_names, label="Filter", value="NONE")
                 g_btn = gr.Button("Edit Image")
+                health_btn = gr.Button("Health Check", visible=False)
+                health_output = gr.Textbox(label="Health Check Output", visible=False)
                 with gr.Accordion("Advanced Options", open=False):
                     guidance_scale = gr.Slider(minimum=0, maximum=20, value=0, step=0.5, label="Guidance Scale")
                     seed = gr.Number(label="Seed", value=8)
@@ -190,4 +195,10 @@ def create_demo() -> gr.Blocks:
             outputs=[restored_image, download_path],
         )
 
+        health_btn.click(
+            fn=health_check,
+            inputs=[],
+            outputs=[health_output],
+        )
+
     return demo