Upload 3 files

app.py

@@ -9,7 +9,7 @@ css = """
 with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css, delete_cache=(60, 3600)) as demo:
     with gr.Column():
         gr.Markdown("# CivitAI to HF🤗 Downloader & Uploader", elem_classes="title")
-        with gr.Accordion("Search Civitai", open=False):
+        with gr.Accordion("Search Civitai", open=True):
             with gr.Row():
                 search_civitai_type = gr.CheckboxGroup(label="Type", choices=CIVITAI_TYPE, value=["Checkpoint", "LORA"])
                 search_civitai_basemodel = gr.CheckboxGroup(label="Base model", choices=CIVITAI_BASEMODEL, value=[])
@@ -23,7 +23,7 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css, delete_ca
             with gr.Row():
                 search_civitai_json = gr.JSON(value={}, visible=False)
                 search_civitai_desc = gr.Markdown(value="", visible=False)
-            search_civitai_result = gr.Dropdown(label="Search Results", choices=[("", "")], value="", allow_custom_value=True, visible=False)
+            search_civitai_result = gr.Dropdown(label="Search Results", choices=[("", "")], value="", allow_custom_value=True, visible=False, multiselect=True)
             search_civitai_add = gr.Button("Add to download URLs")
         dl_url = gr.Textbox(label="Download URL(s)", placeholder="https://civitai.com/api/download/models/28907\n...", value="", lines=2, max_lines=255)
         civitai_key = gr.Textbox(label="Your Civitai Key", value="", max_lines=1)

civitai_to_hf.py

@@ -6,7 +6,7 @@ import gc
 import requests
 from requests.adapters import HTTPAdapter
 from urllib3.util import Retry
-from utils import get_token, set_token, is_repo_exists, get_user_agent, get_download_file
+from utils import get_token, set_token, is_repo_exists, get_user_agent, get_download_file, list_uniq
 import re
 
 
@@ -19,6 +19,14 @@ def parse_urls(s):
         return []
 
 
+def to_urls(l: list[str]):
+    return "\n".join(l)
+
+
+def uniq_urls(s):
+    return to_urls(list_uniq(parse_urls(s)))
+
+
 def upload_safetensors_to_repo(filename, repo_id, repo_type, is_private, progress=gr.Progress(track_tqdm=True)):
     output_filename = Path(filename).name
     hf_token = get_token()
@@ -135,14 +143,18 @@ def search_civitai(query, types, base_model=[], sort=CIVITAI_SORT[0], period=CIV
           gr.update(visible=True), gr.update(visible=True)
 
 
-def select_civitai_item(search_result):
-    if not "http" in search_result: return gr.update(value="None", visible=True)
-    result = civitai_last_results.get(search_result, "None")
+def select_civitai_item(results: list[str]):
+    if "http" not in "".join(results): return gr.update(value="None", visible=True)
+    result = civitai_last_results.get(results[-1], "None")
     md = result['md'] if result else ""
     return gr.update(value=md, visible=True)
 
 
-def add_civitai_item(search_result, dl_url):
-    if not "http" in search_result: return gr.update(value=dl_url)
-    new_url = f"{dl_url}\n{search_result}" if dl_url else search_result
+def add_civitai_item(results: list[str], dl_url: str):
+    if "http" not in "".join(results): return gr.update(value=dl_url)
+    new_url = dl_url if dl_url else ""
+    for result in results:
+        if "http" not in result: continue
+        new_url += f"\n{result}" if new_url else f"{result}"
+    new_url = uniq_urls(new_url)
     return gr.update(value=new_url)

utils.py

@@ -63,6 +63,10 @@ def get_model_type(repo_id: str):
     return default
 
 
+def list_uniq(l):
+        return sorted(set(l), key=l.index)
+
+
 def list_sub(a, b):
     return [e for e in a if e not in b]