Add password

app.py

@@ -434,9 +434,16 @@ css = r"""
 """
 
 
+def check_password(password: str) -> bool:
+    if password == PASSWORD:
+        return [gr.update(visible=False), gr.update(visible=True)]
+    return [gr.update(visible=True), gr.update(visible=False)]
+
+
 with gr.Blocks(css=css) as demo:
     gr.Markdown(intro_markdown)
-    with gr.Row():
+
+    with gr.Row(visible=False) as content:
         with gr.Column(elem_id="col-left"):
             gr.HTML(
                 """
@@ -518,11 +525,7 @@ with gr.Blocks(css=css) as demo:
             )
             run_button = gr.Button("Run")
             flag_button = gr.Button("Flag")
-            # Reset the results when the run button is clicked
-            run_button.click(
-                outputs=results,
-                fn=lambda: None,
-            )
+
             with gr.Accordion("Advanced Settings", open=False):
                 seed = gr.Slider(
                     label="Seed",
@@ -573,6 +576,15 @@ with gr.Blocks(css=css) as demo:
                         value=20,
                     )
 
+    with gr.Row():
+        password = gr.Textbox(label="Password", type="password")
+        submit = gr.Button("Submit")
+        submit.click(
+            fn=check_password,
+            inputs=[password],
+            outputs=[submit, content],
+        )
+
     # This needs to be called at some point prior to the first call to callback.flag()
     callback.setup(
         [
@@ -600,26 +612,6 @@ with gr.Blocks(css=css) as demo:
             examples_per_page=12,
         )
 
-    run_button.click(
-        fn=infer,
-        inputs=[
-            furniture_image_input,
-            room_image_input,
-            furniture_prompt,
-            seed,
-            randomize_seed,
-            guidance_scale,
-            num_inference_steps,
-            max_dimension,
-            num_images_per_prompt,
-        ],
-        outputs=[results, seed],
-    )
-    flag_button.click(
-        fn=flag,
-        inputs=[furniture_image_input, room_image_input, results],
-        preprocess=False,
-    )
 
 # demo.launch(auth=[(USER, PASSWORD)])
 demo.launch()