Update src/gradio_s3diff.py

src/gradio_s3diff.py

@@ -139,14 +139,17 @@ MARKDOWN = \
 If S3Diff is helpful for you, please help star the GitHub Repo. Thanks!
 """
 
+# Gradioアプリの作成
 block = gr.Blocks().queue()
 with block:
     with gr.Row():
         gr.Markdown(MARKDOWN)
+    with gr.Row():
+        gr.LoginButton()  # サインインボタンを追加
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(type="pil")
-            run_button = gr.Button("Run")  # `label="Run"` を削除してボタンのテキストを直接渡す
+            run_button = gr.Button("Run")
 
             with gr.Accordion("Options", open=True):
                 cfg_scale = gr.Slider(label="Classifier Free Guidance Scale (Set a value larger than 1 to enable it!)", minimum=1.0, maximum=1.1, value=1.07, step=0.01)
@@ -155,9 +158,14 @@ with block:
                 latent_tiled_overlap = gr.Slider(label="Tile Overlap", minimum=16, maximum=48, value=32, step=1)
                 align_method = gr.Dropdown(label="Color Correction", choices=["wavelet", "adain", "no fix"], value="wavelet")
         with gr.Column():
-            #result_image = gr.Image(label="Output", show_label=False, elem_id="result_image", source="canvas", width="100%", height="auto")
             result_image = gr.Image(label="Output", show_label=False, elem_id="result_image", width="100%", height="auto")
 
+    # サインイン後に実行する関数や出力表示の設定
+    with gr.Row():
+        m1 = gr.Markdown()
+        m2 = gr.Markdown()
+        block.load(lambda profile: f"Hello {profile.name}" if profile else "Please log in", inputs=None, outputs=m1)
+        block.load(lambda oauth_token: f"You belong to {', '.join([org['name'] for org in whoami(oauth_token.token)['orgs']])}" if oauth_token else "Log in to view organizations", inputs=None, outputs=m2)
 
     inputs = [
         input_image,