Upload 4 files

app.py

@@ -23,8 +23,10 @@ from stablepy import (
     SDXL_TASKS,
 )
 import time
+from PIL import ImageFile
 #import urllib.parse
 
+ImageFile.LOAD_TRUNCATED_IMAGES = True
 print(os.getenv("SPACES_ZERO_GPU"))
 
 PREPROCESSOR_CONTROLNET = {
@@ -399,7 +401,7 @@ class GuiSD:
             device="cpu",
         )
         self.model.load_beta_styles()
-        self.model.device = torch.device("cpu") #
+        #self.model.device = torch.device("cpu") #
 
     def load_new_model(self, model_name, vae_model, task, progress=gr.Progress(track_tqdm=True)):
 
@@ -758,12 +760,10 @@ def dynamic_gpu_duration(func, duration, *args):
 
     return wrapped_func()
 
-
 @spaces.GPU
 def dummy_gpu():
     return None
 
-
 def sd_gen_generate_pipeline(*args):
 
     gpu_duration_arg = int(args[-1]) if args[-1] else 59
@@ -831,6 +831,42 @@ def sd_gen_generate_pipeline(*args):
         gr.Info(msg_task_complete)
         print(msg_task_complete)
 
+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)}"
+
+@spaces.GPU(duration=20)
+def esrgan_upscale(image, upscaler_name, upscaler_size):
+    if image is None: return None
+
+    from stablepy.diffusers_vanilla.utils import save_pil_image_with_metadata
+    from stablepy import UpscalerESRGAN
+
+    exif_image = extract_exif_data(image)
+
+    url_upscaler = UPSCALER_DICT_GUI[upscaler_name]
+    directory_upscalers = 'upscalers'
+    os.makedirs(directory_upscalers, exist_ok=True)
+    if not os.path.exists(f"./upscalers/{url_upscaler.split('/')[-1]}"):
+        download_things(directory_upscalers, url_upscaler, HF_TOKEN)
+
+    scaler_beta = UpscalerESRGAN(0, 0)
+    image_up = scaler_beta.upscale(image, upscaler_size, f"./upscalers/{url_upscaler.split('/')[-1]}")
+
+    image_path = save_pil_image_with_metadata(image_up, f'{os.getcwd()}/up_images', exif_image)
+
+    return image_path
 
 dynamic_gpu_duration.zerogpu = True
 sd_gen_generate_pipeline.zerogpu = True
@@ -1460,21 +1496,6 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                 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"])
@@ -1488,6 +1509,23 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                         outputs=[result_metadata],
                     )
 
+        with gr.Tab("Upscaler"):
+            with gr.Row():
+                with gr.Column():
+                    image_up_tab = gr.Image(label="Image", type="pil", sources=["upload"])
+                    upscaler_tab = gr.Dropdown(label="Upscaler", choices=UPSCALER_KEYS[9:], value=UPSCALER_KEYS[11])
+                    upscaler_size_tab = gr.Slider(minimum=1., maximum=4., step=0.1, value=1.1, label="Upscale by")
+                    generate_button_up_tab = gr.Button(value="START UPSCALE", variant="primary")
+
+                with gr.Column():
+                    result_up_tab = gr.Image(label="Result", type="pil", interactive=False, format="png")
+
+                    generate_button_up_tab.click(
+                        fn=esrgan_upscale,
+                        inputs=[image_up_tab, upscaler_tab, upscaler_size_tab],
+                        outputs=[result_up_tab],
+                    )
+
 ## BEGIN MOD
         interface_mode_gui.change(
             change_interface_mode,

env.py

@@ -82,9 +82,11 @@ load_diffusers_format_model = [
     'KBlueLeaf/Kohaku-XL-Zeta',
     'kayfahaarukku/UrangDiffusion-1.4',
     'Eugeoter/artiwaifu-diffusion-2.0',
+    'comin/IterComp',
     "OnomaAIResearch/Illustrious-xl-early-release-v0",
     'Raelina/Rae-Diffusion-XL-V2',
     'Raelina/Raemu-XL-V4',
+    'Raelina/Raehoshi-illust-XL',
     "camenduru/FLUX.1-dev-diffusers",
     "black-forest-labs/FLUX.1-schnell",
     "sayakpaul/FLUX.1-merged",

lora_dict.json

@@ -1371,6 +1371,13 @@
     "https://civitai.com/models/350199",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/385edbe8-3fad-4278-bfed-43d551e8f78b/width=450/8901842.jpeg"
   ],
+  "LECO-mature_female_XL(ill)v10": [
+    "mature female,",
+    "SDXL 1.0",
+    "[illustrious LECO] mature female...? / \u719f\u5973\u304b\u3089\u306a\u305c\u304b\u30ed\u30dc\u306b\u306a\u308bLECO",
+    "https://civitai.com/models/846750",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/f9f14c8b-63cd-4590-872b-4108079b1cb2/width=450/34156304.jpeg"
+  ],
   "LapPillowXL": [
     "LapPillowXL",
     "Pony",
@@ -1476,6 +1483,13 @@
     "https://civitai.com/models/566542",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/e2daae6e-a9fb-43c5-b4ee-cfa264a6455d/width=450/19074383.jpeg"
   ],
+  "MORTARHEADD_like_mecha_v1": [
+    "MORTARHEADD / MH-POSE / MH-BUSTER / MH-KOG / MH-LED-MIRAGE / MH-Engage-SR1 / MH-the-BANG / MH-Neptune / MH-BTK",
+    "SDXL 1.0",
+    "FSS Mortar headd Like mecha",
+    "https://civitai.com/models/844457",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0740f561-8808-425c-a4e0-90575f887262/width=450/34033551.jpeg"
+  ],
   "MS_PDXL_AfterSex_Lite": [
     "after sex, cum, lying, cumdrip, ass, on stomach, on back, fucked silly, sweat, cum pool, bukkake, trembling",
     "Pony",
@@ -2834,6 +2848,20 @@
     "https://civitai.com/models/238419",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/74053ca5-9486-44ee-94d1-9cebcbbe7250/width=450/22207105.jpeg"
   ],
+  "airguitar_XL_v2": [
+    "airgui",
+    "SDXL 1.0",
+    "air guitar / \u30a8\u30a2\u30ae\u30bf\u30fc",
+    "https://civitai.com/models/827890",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/bd10aea5-edc4-4e24-9b3d-58651faf5892/width=450/33009474.jpeg"
+  ],
+  "akanbe-ponyxl-lora-nochekaiser": [
+    "<lora:akanbe-ponyxl-lora-nochekaiser:1>, akanbe, eyelid pull, finger to eye, tongue out, :p, tongue, ;p,",
+    "Pony",
+    "Akanbe (\u3042\u3063\u304b\u3093\u3079\u30fc) - Concept",
+    "https://civitai.com/models/828913",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/f3cfd5c3-725a-4397-ab71-531baeb28533/width=450/33420099.jpeg"
+  ],
   "akifn_pony_v2": [
     "",
     "Pony",
@@ -3149,6 +3177,20 @@
     "https://civitai.com/models/413240",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/b70b41a8-6a37-4cc9-894a-904de8e1f238/width=450/10439295.jpeg"
   ],
+  "betweenbreasts_Pony_v1": [
+    "between breasts",
+    "Pony",
+    "[SDXL&Pony]between breasts / \u7591\u4f3c\u30d1\u30a4\u30ba\u30ea / \u304a\u3063\u3071\u3044\u30dd\u30b1\u30c3\u30c8",
+    "https://civitai.com/models/402979",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/3742babb-c058-47f9-b0bc-2d541da43695/width=450/33014304.jpeg"
+  ],
+  "betweenbreasts_XL_v1": [
+    "between breasts",
+    "SDXL 1.0",
+    "[SDXL&Pony]between breasts / \u7591\u4f3c\u30d1\u30a4\u30ba\u30ea / \u304a\u3063\u3071\u3044\u30dd\u30b1\u30c3\u30c8",
+    "https://civitai.com/models/402979",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/6757b641-e0a7-4fd7-a2eb-f39ed2485303/width=450/9971972.jpeg"
+  ],
   "big_clitoris_pony_V1_0": [
     "big clitoris / clitoris / erection",
     "Pony",
@@ -3268,6 +3310,13 @@
     "https://civitai.com/models/539285",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/03d304a7-fce1-4ed2-b254-1b82698d53f2/width=450/17279266.jpeg"
   ],
+  "breakdance_XL_v2": [
+    "breakdance",
+    "SDXL 1.0",
+    "breakdance / breaking / \u30d6\u30ec\u30a4\u30af\u30c0\u30f3\u30b9 / \u30d6\u30ec\u30a4\u30ad\u30f3",
+    "https://civitai.com/models/851015",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/5ee7371d-7aab-471c-8717-810132afc44c/width=450/34389576.jpeg"
+  ],
   "breast_open_bikini_pony_V1_0": [
     "breast open bikini",
     "Pony",
@@ -3940,6 +3989,20 @@
     "https://civitai.com/models/728624",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a29c4df9-07eb-4f8f-b984-c24b7531225a/width=450/28137263.jpeg"
   ],
+  "defence_pose_XL_V1_0": [
+    "defense pose, crossed arms",
+    "SDXL 1.0",
+    "\u80f8\u3092\u96a0\u3059/defence pose",
+    "https://civitai.com/models/838545",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a0c09c07-ecb6-4e08-80a4-8db895feb3f1/width=450/33960236.jpeg"
+  ],
+  "defence_pose_pony_V1_0": [
+    "defense pose,crossed arms",
+    "Pony",
+    "\u80f8\u3092\u96a0\u3059/defence pose",
+    "https://civitai.com/models/838545",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/4611cac5-7c02-484c-9937-a4b7fb5c3817/width=450/33675457.jpeg"
+  ],
   "deflowered_pony": [
     "1woman, perfect face, fully nude, imminent penetration:1.4, deflowered:1.1, blood on thighs, blood dripping, semen, cum, overflow:1.2, / 1male, huge penis, large testicles, / female pubic hair, uncensored pussy, hairy pussy, blush / <lora:deflowered_pony:0.7>,",
     "Pony",
@@ -4374,6 +4437,13 @@
     "https://civitai.com/models/576705",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/af821c46-6ce4-4726-921c-6cd723f66a0f/width=450/19739141.jpeg"
   ],
+  "glory_wall_pony_V1_0": [
+    " glory wall, ass",
+    "Pony",
+    "\u58c1\u5c3b/glory wall",
+    "https://civitai.com/models/828936",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/680ddc61-36b7-4407-b4b1-b6e63d86581a/width=450/33087471.jpeg"
+  ],
   "gm_meme": [
     "sitting, girl on top, bed, downblouse",
     "Pony",
@@ -4983,6 +5053,13 @@
     "https://civitai.com/models/511064",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ac1f7ec5-bd8c-4694-9aa4-d158c1fa878a/width=450/21578898.jpeg"
   ],
+  "kijyou_fella_pony_V1_0": [
+    " fellatio,oral / on back / 1boy / penis",
+    "Pony",
+    "\u9a0e\u4e57\u30d5\u30a7\u30e9",
+    "https://civitai.com/models/846716",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/886182f3-511e-4b82-ab63-78cc9e246443/width=450/34323618.jpeg"
+  ],
   "kingyo_fukuro_SDXL_V1": [
     "kingyofukuro, bagged fish, fish, bag, goldfish, ",
     "SDXL 1.0",
@@ -5522,6 +5599,13 @@
     "https://civitai.com/models/573056",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/c3c99a9a-f2cf-4c20-ae4b-a03ded327311/width=450/19512503.jpeg"
   ],
+  "multi_chan_v2": [
+    "multi_chan",
+    "Pony",
+    "multiple views chan",
+    "https://civitai.com/models/822633",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/845c7664-e4aa-48a1-a1e7-059531b679ff/width=450/33096611.jpeg"
+  ],
   "multi_prolapse_ponyXL_v1": [
     "anal prolapse /  uterine prolapse /  double prolapse /  ovary prolapse",
     "Pony",
@@ -6033,6 +6117,13 @@
     "https://civitai.com/models/470923",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9870ea43-5d3b-4403-8d3e-48bb122c3897/width=450/13800000.jpeg"
   ],
+  "pet_shaming_pony_v1": [
+    "hansei_b,sitting,seiza,sign",
+    "Pony",
+    "Pet shaming / \u53cd\u7701\u672d",
+    "https://civitai.com/models/851053",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/3d962f39-b4e6-4165-9bb7-b6ff074f2e9f/width=450/34391608.jpeg"
+  ],
   "phallic_food_-_XL_Pony_V2_0": [
     "Banana, half peeled, half peeled banana, / sausage, / sausage, sausage on stick, / Makizushi, sushi roll, ehomaki, / Hot dog, / ketchup, / mustard, / sucking, / holding, / eating, / biting, / licking, / saliva, / fork, / chocolate banana / sandwich / popsicle / cucumber / mushroom",
     "Pony",
@@ -6383,6 +6474,13 @@
     "https://civitai.com/models/575758",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/baccd194-7d91-4d04-858d-627c3ed263f9/width=450/19683271.jpeg"
   ],
+  "railgun_technique": [
+    "coin, holding coin, facing viewer, looking at the viewer, electricity, electrokinesis / orange beam, energy beam, from side, looking to the side, electricity, electrokinesis,  one-arm outstretched, city background  ",
+    "Pony",
+    "Railgun pose and technique (Misaka Mikoto) [To aru majutsu / kagaku]",
+    "https://civitai.com/models/822171",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/d32e5d7b-e17d-4299-bb62-ba3caff139e0/width=450/32814916.jpeg"
+  ],
   "ramune_SDXL_V1": [
     "ramune, food, bottle, ",
     "SDXL 1.0",
@@ -6474,6 +6572,13 @@
     "https://civitai.com/models/498892",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9822ab8f-9832-4471-9ff7-3fe6a00714c8/width=450/14826239.jpeg"
   ],
+  "returningaxis_Illust0_1XL1": [
+    "returningaxis",
+    "SDXL 1.0",
+    "Returning the Axis / \u30a2\u30af\u30b7\u30ba\u8fd4\u3057\u3000SDXL",
+    "https://civitai.com/models/834850",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/3ba9433d-8f5e-465b-bcf4-1568a9624059/width=450/33452531.jpeg"
+  ],
   "revealing_forehead_concept-soralz": [
     "revealing forehead, hand on own hair, forehead, head tilt",
     "Pony",
@@ -6726,6 +6831,20 @@
     "https://civitai.com/models/594617",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/8598e698-88d1-4650-a447-a307a52ce7f4/width=450/20899405.jpeg"
   ],
+  "shiDioV2": [
+    "konodiopose, pointing, parody, meme, smile, open mouth, style parody, upper body",
+    "Pony",
+    "Kono Dio Da! JoJo Meme Pose | Single Concept | PonyXL",
+    "https://civitai.com/models/834937",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a4fd210d-90db-4995-8868-482e597fe927/width=450/33459038.jpeg"
+  ],
+  "shiReroVT1": [
+    "shirerocrazy, cross-eyed, tongue, cherry on tongue, saliva, tongue out, looking at viewer, open mouth, parody, cherry, ",
+    "Pony",
+    "Rerorerorerorero Noriaki Kakyoin Cherry Lick Concept | PonyXL",
+    "https://civitai.com/models/837489",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/05c6aefc-9791-4be4-a5e4-54853cc4a503/width=450/33616097.jpeg"
+  ],
   "shibari_V3_sdxl": [
     "shibari",
     "SDXL 1.0",
@@ -6838,6 +6957,13 @@
     "https://civitai.com/models/602418",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/cf5b0816-ccd6-49bb-8c5b-0fdc56a30dcf/width=450/21430280.jpeg"
   ],
+  "small_penis_cross-section_illustrious-000035": [
+    "tiny penis / cross-section / internal cumshot / excessive cum / deficient cum",
+    "SDXL 1.0",
+    "[Pony / Illustrious XL] cross-section for small penis / \u77ed\u5c0f\u7c97\u30c1\u30f3\u65ad\u9762\u56f3",
+    "https://civitai.com/models/430102",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/426490c3-9646-418e-9741-84f1d9ef0ba7/width=450/33670775.jpeg"
+  ],
   "smalldombigsubV3": [
     "small dom big sub / size difference",
     "Pony",
@@ -6887,6 +7013,13 @@
     "https://civitai.com/models/717353",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/81be9bdf-990b-4a1f-92f4-2dd5ac78fe79/width=450/27574152.jpeg"
   ],
+  "spacium_beam": [
+    "spacium beam / (energy beam:1.1),",
+    "Pony",
+    "Spacium Beam \u30b9\u30da\u30b7\u30a6\u30e0\u5149\u7dda",
+    "https://civitai.com/models/836143",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/a1e6fe94-c09b-4a15-95c7-43cc3f80e4ae/width=450/33548979.jpeg"
+  ],
   "spagainstScreenXLPony": [
     "against glass, breast press, cheek press",
     "Pony",
@@ -7440,6 +7573,20 @@
     "https://civitai.com/models/522462",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/73988a4c-c9df-4859-84d5-c88b36ce2b19/width=450/16298052.jpeg"
   ],
+  "triangularheadpiece_Pony_v1": [
+    "triangular headpiece",
+    "Pony",
+    "[SDXL&Pony] triangular headpiece / shiroshouzoku / \u767d\u88c5\u675f / \u5929\u51a0 / \u4e09\u89d2\u5dfe / \u5e7d\u970a",
+    "https://civitai.com/models/491760",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ef0429ed-90d3-43e3-9da0-85f87fb42e04/width=450/32850050.jpeg"
+  ],
+  "triangularheadpiece_XL_v1": [
+    "triangular headpiece",
+    "SDXL 1.0",
+    "[SDXL&Pony] triangular headpiece / shiroshouzoku / \u767d\u88c5\u675f / \u5929\u51a0 / \u4e09\u89d2\u5dfe / \u5e7d\u970a",
+    "https://civitai.com/models/491760",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/36267afb-7093-4c71-ae47-516c7219fabd/width=450/14403049.jpeg"
+  ],
   "triple_oral_-_triple_barrel_fellatio_-_triple_fellatio-000014": [
     "tripplebj / read description",
     "Pony",
@@ -7971,5 +8118,12 @@
     "\u0421olor palette enhancement",
     "https://civitai.com/models/570834",
     "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/45d05dab-397c-4bc7-8150-8530aa524899/width=450/19621390.jpeg"
+  ],
+  "\u6307\u306e\u8f2a\u30b3\u30ad": [
+    "handjob, hold glans in finger circle, open hand, large breasts, maid, grin, upper body",
+    "Pony",
+    "\u6307\u306e\u8f2a\u30b3\u30ad",
+    "https://civitai.com/models/850497",
+    "https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/0a48013a-809a-41a5-823d-e9183eae0dd0/width=450/34362448.jpeg"
   ]
 }

modutils.py

@@ -28,7 +28,7 @@ def to_list(s):
 
 
 def list_uniq(l):
-        return sorted(set(l), key=l.index)
+    return sorted(set(l), key=l.index)
 
 
 def list_sub(a, b):