Update app.py

app.py

@@ -181,20 +181,22 @@ def check_civit_link(profile: Optional[gr.OAuthProfile], url):
     info, _ = process_url(url, download_files=False)
     hf_username = extract_huggingface_username(info['creator'])
     if(not hf_username):
-        no_username_text = 'Oops, your CivitAI profile seems to not have information about your Hugging Face account. Please visit <a href="https://civitai.com/user/account">https://civitai.com/user/account</a> and include it there<br><img src="https://i.imgur.com/hCbo9uL.png" />'
-        return no_username_text, gr.update()
+        no_username_text = 'Oops, your CivitAI profile seems to not have information about your Hugging Face account. Please visit <a href="https://civitai.com/user/account">https://civitai.com/user/account</a> and include it there<br><img width="60%" src="https://i.imgur.com/hCbo9uL.png" />'
+        return no_username_text, gr.update(), gr.update(visible=True)
     if(profile.name != hf_username):
         unmatched_username_text = 'Oops, the Hugging Face account in your CivitAI profile seems to be different than the one your are using here. Please visit <a href="https://civitai.com/user/account">https://civitai.com/user/account</a> and update it there<br><img src="https://i.imgur.com/hCbo9uL.png" />'
-        return unmatched_username_text, gr.update()
+        return unmatched_username_text, gr.update(), gr.update(visible=True)
     else:
-        return '', gr.update(interactive=True)
+        return '', gr.update(interactive=True), gr.update(visible=False)
         
 def swap_fill(profile: Optional[gr.OAuthProfile]):
     if profile is None:
         return gr.update(visible=True), gr.update(visible=False)
     else:
         return gr.update(visible=False), gr.update(visible=True)
-    
+
+def show_output():
+    return gr.update(visible=True)
 css = '''
 #login {
     font-size: 0px;
@@ -227,7 +229,6 @@ with gr.Blocks(css=css) as demo:
                             label="CivitAI model URL",
                             info="URL of the CivitAI model, make sure it is a SDXL LoRA",
                         )
-                        #is_author = gr.Checkbox(label="Are you the model author?", info="If you are not the author, a disclaimer with information about the author and the CivitAI source will be added", value=False)
         submit_button_civit = gr.Button("Upload model to Hugging Face and submit")
         output = gr.Textbox(label="Output progress")
     with gr.Column(visible=False) as enabled_area:
@@ -236,14 +237,15 @@ with gr.Blocks(css=css) as demo:
                             label="CivitAI model URL",
                             info="URL of the CivitAI model, make sure it is a SDXL LoRA",
                         )
-                        #is_author = gr.Checkbox(label="Are you the model author?", info="If you are not the author, a disclaimer with information about the author and the CivitAI source will be added", value=False)
         instructions = gr.HTML("")
+        try_again_button = gr.Button("I have added my HF profile to my account", visible=False)
         submit_button_civit = gr.Button("Upload model to Hugging Face", interactive=False)
-        output = gr.Textbox(label="Output progress")
+        output = gr.Textbox(label="Output progress", visible=False)
    
     demo.load(fn=swap_fill, outputs=[disabled_area, enabled_area])
-    submit_source_civit.change(fn=check_civit_link, inputs=[submit_source_civit], outputs=[instructions, submit_button_civit])
-    submit_button_civit.click(fn=upload_civit_to_hf, inputs=[submit_source_civit], outputs=[output])
+    submit_source_civit.change(fn=check_civit_link, inputs=[submit_source_civit], outputs=[instructions, submit_button_civit, try_again_button])
+    try_again_button.change(fn=check_civit_link, inputs=[submit_source_civit], outputs=[instructions, submit_button_civit, try_again_button])
+    submit_button_civit.click(fn=show_output, inputs=[], outputs=[output]).then(fn=upload_civit_to_hf, inputs=[submit_source_civit], outputs=[output])
     
 demo.queue()
 demo.launch()