Update app.py

app.py

@@ -23,8 +23,8 @@ MAX_MODEL_NUM = 300
 Yntec digplay
 '''
 
-hf_civital_image_info_dataset = load_dataset("svjack/hf_civital_image_info_v0", token = API_TOKEN)
-hf_civital_image_info_df = hf_civital_image_info_dataset["train"].to_pandas()
+hf_civitai_image_info_dataset = load_dataset("svjack/hf_civitai_image_info_v0", token = API_TOKEN)
+hf_civitai_image_info_df = hf_civitai_image_info_dataset["train"].to_pandas()
 
 def gen_interface(model_name, max_times = 3):
     times = 0
@@ -53,7 +53,7 @@ def jpg_val_to_img(jpg_bytes):
     return toImgPIL(img)
 
 
-model_list = hf_civital_image_info_df["hf_repo_id"].drop_duplicates().values.tolist()
+model_list = hf_civitai_image_info_df["hf_repo_id"].drop_duplicates().values.tolist()
 model_interface_list = []
 for model_name in tqdm(model_list):
     gr_model_interface = gen_interface(model_name)
@@ -63,7 +63,7 @@ for model_name in tqdm(model_list):
             break
 print("load model num : {}".format(len(model_interface_list)))
 
-def get_civital_iframe(url, width = 1400, height = 768, as_html = True, visible = False):
+def get_civitai_iframe(url, width = 1400, height = 768, as_html = True, visible = False):
     html= '''
             <div style="justify-content: center; display: flex;">
                     <iframe
@@ -79,23 +79,23 @@ def get_civital_iframe(url, width = 1400, height = 768, as_html = True, visible
     return html
 
 def get_info_by_interface(gr_interface, model_interface_list = model_interface_list):
-    #### out: (gr_interface, civital_url, civital_info)
+    #### out: (gr_interface, civitai_url, civitai_info)
     if hasattr(gr_interface, "app"):
-        civital_url = hf_civital_image_info_df[
-            hf_civital_image_info_df["hf_repo_id"] == gr_interface.title
-        ]["civital_url"].iloc[0]
-        civital_info = hf_civital_image_info_df[
-            hf_civital_image_info_df["hf_repo_id"] == gr_interface.title
+        civitai_url = hf_civitai_image_info_df[
+            hf_civitai_image_info_df["hf_repo_id"] == gr_interface.title
+        ]["civitai_url"].iloc[0]
+        civitai_info = hf_civitai_image_info_df[
+            hf_civitai_image_info_df["hf_repo_id"] == gr_interface.title
         ][["prompt", "image"]].values.tolist()
-        return gr_interface ,civital_url, civital_info
+        return gr_interface ,civitai_url, civitai_info
     else:
-        civital_url = hf_civital_image_info_df[
-            hf_civital_image_info_df["hf_repo_id"] == gr_interface
-        ]["civital_url"].iloc[0]
-        civital_info = hf_civital_image_info_df[
-            hf_civital_image_info_df["hf_repo_id"] == gr_interface
+        civitai_url = hf_civitai_image_info_df[
+            hf_civitai_image_info_df["hf_repo_id"] == gr_interface
+        ]["civitai_url"].iloc[0]
+        civitai_info = hf_civitai_image_info_df[
+            hf_civitai_image_info_df["hf_repo_id"] == gr_interface
         ][["prompt", "image"]].values.tolist()
-        return list(filter(lambda x:x.title == gr_interface, model_interface_list))[0] ,civital_url, civital_info
+        return list(filter(lambda x:x.title == gr_interface, model_interface_list))[0] ,civitai_url, civitai_info
 
 def read_image_from_url(url):
     response = requests.get(url)
@@ -131,9 +131,9 @@ def image_click(images, evt: gr.SelectData, gr_interface_value,
     return prompt
     #return prompt, im
 
-def try_repo_act_func(civital_url, show_civital_button):
-    repo_html_iframe_hide = get_civital_iframe(civital_url, visible = True if show_civital_button == "Show Civital Page" else False)
-    return repo_html_iframe_hide,  gr.Button("Hide Civital Page" if show_civital_button == "Show Civital Page" else "Show Civital Page")
+def try_repo_act_func(civitai_url, show_civitai_button):
+    repo_html_iframe_hide = get_civitai_iframe(civitai_url, visible = True if show_civitai_button == "Show Civitai Page" else False)
+    return repo_html_iframe_hide,  gr.Button("Hide Civitai Page" if show_civitai_button == "Show Civitai Page" else "Show Civitai Page")
 
 with gr.Blocks(
     css = '''
@@ -152,7 +152,7 @@ with gr.Blocks(
         '''
         <center>
         <div class="header">
-        <h1 class = "logo"> <img src="https://huggingface.co/spaces/svjack/Civital-Stable-Diffusion-HF/resolve/main/civital_logo.webp" alt="logo" />  🤗 Civital Model on Huggingface </h1>
+        <h1 class = "logo"> <img src="https://huggingface.co/spaces/svjack/Civitai-Stable-Diffusion-HF/resolve/main/civitai_logo.webp" alt="logo" />  🤗 Civitai Model on Huggingface </h1>
         </center>
         '''
     )
@@ -165,26 +165,26 @@ with gr.Blocks(
                     value=sorted(map(lambda x: x.title, model_interface_list))[0],)
             with gr.Column():
                 with gr.Row():
-                    civital_prompt = gr.Textbox(label = "🤭 Civital Prompt (Click from 👉 right gallery to get them, and You can edit ✍️ yourself) Don't use them to generate NSFW content, such as porns. 😬👎 Or you will get ❌👀",
+                    civitai_prompt = gr.Textbox(label = "🤭 Civitai Prompt (Click from 👉 right gallery to get them, and You can edit ✍️ yourself) Don't use them to generate NSFW content, such as porns. 😬👎 Or you will get ❌👀",
                         interactive = True,
                     )
                 gen_button = gr.Button(label = "Generate")
                 hf_image = gr.Image(label = "🤭 Image generate by 🤗 Huggingface", height = 768)
 
         with gr.Column():
-            civital_info_gallery = gr.Gallery(
+            civitai_info_gallery = gr.Gallery(
                 pd.Series(
                 get_info_by_interface(hf_model_dropdown.value)[2]
                 ).sample(n = min(len(get_info_by_interface(hf_model_dropdown.value)[2]), 30)).map(lambda t2: t2[1]).map(lambda x: x["bytes"]).map(jpg_val_to_img).values.tolist(),
                 height = 1024,
-                label = "🖱️👇 ➡️ 👈 Civital image samples",
+                label = "🖱️👇 ➡️ 👈 Civitai image samples",
                 object_fit = "contain"
             )
 
     with gr.Row():
         with gr.Column():
-            try_repo_button = gr.Button("Show Civital Page")
-            civital_iframe_html = get_civital_iframe(
+            try_repo_button = gr.Button("Show Civitai Page")
+            civitai_iframe_html = get_civitai_iframe(
                 get_info_by_interface(hf_model_dropdown.value)[1]
             )
 
@@ -193,22 +193,22 @@ with gr.Blocks(
         get_info_by_interface(x)[2]
         ).sample(n = min(len(get_info_by_interface(x)[2]), 30)).map(lambda t2: t2[1]).map(lambda x: x["bytes"]).map(jpg_val_to_img).values.tolist(),
         hf_model_dropdown,
-        civital_info_gallery
+        civitai_info_gallery
     )
     hf_model_dropdown.change(
-        lambda _: (gr.Button("Show Civital Page"), gr.HTML(visible = False)),
+        lambda _: (gr.Button("Show Civitai Page"), gr.HTML(visible = False)),
         None,
-        [try_repo_button, civital_iframe_html]
+        [try_repo_button, civitai_iframe_html]
     )
 
-    civital_info_gallery.select(
+    civitai_info_gallery.select(
             image_click,
-            [civital_info_gallery, hf_model_dropdown],
-            civital_prompt
+            [civitai_info_gallery, hf_model_dropdown],
+            civitai_prompt
     )
     gen_button.click(lambda hf_model_name, text_prompt:
             get_info_by_interface(hf_model_name)[0](text_prompt),
-            [hf_model_dropdown, civital_prompt],
+            [hf_model_dropdown, civitai_prompt],
             hf_image
     )
 
@@ -217,7 +217,7 @@ with gr.Blocks(
             get_info_by_interface(hf_model_name)[1]
             , button),
         [hf_model_dropdown, try_repo_button],
-        [civital_iframe_html, try_repo_button]
+        [civitai_iframe_html, try_repo_button]
     )
 
 demo.launch(show_api = False)