Upload 4 files

app.py

@@ -25,6 +25,8 @@ from stablepy import (
 import time
 #import urllib.parse
 
+print(os.getenv("SPACES_ZERO_GPU"))
+
 PREPROCESSOR_CONTROLNET = {
   "openpose": [
     "Openpose",
@@ -32,14 +34,14 @@ PREPROCESSOR_CONTROLNET = {
   ],
   "scribble": [
     "HED",
-    "Pidinet",
+    "PidiNet",
     "None",
   ],
   "softedge": [
-    "Pidinet",
+    "PidiNet",
     "HED",
     "HED safe",
-    "Pidinet safe",
+    "PidiNet safe",
     "None",
   ],
   "segmentation": [
@@ -324,18 +326,21 @@ def extract_parameters(input_string):
     input_string = input_string.replace("\n", "")
 
     if "Negative prompt:" not in input_string:
-        print("Negative prompt not detected")
-        parameters["prompt"] = input_string
-        return parameters
+        if "Steps:" in input_string:
+            input_string = input_string.replace("Steps:", "Negative prompt: Steps:")
+        else:
+            print("Invalid metadata")
+            parameters["prompt"] = input_string
+            return parameters
 
     parm = input_string.split("Negative prompt:")
-    parameters["prompt"] = parm[0]
+    parameters["prompt"] = parm[0].strip()
     if "Steps:" not in parm[1]:
         print("Steps not detected")
-        parameters["neg_prompt"] = parm[1]
+        parameters["neg_prompt"] = parm[1].strip()
         return parameters
     parm = parm[1].split("Steps:")
-    parameters["neg_prompt"] = parm[0]
+    parameters["neg_prompt"] = parm[0].strip()
     input_string = "Steps:" + parm[1]
 
     # Extracting Steps
@@ -394,7 +399,8 @@ class GuiSD:
             retain_task_model_in_cache=False,
             device="cpu",
         )
-        self.model.device = torch.device("cpu") #
+        self.model.load_beta_styles()
+        #self.model.device = torch.device("cpu") #
 
     def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
 
@@ -722,7 +728,7 @@ class GuiSD:
                 if msg_lora:
                     info_state += msg_lora
 
-                info_state = info_state + "<br>" + "GENERATION DATA:<br>" + "<br>-------<br>".join(metadata).replace("\n", "<br>") 
+                info_state = info_state + "<br>" + "GENERATION DATA:<br>" + metadata[0].replace("\n", "<br>") + "<br>-------<br>"
 
                 download_links = "<br>".join(
                     [
@@ -732,9 +738,9 @@ class GuiSD:
                 )
                 if save_generated_images:
                     info_state += f"<br>{download_links}"
-                
+
                 img = save_images(img, metadata)
-                
+
             yield img, info_state
 
 def update_task_options(model_name, task_name):
@@ -953,12 +959,12 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                         with gr.Row():
                             seed_gui = gr.Number(minimum=-1, maximum=2**32-1, value=-1, label="Seed")
                             pag_scale_gui = gr.Slider(minimum=0.0, maximum=10.0, step=0.1, value=0.0, label="PAG Scale")
-                            num_images_gui = gr.Slider(minimum=1, maximum=4, step=1, value=1, label="Images")
+                            num_images_gui = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Images")
                             clip_skip_gui = gr.Checkbox(value=False, label="Layer 2 Clip Skip")
                             free_u_gui = gr.Checkbox(value=False, label="FreeU")
                         with gr.Row():
                             sampler_gui = gr.Dropdown(label="Sampler", choices=scheduler_names, value="Euler a")
-                            vae_model_gui = gr.Dropdown(label="VAE Model", choices=vae_model_list)
+                            vae_model_gui = gr.Dropdown(label="VAE Model", choices=vae_model_list, value=vae_model_list[0])
                             prompt_s_options = [
                                 ("Compel format: (word)weight", "Compel"),
                                 ("Classic format: (word:weight)", "Classic"),
@@ -969,29 +975,35 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                             ]
                             prompt_syntax_gui = gr.Dropdown(label="Prompt Syntax", choices=prompt_s_options, value=prompt_s_options[1][1])
 
-                        with gr.Row(equal_height=False):
-                            def run_set_params_gui(base_prompt):
-                                valid_receptors = {  # default values
-                                    "prompt": gr.update(value=base_prompt),
-                                    "neg_prompt": gr.update(value=""),
-                                    "Steps": gr.update(value=30),
-                                    "width": gr.update(value=1024),
-                                    "height": gr.update(value=1024),
-                                    "Seed": gr.update(value=-1),
-                                    "Sampler": gr.update(value="Euler a"),
-                                    "scale": gr.update(value=7.5), # cfg
-                                    "skip": gr.update(value=True),
-                                }
-                                valid_keys = list(valid_receptors.keys())
-
-                                parameters = extract_parameters(base_prompt)
-                                for key, val in parameters.items():
-                                    # print(val)
-                                    if key in valid_keys:
+                    with gr.Row(equal_height=False):
+
+                        def run_set_params_gui(base_prompt, name_model):
+                            valid_receptors = {  # default values
+                                "prompt": gr.update(value=base_prompt),
+                                "neg_prompt": gr.update(value=""),
+                                "Steps": gr.update(value=30),
+                                "width": gr.update(value=1024),
+                                "height": gr.update(value=1024),
+                                "Seed": gr.update(value=-1),
+                                "Sampler": gr.update(value="Euler a"),
+                                "scale": gr.update(value=7.),  # cfg
+                                "skip": gr.update(value=True),
+                                "Model": gr.update(value=name_model),
+                            }
+                            valid_keys = list(valid_receptors.keys())
+
+                            parameters = extract_parameters(base_prompt)
+
+                            for key, val in parameters.items():
+                                # print(val)
+                                if key in valid_keys:
+                                    try:
                                         if key == "Sampler":
                                             if val not in scheduler_names:
                                                 continue
                                         elif key == "skip":
+                                            if "," in str(val):
+                                                val = val.replace(",", "")
                                             if int(val) >= 2:
                                                 val = True
                                         if key == "prompt":
@@ -999,43 +1011,52 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                                                 val = re.sub(r'<[^>]+>', '', val)
                                                 print("Removed LoRA written in the prompt")
                                         if key in ["prompt", "neg_prompt"]:
-                                            val = val.strip()
+                                            val = re.sub(r'\s+', ' ', re.sub(r',+', ',', val)).strip()
                                         if key in ["Steps", "width", "height", "Seed"]:
                                             val = int(val)
                                         if key == "scale":
                                             val = float(val)
+                                        if key == "Model":
+                                            filtered_models = [m for m in model_list if val in m]
+                                            if filtered_models:
+                                                val = filtered_models[0]
+                                            else:
+                                                val = name_model
                                         if key == "Seed":
                                             continue
                                         valid_receptors[key] = gr.update(value=val)
                                         # print(val, type(val))
                                         # print(valid_receptors)
-                                return [value for value in valid_receptors.values()]
-
-                            set_params_gui.click(
-                                run_set_params_gui, [prompt_gui],[
-                                    prompt_gui,
-                                    neg_prompt_gui,
-                                    steps_gui,
-                                    img_width_gui,
-                                    img_height_gui,
-                                    seed_gui,
-                                    sampler_gui,
-                                    cfg_gui,
-                                    clip_skip_gui,
-                                ],
-                            )
-                            
-                            def run_clear_prompt_gui():
-                                return gr.update(value=""), gr.update(value="")
-                            clear_prompt_gui.click(
-                                run_clear_prompt_gui, [], [prompt_gui, neg_prompt_gui]
-                            )
-
-                            def run_set_random_seed():
-                                return -1
-                            set_random_seed.click(
-                                run_set_random_seed, [], seed_gui
-                            )
+                                    except Exception as e:
+                                        print(str(e))
+                            return [value for value in valid_receptors.values()]
+
+                        set_params_gui.click(
+                            run_set_params_gui, [prompt_gui, model_name_gui], [
+                                prompt_gui,
+                                neg_prompt_gui,
+                                steps_gui,
+                                img_width_gui,
+                                img_height_gui,
+                                seed_gui,
+                                sampler_gui,
+                                cfg_gui,
+                                clip_skip_gui,
+                                model_name_gui,
+                            ],
+                        )
+
+                        def run_clear_prompt_gui():
+                            return gr.update(value=""), gr.update(value="")
+                        clear_prompt_gui.click(
+                            run_clear_prompt_gui, [], [prompt_gui, neg_prompt_gui]
+                        )
+
+                        def run_set_random_seed():
+                            return -1
+                        set_random_seed.click(
+                            run_set_random_seed, [], seed_gui
+                        )
 
                     with gr.Accordion("LoRA", open=False, visible=True) as menu_lora:
                         def lora_dropdown(label):
@@ -1227,7 +1248,7 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                                     mode_ip2 = gr.Dropdown(value="style", label="Mode", choices=MODE_IP_OPTIONS)
                                 scale_ip2 = gr.Slider(minimum=0., maximum=2., step=0.01, value=0.7, label="Scale")
 
-                        with gr.Accordion("T2I adapter", open=False, visible=True) as menu_t2i:
+                        with gr.Accordion("T2I adapter", open=False, visible=False) as menu_t2i:
                             t2i_adapter_preprocessor_gui = gr.Checkbox(value=True, label="T2i Adapter Preprocessor")
                             with gr.Row():
                                 adapter_conditioning_scale_gui = gr.Slider(minimum=0, maximum=5., step=0.1, value=1, label="Adapter Conditioning Scale")
@@ -1368,6 +1389,11 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                     cache_examples=False,
                     #elem_id="examples",
                 )
+                gr.Markdown(
+                    """### Resources
+                    - You can also try the image generator in Colab’s free tier, which provides free GPU [link](https://github.com/R3gm/SD_diffusers_interactive).
+                    """
+                )
 ## END MOD
 
         with gr.Tab("Inpaint mask maker", render=True):
@@ -1433,6 +1459,35 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                     return img_source, img_result
                 btn_send.click(send_img, [img_source, img_result], [image_control, image_mask_gui])
 
+        with gr.Tab("PNG Info"):
+            def extract_exif_data(image):
+                if image is None: return ""
+
+                try:
+                    metadata_keys = ['parameters', 'metadata', 'prompt', 'Comment']
+
+                    for key in metadata_keys:
+                        if key in image.info:
+                            return image.info[key]
+
+                    return str(image.info)
+
+                except Exception as e:
+                    return f"Error extracting metadata: {str(e)}"
+
+            with gr.Row():
+                with gr.Column():
+                    image_metadata = gr.Image(label="Image with metadata", type="pil", sources=["upload"])
+
+                with gr.Column():
+                    result_metadata = gr.Textbox(label="Metadata", show_label=True, show_copy_button=True, interactive=False, container=True, max_lines=99)
+
+                    image_metadata.change(
+                        fn=extract_exif_data,
+                        inputs=[image_metadata],
+                        outputs=[result_metadata],
+                    )
+
 ## BEGIN MOD
         interface_mode_gui.change(
             change_interface_mode,

env.py

@@ -7,6 +7,11 @@ hf_read_token = os.environ.get('HF_READ_TOKEN') # only use for private repo
 # - **List Models**
 load_diffusers_format_model = [
     'stabilityai/stable-diffusion-xl-base-1.0',
+    'John6666/blue-pencil-flux1-v021-fp8-flux',
+    'John6666/wai-ani-flux-v10forfp8-fp8-flux',
+    'John6666/xe-anime-flux-v04-fp8-flux',
+    'John6666/lyh-anime-flux-v2a1-fp8-flux',
+    'John6666/carnival-unchained-v10-fp8-flux',
     'cagliostrolab/animagine-xl-3.1',
     'John6666/epicrealism-xl-v8kiss-sdxl',
     'misri/epicrealismXL_v7FinalDestination',
@@ -22,15 +27,30 @@ load_diffusers_format_model = [
     'WhiteAiZ/autismmixSDXL_autismmixConfetti_diffusers',
     'kitty7779/ponyDiffusionV6XL',
     'GraydientPlatformAPI/aniverse-pony',
+    'John6666/ras-real-anime-screencap-v1-sdxl',
+    'John6666/duchaiten-pony-xl-no-score-v60-sdxl',
     'John6666/mistoon-anime-ponyalpha-sdxl',
+    'John6666/3x3x3mixxl-v2-sdxl',
+    'John6666/3x3x3mixxl-3dv01-sdxl',
     'John6666/ebara-mfcg-pony-mix-v12-sdxl',
     'John6666/t-ponynai3-v51-sdxl',
+    'John6666/t-ponynai3-v65-sdxl',
+    'John6666/prefect-pony-xl-v3-sdxl',
     'John6666/mala-anime-mix-nsfw-pony-xl-v5-sdxl',
     'John6666/wai-real-mix-v11-sdxl',
+    'John6666/wai-c-v6-sdxl',
+    'John6666/iniverse-mix-xl-sfwnsfw-pony-guofeng-v43-sdxl',
+    'John6666/photo-realistic-pony-v5-sdxl',
+    'John6666/pony-realism-v21main-sdxl',
+    'John6666/pony-realism-v22main-sdxl',
     'John6666/cyberrealistic-pony-v63-sdxl',
+    'John6666/cyberrealistic-pony-v64-sdxl',
     'GraydientPlatformAPI/realcartoon-pony-diffusion',
     'John6666/nova-anime-xl-pony-v5-sdxl',
     'John6666/autismmix-sdxl-autismmix-pony-sdxl',
+    'John6666/aimz-dream-real-pony-mix-v3-sdxl',
+    'John6666/duchaiten-pony-real-v11fix-sdxl',
+    'John6666/duchaiten-pony-real-v20-sdxl',
     'yodayo-ai/kivotos-xl-2.0',
     'yodayo-ai/holodayo-xl-2.1',
     'yodayo-ai/clandestine-xl-1.0',

requirements.txt

@@ -11,4 +11,4 @@ huggingface_hub
 translatepy
 timm
 rapidfuzz
-sentencepiece
+sentencepiece