Update app.py

app.py

@@ -104,9 +104,10 @@ def process_url(url, profile, do_download=True, folder="."):
     else:
         raise gr.Error("Something went wrong in fetching CivitAI API")
 
-def create_readme(info, downloaded_files, is_author=True, folder="."):
+def create_readme(info, downloaded_files, link_civit=False, is_author=True, folder="."):
     readme_content = ""
     original_url = f"https://civitai.com/models/{info['id']}"
+    link_civit_disclaimer = f'([CivitAI]({original_url})'
     non_author_disclaimer = f'This model was originally uploaded on [CivitAI]({original_url}), by [{info["creator"]}](https://civitai.com/user/{info["creator"]}/models). The information below was provided by the author on CivitAI:'
     default_tags = ["text-to-image", "stable-diffusion", "lora", "diffusers"]
     civit_tags = [t for t in info["tags"] if t not in default_tags]
@@ -124,7 +125,7 @@ widget:
 - text: {widget_prompts}
 ---
 
-# {info["name"]}
+# {info["name"]} {link_civit_disclaimer if link_civit else ''}
 
 {non_author_disclaimer if not is_author else ''}
 
@@ -203,15 +204,35 @@ def swap_fill(profile: Optional[gr.OAuthProfile]):
 
 def show_output():
     return gr.update(visible=True)
+
+def list_civit_models(username):
+    url = f"https://civitai.com/api/v1/models?username={username}&limit=100"
+    json_models_list = []
+
+    while url:
+        response = requests.get(url)
+        data = response.json()
+
+        # Add current page items to the list
+        json_models_list.extend(data.get('items', []))
+
+        # Check if there is a nextPage URL in the metadata
+        metadata = data.get('metadata', {})
+        url = metadata.get('nextPage', None)
+    urls = ""
+    for model in json_models_list:
+        urls += f'https://civitai.com/models/{model["id"]}/{slugify(model["name"])}\n'
     
-def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Progress(track_tqdm=True)):
+    return urls
+
+def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, link_civit=False, progress=gr.Progress(track_tqdm=True)):
     if not profile.name:
         return gr.Error("Are you sure you are logged in?")
     
     folder = str(uuid.uuid4())
     os.makedirs(folder, exist_ok=False)
     info, downloaded_files = process_url(url, profile, folder=folder)
-    create_readme(info, downloaded_files, folder=folder)
+    create_readme(info, downloaded_files, link_civit, folder=folder)
     try:
         api = HfApi(token=os.environ["HUGGING_FACE_HUB_TOKEN"])
         username = api.whoami()["name"]
@@ -252,6 +273,10 @@ def upload_civit_to_hf(profile: Optional[gr.OAuthProfile], url, progress=gr.Prog
     return f'''# Model uploaded to 🤗!
     ## Access it here [{user_repo_id}](https://huggingface.co/{user_repo_id}) '''
 
+def bulk_upload(profile: Optional[gr.OAuthProfile], urls, link_civit=False, progress=gr.Progress(track_tqdm=True)):
+    for url in urls.split("\n"):
+        if(url):
+            yield upload_civit_to_hf(profile, url, link_civit)
 
 css = '''
 #login {
@@ -294,11 +319,21 @@ Get diffusers compatibility, a free GPU-based Inference Widget and possibility t
                         )
         submit_button_civit = gr.Button("Upload model to Hugging Face and submit", interactive=False)
     with gr.Column(visible=False) as enabled_area:
-        with gr.Row():
-                        submit_source_civit = gr.Textbox(
-                            label="CivitAI model URL",
-                            info="URL of the CivitAI model, for now only SDXL LoRAs are supported",
-                        )
+        with gr.Column():
+            submit_source_civit = gr.Textbox(
+                label="CivitAI model URL",
+                info="URL of the CivitAI model, for now only SDXL LoRAs are supported",
+            )
+            with gr.Accordion("Advanced options", open=False):
+                civit_username_to_bulk = gr.Textbox(label="CivitAI username (optional)", info="Type your CivitAI username here to automagically fill the bulk models URLs list below (optional, you can paste links down here directly)")
+                submit_bulk_civit = gr.Textbox(
+                    label="CivitAI bulk models URLs",
+                    info="Add one URL per line",
+                    lines=6,
+                )
+                link_civit = gr.Checkbox(label="Link back to CivitAI?", value=False)
+                bulk_button = gr.Button("Bulk upload")
+                
         instructions = gr.HTML("")
         try_again_button = gr.Button("I have added my HF profile to my account (it may take 1 minute to refresh)", visible=False)
         submit_button_civit = gr.Button("Upload model to Hugging Face", interactive=False)
@@ -306,8 +341,10 @@ Get diffusers compatibility, a free GPU-based Inference Widget and possibility t
 
     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, try_again_button, submit_button_civit])
+    civit_username_to_bulk.change(fn=list_civit_models, inputs=[civit_username_to_bulk], outputs=[submit_bulk_civit])
     try_again_button.click(fn=check_civit_link, inputs=[submit_source_civit], outputs=[instructions, submit_button_civit, try_again_button, submit_button_civit])
-    submit_button_civit.click(fn=show_output, inputs=[], outputs=[output]).then(fn=upload_civit_to_hf, inputs=[submit_source_civit], outputs=[output])
+    submit_button_civit.click(fn=show_output, inputs=[], outputs=[output]).then(fn=upload_civit_to_hf, inputs=[submit_source_civit, link_civit], outputs=[output])
+    bulk_button.click(fn=show_output, inputs=[], outputs=[output]).then(fn=bulk_upload, inputs=[submit_bulk_civit, link_civit], outputs=[output])
     gr.LogoutButton(elem_id="logout")
 demo.queue()
 demo.launch()