Delete app.py

app.py

@@ -1,366 +0,0 @@
-import spaces  # oltava ensimmäisenä!
-
-import torch
-from diffusers import StableDiffusionXLPipeline
-import gradio as gr
-import random
-import re
-
-model_id = "kkntr/sdxl-kkntr"
-
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    model_id,
-    torch_dtype=torch.float16,
-    use_safetensors=True
-).to("cuda")
-
-# --- Rekisteröidään pysyvät LoRAt ADAPTEREINA (ei painoja vielä) ---
-PERSISTENT_LORAS = {
-    "base1": ("kkntr/lora1", 0.2),  # oletuspaino; muutettavissa generate():ssa
-    "base2": ("kkntr/lora2", 0.3),
-    "base3": ("kkntr/lora12", 0.4), 
-    "base4": ("kkntr/lora10", 0.1),
-}
-for name, (repo, _) in PERSISTENT_LORAS.items():
-    pipe.load_lora_weights(repo, adapter_name=name)
-
-# --- Rekisteröidään tyyli-LoRAt ADAPTEREINA (ei painoja vielä) ---
-STYLE_LORAS = {
-    "None": (None, 0.0, "", None),
-    "Foreskin play": ("kkntr/lora5", 1.0, "foreskin insertion", "style_fp"),
-    "Cum through clothing": ("kkntr/lora6", 1.0, "cum in clothing", "style_cc"),
-    "Penis through leghole": ("kkntr/lora7", 1.0, "detailed bulge", "shorts", "style_ptl"),
-    "Frottage": ("kkntr/lora8", 1.0, "byfrottage, frottage"),
-    "Penis towards viewer": ("kkntr/lora9", 1.0, "penis_towards_viewer", "style_ptv"),
-    "Foreskin play closeup": ("kkntr/lora11", 1.0, "ZonkFS, close-up, furry, anthro, faceless male, penis, foreskin, fellatio, oral, foreskin kiss, foreskin play, side view, steam", "style_fscl"),
-}
-for key, spec in STYLE_LORAS.items():
-    repo = spec[0]
-    adapter_name = spec[3]
-    if repo and adapter_name:
-        pipe.load_lora_weights(repo, adapter_name=adapter_name)
-
-scratchpad = {
-    "animal": [
-        "kangaroo", "otter", "tiger", "lion", "wolf", "fox", "polar bear", "deer", "reindeer", "hyena", "bull", 
-"rabbit", "panther", "moose", "cheetah", "badger", "striped hyena", "rat", "leopard", "thylacine", "bighorn sheep", 
-"zebra", "horse", "donkey", "unicorn", "white unicorn", "alligator", "western dragon", "lizard", "crocodile", "theropod",
-"kangaroo", "otter", "tiger", "lion", "wolf", "fox", "polar bear", "deer", "reindeer", "hyena", "bull", 
-"rabbit", "panther", "moose", "cheetah", "badger", "striped hyena", "rat", "leopard", "thylacine", "bighorn sheep", 
-"zebra", "horse", "donkey", "unicorn", "white unicorn", "alligator", "western dragon", "lizard", "crocodile", "theropod",
-"kangaroo", "otter", "tiger", "lion", "wolf", "fox", "polar bear", "deer", "reindeer", "hyena", "bull", 
-"rabbit", "panther", "moose", "cheetah", "badger", "striped hyena", "rat", "leopard", "thylacine", "bighorn sheep", 
-"zebra", "horse", "donkey", "unicorn", "white unicorn", "alligator", "western dragon", "lizard", "crocodile", "theropod",
-"black russian terrier", "pit bull", "schnauzer", "dobermann", "mastiff", "molosser", "pinscher", "rottweiler", "tamaskan dog", 
-"wolfdog", "alaskan husky", "alaskan malamute", "canadian eskimo dog", "german shepherd", "labrador husky", "siberian husky", 
-"labrador", "golden retriever", "dalmatian", "border collie", "collie", "herding dog", "pastoral dog", "sheepdog", 
-"carolina dog", "rhodesian ridgeback", "nordic sled dog", "spitz", "husky", "akita", "mountain dog", "primitive dog", 
-"siberian retriever", "hunting dog", "australian cattle dog", "crash bandicoot", "montgomery gator, fnaf", "bear",
-    ],
-    "pose": [
-        "flexing", "muscle flex", "pinup pose", "victory pose", "squatting", "crouching", 
-"reclining", "lying", "lying on back", "lying on stomach", "lying on side", 
-"sitting", "sitting cross-legged", "sitting on edge", "leaning", "lounging", 
-"hands behind head", "hands on hips", "arms crossed", "arms raised", "arms behind back",
-"jumping", "running pose", "walking pose", "sprinting", "mid-air pose", "push-up position", "plank position", "yoga pose",
-"stretching", "twisting torso", "bending forward", "side bend", "backbend",
-"hands in pockets", "hands on knees", "hands on thighs", "arms extended", "arms flexed",
-"looking over shoulder", "looking up", "looking down", "looking straight", "gaze at camera",
-"hands clasped", "hands together", "resting chin on hand", "leaning on wall",
-"casual pose", "relaxed pose", "confident pose", "triumphant pose",
-"athletic pose", "dynamic pose", "action pose", "defensive stance",
-"martial arts pose", "superhero pose", "dramatic pose",
-    ],
-    "swimwear": [
-        "swimming trunks", "swimming shorts", "swim briefs", "speedo", "bermuda shorts", "swim thong", "athletic swim shorts", "compression swim shorts",
-    ],
-    "headwear": [
-        "baseball cap", "backwards cap", "snapback", "trucker hat", 
-"beanie", "knit cap", "watch cap", "skullcap", 
-"bandana", "headband", "sweatband", 
-"visor", "bucket hat", "sun hat", 
-"cowboy hat", "straw hat", "fedora", "trilby", 
-"flat cap", "newsboy cap", "beret", 
-"military cap", "officer cap", "camouflage helmet", "combat helmet", 
-"motorcycle helmet", "bike helmet", "hard hat", "construction helmet", 
-"hood", "hoodie hood", "cowl", 
-"crown", "circlet", "horned helmet", "samurai helmet", "viking helmet",
-    ],
-    "setting": [
-        "bedroom", "living room", "locker room", "gym", "shower room", 
-"bathroom", "sauna", "kitchen", "garage", "workshop", "basement", 
-
-"beach", "swimming pool", "hot tub", "lake", "waterfall", "forest", 
-"mountains", "cave", "desert", "snowy landscape", "tundra", 
-
-"city street", "alleyway", "rooftop", "nightclub", "bar", "concert stage", 
-"festival", "arcade", "skate park", "basketball court", "football field", 
-"stadium", "boxing ring", "wrestling ring", 
-
-"office", "warehouse", "factory", "construction site", "train station", 
-"subway", "parking lot", "gas station", "motel room", "hotel lobby", 
-
-"campfire", "tent", "cabin", "dorm room", "classroom", 
-"library", "museum", "gallery",
-    ],
-    "hair": [
-        "short hair", "medium hair", "crew cut", "buzz cut", "flat top", 
-"undercut", "slicked back hair", "side part", "center part", 
-"pompadour", "quiff", "spiky hair", "messy hair", 
-"mohawk", "faux hawk", 
-"man bun", "top knot", "low bun",
-    ],
-    "accessories": [
-        "chain necklace", "dog tag necklace", "leather bracelet", "wristband", 
-"watch", "ring", "earring", "stud earring", "hoop earring", 
-"septum piercing", "nose ring", "eyebrow piercing", "lip piercing", 
-"tongue piercing", "fingerless gloves", "leather gloves", 
-"armband", "bandana", "headband", "sunglasses", "aviator sunglasses", "cap", "beanie", "backwards cap", "finger rings", "spiked bracelet", "spiked collar", "chain belt", 
-"pendant necklace", "cuff bracelet", "knuckle rings",
-    ],
-    "hairy": [
-        "chest hair", "belly tuft", "happy trail", "armpit hair", "facial hair", 
-"beard", "mustache", "goatee", "soul patch", "sideburns", "hairy arms", 
-"hairy legs", "hairy back", "full body hair",
-
-"chest hair, belly tuft", "chest hair, happy trail", "chest hair, armpit hair", 
-"chest hair, facial hair", "chest hair, beard", "chest hair, mustache", "chest hair, goatee", 
-"chest hair, soul patch", "chest hair, sideburns", "chest hair, hairy arms", 
-"chest hair, hairy legs", "chest hair, hairy back",
-
-"belly tuft, happy trail", "belly tuft, armpit hair", "belly tuft, facial hair", 
-"belly tuft, beard", "belly tuft, mustache", "belly tuft, goatee", "belly tuft, soul patch", 
-"belly tuft, sideburns", "belly tuft, hairy arms", "belly tuft, hairy legs", "belly tuft, hairy back",
-
-"happy trail, armpit hair", "happy trail, facial hair", "happy trail, beard", 
-"happy trail, mustache", "happy trail, goatee", "happy trail, soul patch", 
-"happy trail, sideburns", "happy trail, hairy arms", "happy trail, hairy legs", "happy trail, hairy back",
-
-"armpit hair, facial hair", "armpit hair, beard", "armpit hair, mustache", 
-"armpit hair, goatee", "armpit hair, soul patch", "armpit hair, sideburns", 
-"armpit hair, hairy arms", "armpit hair, hairy legs", "armpit hair, hairy back",
-
-"facial hair, beard", "facial hair, mustache", "facial hair, goatee", 
-"facial hair, soul patch", "facial hair, sideburns", "facial hair, hairy arms", 
-"facial hair, hairy legs", "facial hair, hairy back",
-
-"beard, mustache", "beard, goatee", "beard, soul patch", "beard, sideburns", 
-"beard, hairy arms", "beard, hairy legs", "beard, hairy back",
-
-"mustache, goatee", "mustache, soul patch", "mustache, sideburns", 
-"mustache, hairy arms", "mustache, hairy legs", "mustache, hairy back",
-
-"goatee, soul patch", "goatee, sideburns", "goatee, hairy arms", 
-"goatee, hairy legs", "goatee, hairy back",
-
-"soul patch, sideburns", "soul patch, hairy arms", "soul patch, hairy legs", "soul patch, hairy back",
-
-"sideburns, hairy arms", "sideburns, hairy legs", "sideburns, hairy back",
-
-"hairy arms, hairy legs", "hairy arms, hairy back", "hairy legs, hairy back",
-
-"chest hair, belly tuft, happy trail", "chest hair, belly tuft, beard", 
-"chest hair, happy trail, beard", "chest hair, armpit hair, beard", 
-"chest hair, armpit hair, mustache", "chest hair, armpit hair, goatee", 
-"chest hair, facial hair, beard", "chest hair, facial hair, mustache", 
-"chest hair, facial hair, goatee", "chest hair, hairy arms, beard", 
-"chest hair, hairy legs, beard", "chest hair, hairy back, beard",
-
-"full body hair, chest hair", "full body hair, beard", "full body hair, armpit hair", 
-"full body hair, happy trail", "full body hair, facial hair"
-,
-    ],
-    "body": [
-        "skinny", "underweight", "slim", "lean", "fit", "athletic", "toned", "muscular", 
-"big muscles", "hyper muscles", "bodybuilder", "buff", "beefy", "manly", "masculine", 
-"stocky", "broad-shouldered", "medium build", "average build", "dad body", 
-"big body", "heavyset", "obese", "large frame", "solid build", "powerful build", 
-"athletic, muscular", "athletic, toned", "muscular, big body", "buff, muscular", 
-"beefy, big body", "stocky, muscular", "broad-shouldered, muscular", 
-"average build, toned", "manly, muscular", "masculine, buff", "bodybuilder, big muscles", 
-"lean, muscular", "fit, muscular", "toned, muscular", "athletic, big muscles", 
-"muscular, beefy", "stocky, strong", "broad-shouldered, fit", "medium build, muscular", 
-"dad body, big body", "heavyset, muscular", "obese, strong", "powerful build, muscular", 
-"solid build, athletic", "large frame, muscular", "buff, manly", "beefy, masculine", 
-"athletic, solid build", "lean, fit", "toned, fit", "muscular, manly", "big muscles, masculine",
-    ],
-    "underwear": [
-        "boxer briefs", "briefs", "trunks", "boxers", "jockstrap", "thong", "g-string",
-"athletic briefs", "sport trunks", "compression shorts",
-"open-back jockstrap", "wet look briefs",
-    ],
-    "sex": [
-        "anvil position", "mating press", "piledriver position", "prison guard position", "chair position", "perching position", "cowgirl position", "doggystyle", "guard position", "leg glider position", "mastery position",
-"mermaid position, tabletop", "Spoon Position", "Speed Bump Position", "reverse missionary position", "Reverse Stand and Carry Position", "butler position", "jockey position", "bent spoon position", "bodyguard position", "missionary position", "stand and carry position",
-"reverse cowgirl position", "full nelson position", "deck chair position", "amazon position", "ballerina position", "dancer position", "bumper cars position", "arch position",
-    ],
-    "topwear": [
-        "t-shirt", "tank top", "polo shirt", 
-"shirt", "button-up shirt", "flannel shirt", "unbuttoned shirt",
-"long sleeve shirt", 
-"sweatshirt", "hoodie", "sweater",
-"jacket", "coat", "vest", "jersey",
-    ],
-    "duo": [
-        "duo", "two characters", "2 characters", "two figures", "2 figures", "pair",
-    ],
-    "solo": [
-        "solo", "1 male", "single character", "one figure",
-    ],
-    "bottomwear": [
-        "jeans", "cargo pants", "khaki pants", "chinos", "linen pants",
-"sweatpants", "joggers", "shorts", "denim shorts", "cargo shorts",
-"corduroy pants", "drawstring pants", "relaxed fit trousers", "capri pants", "rolled-up pants",
-"leather pants",
-"rubber pants",
-    ],
-    "makingout": [
-        "making out", "kissing", "necking", "cuddling", "spooning", "hugging", 
-"back hug", "forehead kiss", "cheek kiss", "nose touch", "hand on waist", 
-"gentle embrace", "playful flirting", "body contact", "touching face", 
-"lip bite", "biting lips", "tongue kissing", "passionate kiss", 
-"holding hands", "caressing", "stroking hair", "nuzzling", "licking lips", 
-"seductive pose", "intimate embrace", "pressing bodies together", "sensual touch",
-    ],
-    "light": [
-        "day", "bright", "sunlight", "sunrise", "sunset", "golden hour", "morning light", "afternoon light", "noon light", 
-"backlight", "rim light", "lens flare", "volumetric light", "god rays", "ray casting light", "neon glow", 
-"night", "dark", "dim light", "moonlight", "shadow", "silhouette", "starlight", "twilight", 
-"studio light", "window light", "candle light", "fireplace light", "ambient light", "soft light", "hard light", 
-"fog", "mist", "haze", "overcast", "diffused light", "dappled light", "reflected light", "spotlight", "floodlight", "colored light",
-    ],
-    "perspective": [
-        "bird's-eye view", "worm's-eye view", "high-angle view", "low-angle view", "front view", "side view", "back view", "three-quarter view", 
-"dutch angle", "fisheye lens", "wide-angle lens", "telephoto lens", "macro lens", "close-up", "extreme close-up", 
-"over-the-shoulder view", "panoramic view", "foreshortening", "tilted perspective", "distorted perspective", "atmospheric perspective", 
-"isometric view", "top-down view", "bottom-up view", "diagonal view", "tilted camera angle", "dynamic angle", "eye-level view", 
-"low-key perspective", "high-key perspective", "tilt-shift perspective", "point-of-view", "bird’s eye panorama", "360-degree view", 
-"overhead shot", "tracking shot", "zoomed-in view", "zoomed-out view",
-    ],
-    "face": [
-        "tongue out", "seductive", "bedroom eyes", "looking at viewer", 
-"looking embarrassed", "blushing", "nervous", "shy", "open mouth", 
-"grin", "playful", "quirky", "mischievous", "laughing", "happy", "joyful", 
-"caring", "kind face", "gentle smile", "smirk", "smile", 
-"surprised", "shocked", "astonished", "confused", "thinking", 
-"pensive", "curious", "serious", "determined", "focused", 
-"disappointed", "neutral", "calm", "relaxed", "content", "tired",
-    ],
-    "colour": [
-        "yellow", "orange", "red", "pink", "purple", "blue", "light blue",
-"green", "lime", "turquoise", "teal", "mint",
-"brown", "beige", "tan", "olive", "khaki",
-"white", "grey", "black",
-"navy", "violet", "camo", "burgundy",
-"ivory",
-"gold", "silver",
-    ],
-}
-
-def apply_scratchpad(prompt: str) -> str:
-    """Korvaa [tagi] scratchpadin vaihtoehdoilla satunnaisesti.
-       Jos tagia ei löydy, poistetaan hakasulkeet kokonaan.
-    """
-    matches = re.findall(r"\[(.+?)\]", prompt)
-    for tag in matches:
-        if tag in scratchpad:
-            choice = random.choice(scratchpad[tag])
-            # korvataan ilman hakasulkeita
-            prompt = prompt.replace(f"[{tag}]", choice, 1)
-        else:
-            # poistetaan tuntemattomat tagit
-            prompt = prompt.replace(f"[{tag}]", "", 1)
-    return prompt.strip()
-
-
-resolutions = [
-    (592, 1768), (608, 1696), (640, 1624), (672, 1544),
-    (720, 1440), (744, 1392), (784, 1320), (832, 1248),
-    (864, 1208), (880, 1184), (912, 1144), (976, 1072),
-    (1024, 1024), (1072, 912), (1144, 912), (1184, 880),
-    (1208, 864), (1248, 832), (1320, 784), (1392, 744),
-    (1440, 720), (1544, 672), (1624, 640), (1696, 608),
-    (1768, 592),
-]
-
-@spaces.GPU
-def generate(prompt, negative_prompt, width, height, random_res, selected_style):
-    if random_res:
-        width, height = random.choice(resolutions)
-
-    prompt = apply_scratchpad(prompt)
-
-    # Rakennetaan listat aktiivisista adaptereista ja niiden painoista
-    adapter_names = []
-    adapter_weights = []
-
-    # 1) Pysyvät LoRAt
-    for name, (_, w) in PERSISTENT_LORAS.items():
-        adapter_names.append(name)
-        adapter_weights.append(w)
-
-    # 2) Tyyli-LoRA + trigger
-    if selected_style in STYLE_LORAS and selected_style != "None":
-        repo, style_w, trigger, style_adapter = STYLE_LORAS[selected_style]
-        prompt = f"{trigger}, {prompt}"
-        adapter_names.append(style_adapter)
-        adapter_weights.append(style_w)   # esim. 7.0 jos haluat “mössöä”
-    # jos "None", ei lisätä tyyliadapteria
-
-    # Varmuuden vuoksi: ei fuseta, vaan käytetään dynaamisia painoja
-    # Jos olet joskus fusennut, nollaa se ensin:
-    try:
-        pipe.unfuse_lora()
-    except Exception:
-        pass
-
-    pipe.set_adapters(adapter_names, adapter_weights=adapter_weights)
-
-    quality_suffix = (
-        ", (anthro), (male), (best quality), high resolution, (ultra detailed),"
-        " 8k, (highly detailed), (masterpiece), (sharp focus), (intricate details),"
-        " (perfect anatomy), (depth of field), high dynamic range, (smooth shading),"
-        " (realistic shadows), (rich texture), professional color grading, (fine details),"
-        " ultra sharp, HDR, (anatomically correct), meesh, by meesh, aaron, 20r0j_2, civifur"
-    )
-
-    full_prompt = prompt + quality_suffix
-
-    image = pipe(
-        prompt=full_prompt,
-        negative_prompt=negative_prompt,
-        width=width,
-        height=height,
-        num_inference_steps=35,
-        guidance_scale=7.5
-    ).images[0]
-
-    # (Valinnainen) palautetaan tekstiin näkyviin mitkä adapterit ja painot oli käytössä
-    used = ", ".join([f"{n}:{w}" for n, w in zip(adapter_names, adapter_weights)])
-    return image, f"{full_prompt}\n\n[adapters] {used}"
-
-# UI sama kuin ennen
-gr.Interface(
-    fn=generate,
-    inputs=[
-        gr.Textbox(label="Prompt"),
-        gr.Textbox(
-            label="Negative Prompt",
-            value="(worst quality), (low quality), (blurry), (bad anatomy), (fused fingers), (extra fingers), (missing fingers), (deformed), (jpeg artifacts), (watermark), (signature), text, error, out of frame, bad hands, bad feet, bad proportions, fused limbs, cropped, username, logo, bad perspective, lowres, overexposure, bad artist, bad composition, distorted, oversaturated, mutated, bad shadow, multiple limbs, extra arms, extra legs, bad background, tilted, bad lighting, bad face, ugly, bad eyes, disfigured, multiple pictures"
-        ),
-        gr.Slider(minimum=512, maximum=1024, step=64, value=768, label="Width"),
-        gr.Slider(minimum=512, maximum=1024, step=64, value=768, label="Height"),
-        gr.Checkbox(label="Random resolution mode (override sliders)", value=True),
-        gr.Dropdown(
-            choices=list(STYLE_LORAS.keys()),
-            value="None",
-            label="Select Style"
-        )
-    ],
-    outputs=[
-        gr.Image(label="Generated Image"),
-        gr.Textbox(label="Used Prompt", lines=6)
-    ],
-    title="FURRYSTYLE",
-    description="Give a prompt"
-).launch()