import os
import json
import copy
import time
import random
import logging
import numpy as np
from typing import Any, Dict, List, Optional, Union

import torch
from PIL import Image
import gradio as gr


from diffusers import (
    DiffusionPipeline,
    AutoencoderTiny,
    AutoencoderKL,
    AutoPipelineForImage2Image,
    FluxPipeline,
    FlowMatchEulerDiscreteScheduler)

from huggingface_hub import (
    hf_hub_download,
    HfFileSystem,
    ModelCard,
    snapshot_download)

from diffusers.utils import load_image

import spaces

#---if workspace = local or colab---

# Authenticate with Hugging Face
# from huggingface_hub import login

# Log in to Hugging Face using the provided token
# hf_token = 'hf-token-authentication'
# login(hf_token)


def calculate_shift(
    image_seq_len,
    base_seq_len: int = 256,
    max_seq_len: int = 4096,
    base_shift: float = 0.5,
    max_shift: float = 1.16,
):
    m = (max_shift - base_shift) / (max_seq_len - base_seq_len)
    b = base_shift - m * base_seq_len
    mu = image_seq_len * m + b
    return mu

def retrieve_timesteps(
    scheduler,
    num_inference_steps: Optional[int] = None,
    device: Optional[Union[str, torch.device]] = None,
    timesteps: Optional[List[int]] = None,
    sigmas: Optional[List[float]] = None,
    **kwargs,
):
    if timesteps is not None and sigmas is not None:
        raise ValueError("Only one of `timesteps` or `sigmas` can be passed. Please choose one to set custom values")
    if timesteps is not None:
        scheduler.set_timesteps(timesteps=timesteps, device=device, **kwargs)
        timesteps = scheduler.timesteps
        num_inference_steps = len(timesteps)
    elif sigmas is not None:
        scheduler.set_timesteps(sigmas=sigmas, device=device, **kwargs)
        timesteps = scheduler.timesteps
        num_inference_steps = len(timesteps)
    else:
        scheduler.set_timesteps(num_inference_steps, device=device, **kwargs)
        timesteps = scheduler.timesteps
    return timesteps, num_inference_steps

# FLUX pipeline
@torch.inference_mode()
def flux_pipe_call_that_returns_an_iterable_of_images(
    self,
    prompt: Union[str, List[str]] = None,
    prompt_2: Optional[Union[str, List[str]]] = None,
    height: Optional[int] = None,
    width: Optional[int] = None,
    num_inference_steps: int = 28,
    timesteps: List[int] = None,
    guidance_scale: float = 3.5,
    num_images_per_prompt: Optional[int] = 1,
    generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
    latents: Optional[torch.FloatTensor] = None,
    prompt_embeds: Optional[torch.FloatTensor] = None,
    pooled_prompt_embeds: Optional[torch.FloatTensor] = None,
    output_type: Optional[str] = "pil",
    return_dict: bool = True,
    joint_attention_kwargs: Optional[Dict[str, Any]] = None,
    max_sequence_length: int = 512,
    good_vae: Optional[Any] = None,
):
    height = height or self.default_sample_size * self.vae_scale_factor
    width = width or self.default_sample_size * self.vae_scale_factor
    
    self.check_inputs(
        prompt,
        prompt_2,
        height,
        width,
        prompt_embeds=prompt_embeds,
        pooled_prompt_embeds=pooled_prompt_embeds,
        max_sequence_length=max_sequence_length,
    )

    self._guidance_scale = guidance_scale
    self._joint_attention_kwargs = joint_attention_kwargs
    self._interrupt = False

    batch_size = 1 if isinstance(prompt, str) else len(prompt)
    device = self._execution_device

    lora_scale = joint_attention_kwargs.get("scale", None) if joint_attention_kwargs is not None else None
    prompt_embeds, pooled_prompt_embeds, text_ids = self.encode_prompt(
        prompt=prompt,
        prompt_2=prompt_2,
        prompt_embeds=prompt_embeds,
        pooled_prompt_embeds=pooled_prompt_embeds,
        device=device,
        num_images_per_prompt=num_images_per_prompt,
        max_sequence_length=max_sequence_length,
        lora_scale=lora_scale,
    )
    
    num_channels_latents = self.transformer.config.in_channels // 4
    latents, latent_image_ids = self.prepare_latents(
        batch_size * num_images_per_prompt,
        num_channels_latents,
        height,
        width,
        prompt_embeds.dtype,
        device,
        generator,
        latents,
    )
    
    sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps)
    image_seq_len = latents.shape[1]
    mu = calculate_shift(
        image_seq_len,
        self.scheduler.config.base_image_seq_len,
        self.scheduler.config.max_image_seq_len,
        self.scheduler.config.base_shift,
        self.scheduler.config.max_shift,
    )
    timesteps, num_inference_steps = retrieve_timesteps(
        self.scheduler,
        num_inference_steps,
        device,
        timesteps,
        sigmas,
        mu=mu,
    )
    self._num_timesteps = len(timesteps)

    guidance = torch.full([1], guidance_scale, device=device, dtype=torch.float32).expand(latents.shape[0]) if self.transformer.config.guidance_embeds else None

    for i, t in enumerate(timesteps):
        if self.interrupt:
            continue

        timestep = t.expand(latents.shape[0]).to(latents.dtype)

        noise_pred = self.transformer(
            hidden_states=latents,
            timestep=timestep / 1000,
            guidance=guidance,
            pooled_projections=pooled_prompt_embeds,
            encoder_hidden_states=prompt_embeds,
            txt_ids=text_ids,
            img_ids=latent_image_ids,
            joint_attention_kwargs=self.joint_attention_kwargs,
            return_dict=False,
        )[0]

        latents_for_image = self._unpack_latents(latents, height, width, self.vae_scale_factor)
        latents_for_image = (latents_for_image / self.vae.config.scaling_factor) + self.vae.config.shift_factor
        image = self.vae.decode(latents_for_image, return_dict=False)[0]
        yield self.image_processor.postprocess(image, output_type=output_type)[0]
        latents = self.scheduler.step(noise_pred, t, latents, return_dict=False)[0]
        torch.cuda.empty_cache()
        
    latents = self._unpack_latents(latents, height, width, self.vae_scale_factor)
    latents = (latents / good_vae.config.scaling_factor) + good_vae.config.shift_factor
    image = good_vae.decode(latents, return_dict=False)[0]
    self.maybe_free_model_hooks()
    torch.cuda.empty_cache()
    yield self.image_processor.postprocess(image, output_type=output_type)[0]

#------------------------------------------------------------------------------------------------------------------------------------------------------------#
loras = [
    #Super-Realism
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Super-Realism-LoRA/resolve/main/images/1.png",
        "title": "Super Realism",
        "repo": "strangerzonehf/Flux-Super-Realism-LoRA",
        "weights": "super-realism.safetensors",
        "trigger_word": "Super Realism"            
    },
    #Dalle-Mix
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Dalle-Mix-LoRA/resolve/main/images/D3.png",
        "title": "Dalle Mix",
        "repo": "prithivMLmods/Flux-Dalle-Mix-LoRA",
        "weights": "dalle-mix.safetensors",
        "trigger_word": "dalle-mix"           
    },
    #anime-v1
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Animeo-v1-LoRA/resolve/main/images/A4.png",
        "title": "Animeo Mix",
        "repo": "strangerzonehf/Flux-Animeo-v1-LoRA",
        "weights": "Animeo.safetensors",
        "trigger_word": "Animeo"        
    },
    #anime-v2
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Animex-v2-LoRA/resolve/main/images/A33.png",
        "title": "Animex Mix",
        "repo": "strangerzonehf/Flux-Animex-v2-LoRA",
        "weights": "Animex.safetensors",
        "trigger_word": "Animex"         
    },
    #Mixer2.0
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Midjourney-Mix2-LoRA/resolve/main/images/3.png",
        "title": "Midjourney Mix 2",
        "repo": "strangerzonehf/Flux-Midjourney-Mix2-LoRA",
        "weights": "mjV6.safetensors",
        "trigger_word": "MJ v6"        
    },
    #Super-Blend
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Super-Blend-LoRA/resolve/main/images/SB1.png",
        "title": "Super Blend",
        "repo": "strangerzonehf/Flux-Super-Blend-LoRA",
        "weights": "Super-Blend.safetensors",
        "trigger_word": "Super Blend"         
    },
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Long-Toon-LoRA/resolve/main/images/LT5.png",
        "title": "Long Toons",
        "repo": "prithivMLmods/Flux-Long-Toon-LoRA",
        "weights": "Long-Toon.safetensors",
        "trigger_word": "Long toons"         
    },
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Isometric-3D-Cinematography/resolve/main/images/IS1.png",
        "title": "Isometric 3D Cinematography",
        "repo": "strangerzonehf/Flux-Isometric-3D-Cinematography",
        "weights": "Isometric-3D-Cinematography.safetensors",
        "trigger_word": "Isometric 3D Cinematography"         
    },
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Cute-3D-Kawaii-LoRA/resolve/main/images/CK3.png",
        "title": "Cute 3D Kawaii",
        "repo": "strangerzonehf/Flux-Cute-3D-Kawaii-LoRA",
        "weights": "Cute-3d-Kawaii.safetensors",
        "trigger_word": "Cute 3d Kawaii"        
    },
    
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Isometric-3D-LoRA/resolve/main/images/ID2.png",
        "title": "Isometric 3D",
        "repo": "strangerzonehf/Flux-Isometric-3D-LoRA",
        "weights": "Isometric-3D.safetensors",
        "trigger_word": "Isometric 3D" 
    },
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Toonic-2.5D-LoRA/resolve/main/images/T2.png",
        "title": "Toon 2.5D",
        "repo": "prithivMLmods/Flux-Toonic-2.5D-LoRA",
        "weights": "toonic2.5D.safetensors",
        "trigger_word": "toonic 2.5D"         
    },
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-YWL-Realism-LoRA/resolve/main/images/R3.png",
        "title": "YWL Realism",
        "repo": "strangerzonehf/Flux-YWL-Realism-LoRA",
        "weights": "ywl-realism.safetensors",
        "trigger_word": "ylw realism"          
    },
    #chill-guy
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Chill-Guy-Zone/resolve/main/images/8.png",
        "title": "Chill Guy",
        "repo": "prithivMLmods/Flux-Chill-Guy-Zone",
        "weights": "chill-guy.safetensors",
        "trigger_word": "chill guy"           
    },
    {
        "image": "https://huggingface.co/p1atdev/flux.1-schnell-pvc-style-lora/resolve/main/images/flux_lora_00221_.png",
        "title": "Anime PVC Style",
        "repo": "p1atdev/flux.1-schnell-pvc-style-lora",
        "weights": "pvc-shnell-7250+7500.safetensors",
        "trigger_word": "pvc figure, nendoroid, figma"         
    },
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-C4C-Design-LoRA/resolve/main/images/4.png",
        "title": "Smiley C4C",
        "repo": "strangerzonehf/Flux-C4C-Design-LoRA",
        "weights": "Smiley-C4C.safetensors",
        "trigger_word": "Smiley C4C"         
    },
#-----------------------------------------------------------------------------------LoRA's----------------------------------------------------------------------#
    #0
    {
        "image": "https://huggingface.co/prithivMLmods/Purple-Dreamy-Flux-LoRA/resolve/main/images/PD3.png",
        "title": "Purple Dream",
        "repo": "prithivMLmods/Purple-Dreamy-Flux-LoRA",
        "weights": "Purple-Dreamy.safetensors",
        "trigger_word": "Purple Dreamy"    
    },
    #1
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-FaceRealism/resolve/main/images/11.png",
        "title": "Flux Face Realism",
        "repo": "prithivMLmods/Canopus-LoRA-Flux-FaceRealism",
        "trigger_word": "Realism"
    },
    #2
    {
        "image": "https://huggingface.co/alvdansen/softserve_anime/resolve/main/images/ComfyUI_00134_.png",
        "title": "Softserve Anime",
        "repo": "alvdansen/softserve_anime",
        "trigger_word": "sftsrv style illustration"
    },
    #3
    {
        "image": "https://huggingface.co/prithivMLmods/Fashion-Hut-Modeling-LoRA/resolve/main/images/MO1.png",
        "title": "Modeling Hut",
        "repo": "prithivMLmods/Fashion-Hut-Modeling-LoRA",
        "trigger_word": "Modeling of"
    },
    #4
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-One-Click-Creative-Template/resolve/main/images/f2cc649985648e57b9b9b14ca7a8744ac8e50d75b3a334ed4df0f368.jpg",
        "title": "Creative Template",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-One-Click-Creative-Template",
        "trigger_word": "The background is 4 real photos, and in the middle is a cartoon picture summarizing the real photos."
    },
    #5
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0/resolve/main/images/XX.png",
        "title": "Ultra Realism",
        "repo": "prithivMLmods/Canopus-LoRA-Flux-UltraRealism-2.0",
        "trigger_word": "Ultra realistic"
    },
    #6
    {
        "image": "https://huggingface.co/gokaygokay/Flux-Game-Assets-LoRA-v2/resolve/main/images/example_y2bqpuphc.png",
        "title": "Game Assets",
        "repo": "gokaygokay/Flux-Game-Assets-LoRA-v2",
        "trigger_word": "wbgmsst, white background"
    },
    #7
    {
        "image": "https://huggingface.co/alvdansen/softpasty-flux-dev/resolve/main/images/ComfyUI_00814_%20(2).png",
        "title": "Softpasty",
        "repo": "alvdansen/softpasty-flux-dev",
        "trigger_word": "araminta_illus illustration style"
    },
    #8
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-add-details/resolve/main/images/0.png",
        "title": "Details Add",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-add-details",
        "trigger_word": ""
    },
    #9
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-Anime/resolve/main/assets/4.png",
        "title": "Flux Anime",
        "repo": "prithivMLmods/Canopus-LoRA-Flux-Anime",
        "trigger_word": "Anime"
    },
    #10
    {
        "image": "https://huggingface.co/aleksa-codes/flux-ghibsky-illustration/resolve/main/images/example5.jpg",
        "title": "Ghibsky Illustration",
        "repo": "aleksa-codes/flux-ghibsky-illustration",
        "trigger_word": "GHIBSKY style painting"
    },
    #11
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Dark-Fantasy/resolve/main/images/c2215bd73da9f14fcd63cc93350e66e2901bdafa6fb8abaaa2c32a1b.jpg",
        "title": "Dark Fantasy",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Dark-Fantasy",
        "trigger_word": ""
    },
    #12
    {
        "image": "https://huggingface.co/Norod78/Flux_1_Dev_LoRA_Paper-Cutout-Style/resolve/main/d13591878d5043f3989dd6eb1c25b710_233c18effb4b491cb467ca31c97e90b5.png",
        "title": "Paper Cutout",
        "repo": "Norod78/Flux_1_Dev_LoRA_Paper-Cutout-Style",
        "trigger_word": "Paper Cutout Style"
    },
    #13
    {
        "image": "https://huggingface.co/alvdansen/mooniverse/resolve/main/images/out-0%20(17).webp",
        "title": "Mooniverse",
        "repo": "alvdansen/mooniverse",
        "trigger_word": "surreal style"
    },
    #14
    {
        "image": "https://huggingface.co/alvdansen/pola-photo-flux/resolve/main/images/out-0%20-%202024-09-22T130819.351.webp",
        "title": "Pola Photo",
        "repo": "alvdansen/pola-photo-flux",
        "trigger_word": "polaroid style"
    },
    #15
    {
        "image": "https://huggingface.co/multimodalart/flux-tarot-v1/resolve/main/images/7e180627edd846e899b6cd307339140d_5b2a09f0842c476b83b6bd2cb9143a52.png",
        "title": "Flux Tarot",
        "repo": "multimodalart/flux-tarot-v1",
        "trigger_word": "in the style of TOK a trtcrd tarot style"
    },
    #16
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Dev-Real-Anime-LoRA/resolve/main/images/111.png",
        "title": "Real Anime",
        "repo": "prithivMLmods/Flux-Dev-Real-Anime-LoRA",
        "trigger_word": "Real Anime"
    },
    #17
    {
        "image": "https://huggingface.co/diabolic6045/Flux_Sticker_Lora/resolve/main/images/example_s3pxsewcb.png",
        "title": "Stickers",
        "repo": "diabolic6045/Flux_Sticker_Lora",
        "trigger_word": "5t1cker 5ty1e"
    },
    #18
    {
        "image": "https://huggingface.co/VideoAditor/Flux-Lora-Realism/resolve/main/images/feel-the-difference-between-using-flux-with-lora-from-xlab-v0-j0ehybmvxehd1.png",
        "title": "Realism",
        "repo": "XLabs-AI/flux-RealismLora",
        "trigger_word": ""
    },
    #19
    {
        "image": "https://huggingface.co/alvdansen/flux-koda/resolve/main/images/ComfyUI_00583_%20(1).png",
        "title": "Koda",
        "repo": "alvdansen/flux-koda",
        "trigger_word": "flmft style"
    },
    #20
    {
        "image": "https://huggingface.co/mgwr/Cine-Aesthetic/resolve/main/images/00019-1333633802.png",
        "title": "Cine Aesthetic",
        "repo": "mgwr/Cine-Aesthetic",
        "trigger_word": "mgwr/cine"
    },
    #21
    {
        "image": "https://huggingface.co/SebastianBodza/flux_cute3D/resolve/main/images/astronaut.webp",
        "title": "Cute 3D",
        "repo": "SebastianBodza/flux_cute3D",
        "trigger_word": "NEOCUTE3D"
    },
    #22
    {
        "image": "https://huggingface.co/bingbangboom/flux_dreamscape/resolve/main/images/3.jpg",
        "title": "Dreamscape",
        "repo": "bingbangboom/flux_dreamscape",
        "trigger_word": "in the style of BSstyle004"
    },   
    #23
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-FaceRealism/resolve/main/images/xc.webp",
        "title": "Cute Kawaii",
        "repo": "prithivMLmods/Canopus-Cute-Kawaii-Flux-LoRA",
        "trigger_word": "cute-kawaii"
    },    
    #24
    {
        "image": "https://cdn-uploads.huggingface.co/production/uploads/64b24543eec33e27dc9a6eca/_jyra-jKP_prXhzxYkg1O.png",
        "title": "Pastel Anime",
        "repo": "Raelina/Flux-Pastel-Anime",
        "trigger_word": "Anime"
    },
    #25
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Vector-Journey/resolve/main/images/f7a66b51c89896854f31bef743dc30f33c6ea3c0ed8f9ff04d24b702.jpg",
        "title": "Vector",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Vector-Journey",
        "trigger_word": "artistic style blends reality and illustration elements"
    },
    #26
    {
        "image": "https://huggingface.co/bingbangboom/flux-miniature-worlds/resolve/main/images/2.jpg",
        "title": "Miniature",
        "repo": "bingbangboom/flux-miniature-worlds",
        "weights": "flux_MNTRWRLDS.safetensors",
        "trigger_word": "Image in the style of MNTRWRLDS"
    },  
    #27
    {
        "image": "https://huggingface.co/glif-loradex-trainer/bingbangboom_flux_surf/resolve/main/samples/1729012111574__000002000_0.jpg",
        "title": "Surf Bingbangboom",
        "repo": "glif-loradex-trainer/bingbangboom_flux_surf",
        "weights": "flux_surf.safetensors",
        "trigger_word": "SRFNGV01"
    }, 
    #28
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-Snoopy-Charlie-Brown-Flux-LoRA/resolve/main/000.png",
        "title": "Snoopy Charlie",
        "repo": "prithivMLmods/Canopus-Snoopy-Charlie-Brown-Flux-LoRA",
        "trigger_word": "Snoopy Charlie Brown"
    }, 
    #29
    {
        "image": "https://huggingface.co/alvdansen/sonny-anime-fixed/resolve/main/images/uqAuIMqA6Z7mvPkHg4qJE_f4c3cbe64e0349e7b946d02adeacdca3.png",
        "title": "Fixed Sonny",
        "repo": "alvdansen/sonny-anime-fixed",
        "trigger_word": "nm22 style"
    }, 
    #30
    {
        "image": "https://huggingface.co/davisbro/flux-multi-angle/resolve/main/multi-angle-examples/3.png",
        "title": "Multi Angle",
        "repo": "davisbro/flux-multi-angle",
        "trigger_word": "A TOK composite photo of a person posing at different angles"
    },
    #31
    {
        "image": "https://huggingface.co/glif/how2draw/resolve/main/images/glif-how2draw-araminta-k-vbnvy94npt8m338r2vm02m50.jpg",
        "title": "How2Draw",
        "repo": "glif/how2draw",
        "trigger_word": "How2Draw"
        
    },
    #32
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Text-Poster/resolve/main/images/6dd1a918d89991ad5e40513ab88e7d892077f89dac93edcf4b660dd2.jpg",
        "title": "Text Poster",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Text-Poster",
        "trigger_word": "text poster"        
    },
    #33
    {
        "image": "https://huggingface.co/SebastianBodza/Flux_Aquarell_Watercolor_v2/resolve/main/images/coffee.webp",
        "title": "Aquarell Watercolor",
        "repo": "SebastianBodza/Flux_Aquarell_Watercolor_v2",
        "trigger_word": "AQUACOLTOK"  
    },
    #34
    {
        "image": "https://huggingface.co/Purz/face-projection/resolve/main/34031797.jpeg",
        "title": "Face Projection ",
        "repo": "Purz/face-projection",
        "trigger_word": "f4c3_p40j3ct10n"  
    },
    #35
    {
        "image": "https://huggingface.co/martintomov/ecom-flux-v2/resolve/main/images/example_z30slf97z.png",
        "title": "Ecom Design Art",
        "repo": "martintomov/ecom-flux-v2",
        "trigger_word": ""
    },
    #36
    {
        "image": "https://huggingface.co/TheAwakenOne/max-headroom/resolve/main/sample/max-headroom_000900_00_20241015234926.png",
        "title": "Max Head-Room",
        "repo": "TheAwakenOne/max-headroom",
        "weights": "max-headroom-v1.safetensors",
        "trigger_word": "M2X, Max-Headroom"
    },
    #37
    {
        "image": "https://huggingface.co/renderartist/toyboxflux/resolve/main/images/3D__00366_.png",
        "title": "Toy Box Flux",
        "repo": "renderartist/toyboxflux",
        "weights": "Toy_Box_Flux_v2_renderartist.safetensors",
        "trigger_word": "t0yb0x, simple toy design, detailed toy design, 3D render"
    },
    #38
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-live-3D/resolve/main/images/51a716fb6fe9ba5d54c260b70e7ff661d38acedc7fb725552fa77bcf.jpg",
        "title": "Live 3D",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-live-3D",
        "trigger_word": ""
    },
    #39
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Garbage-Bag-Art/resolve/main/images/42e944819b43869a03dc252d10409b5944a62494c7082816121016f9.jpg",
        "title": "Garbage Bag Art",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Garbage-Bag-Art",
        "trigger_word": "Inflatable plastic bag"
    },
    #40
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design/resolve/main/images/73e7db6a33550d05836ce285549de60075d05373c7b0660d631dac33.jpg",
        "title": "Logo Design",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Logo-Design",
        "trigger_word": "wablogo, logo, Minimalist"
    }, 
    #41
    {
        "image": "https://huggingface.co/punzel/flux_sadie_sink/resolve/main/images/ComfyUI_Flux_Finetune_00069_.png",
        "title": "Sadie Sink",
        "repo": "punzel/flux_sadie_sink",
        "weights": "flux_sadie_sink.safetensors",
        "trigger_word": "Sadie Sink"
    }, 
    #42
    {
        "image": "https://huggingface.co/punzel/flux_jenna_ortega/resolve/main/images/ComfyUI_Flux_Finetune_00065_.png",
        "title": "Jenna ortega",
        "repo": "punzel/flux_jenna_ortega",
        "weights": "flux_jenna_ortega.safetensors",
        "trigger_word": "Jenna ortega"
    }, 
    #43
    {
        "image": "https://huggingface.co/Wakkamaruh/balatro-poker-cards/resolve/main/samples/01.png",
        "title": "Poker Cards",
        "repo": "Wakkamaruh/balatro-poker-cards",
        "weights": "balatro-poker-cards.safetensors",
        "trigger_word": "balatrocard"
    }, 
    #44
    {
        "image": "https://huggingface.co/lichorosario/flux-cubist-cartoon/resolve/main/samples/albert-einstein.png",
        "title": "Cubist Cartoon",
        "repo": "lichorosario/flux-cubist-cartoon",
        "weights": "lora.safetensors",
        "trigger_word": "CBSTCRTN"
    },
    #45
     {
        "image": "https://huggingface.co/iliketoasters/miniature-people/resolve/main/images/1757-over%20the%20shoulder%20shot%2C%20raw%20photo%2C%20a%20min-fluxcomfy-orgflux1-dev-fp8-128443497-converted.png",
        "title": "Miniature People",
        "repo": "iliketoasters/miniature-people",
        "trigger_word": "miniature people"
    },   
    #46
    {
        "image": "https://huggingface.co/ampp/rough-kids-illustrations/resolve/main/samples/1725115106736__000001000_0.jpg",
        "title": "kids Illustrations",
        "repo": "ampp/rough-kids-illustrations",
        "weights": "rough-kids-illustrations.safetensors",
        "trigger_word": "r0ughkids4rt"  
    },
    #47
    {
        "image": "https://huggingface.co/lichorosario/flux-lora-tstvctr/resolve/main/images/example_mo3jx93o6.png",
        "title": "TSTVCTR Cartoon",
        "repo": "lichorosario/flux-lora-tstvctr",
        "weights": "lora.safetensors",
        "trigger_word": "TSTVCTR cartoon illustration" 
    },
    #48
    {
        "image": "https://huggingface.co/lichorosario/flux-lora-gliff-tosti-vector-no-captions-2500s/resolve/main/images/example_i6h6fi9sq.png",
        "title": "Tosti Vector",
        "repo": "lichorosario/flux-lora-gliff-tosti-vector-no-captions-2500s",
        "weights": "flux_dev_tosti_vector_without_captions_000002500.safetensors",
        "trigger_word": ""     
    },
    #49
    {
        "image": "https://huggingface.co/AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion/resolve/main/Trashy.png",
        "title": "Propaganda Poster",
        "repo": "AlekseyCalvin/Propaganda_Poster_Schnell_by_doctor_diffusion",
        "weights": "propaganda_schnell_v1.safetensors",
        "trigger_word": "propaganda poster"          
    },
    #50
    {
        "image": "https://huggingface.co/WizWhite/Wiz-PunchOut_Ringside_Portrait/resolve/main/images/punch0ut__ringside_pixel_portrait_depicting_chris_brown_wearing_a_veil__moonstone_gray_background_with_white_ropes___1923906484.png",
        "title": "Ringside Portrait",
        "repo": "WizWhite/Wiz-PunchOut_Ringside_Portrait",
        "trigger_word": "punch0ut, ringside pixel portrait depicting"     
    },
    #51
    {
        "image": "https://huggingface.co/glif-loradex-trainer/kklors_flux_dev_long_exposure/resolve/main/samples/1729016926778__000003000_3.jpg",
        "title": "Long Exposure",
        "repo": "glif-loradex-trainer/kklors_flux_dev_long_exposure",
        "weights": "flux_dev_long_exposure.safetensors",
        "trigger_word": "LE"     
    },
    #52
    {
        "image": "https://huggingface.co/DamarJati/streetwear-flux/resolve/main/img/79e891f9-ceb8-4f8a-a51d-bb432789d037.jpeg",
        "title": "Street Wear",
        "repo": "DamarJati/streetwear-flux",
        "weights": "Streetwear.safetensors",
        "trigger_word": "Handling Information Tshirt template"      
    },
    #53
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-NTFv4-Designs-LoRA/resolve/main/images/6.png",
        "title": "NFT V4",
        "repo": "strangerzonehf/Flux-NTFv4-Designs-LoRA",
        "weights": "NFTv4.safetensors",
        "trigger_word": "NFT V4" 
    },
    #54
    {
        "image": "https://huggingface.co/multimodalart/product-design/resolve/main/images/example_vgv87rlfl.png",
        "title": "Product Design",
        "repo": "multimodalart/product-design",
        "weights": "product-design.safetensors",
        "trigger_word": "product designed by prdsgn"   
    },
    #55
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-LoRA-Flux-Typography-ASCII/resolve/main/images/NNN.png",
        "title": "Typography",
        "repo": "prithivMLmods/Canopus-LoRA-Flux-Typography-ASCII",
        "weights": "Typography.safetensors",
        "trigger_word": "Typography, ASCII Art"          
    },
    #56
    {
        "image": "https://huggingface.co/mateo-19182/mosoco/resolve/main/samples/1725714834007__000002000_0.jpg",
        "title": "Mosoco",
        "repo": "mateo-19182/mosoco",
        "weights": "mosoco.safetensors",
        "trigger_word": "moscos0"           
    },
    #57
    {
        "image": "https://huggingface.co/jakedahn/flux-latentpop/resolve/main/images/2.webp",
        "title": "Latent Pop",
        "repo": "jakedahn/flux-latentpop",
        "weights": "lora.safetensors",
        "trigger_word": "latentpop"          
    },
    #58
    {
        "image": "https://huggingface.co/glif-loradex-trainer/ddickinson_dstyl3xl/resolve/main/samples/1728556571974__000001500_2.jpg",
        "title": "Dstyl3xl",
        "repo": "glif-loradex-trainer/ddickinson_dstyl3xl",
        "weights": "dstyl3xl.safetensors",
        "trigger_word": "in the style of dstyl3xl"       
    },
    #59
    {
        "image": "https://huggingface.co/TDN-M/RetouchFLux/resolve/main/images/496f0680-0158-4f37-805d-d227c1a08a7b.png",
        "title": "Retouch FLux",
        "repo": "TDN-M/RetouchFLux",
        "weights": "TDNM_Retouch.safetensors",
        "trigger_word": "luxury, enhance, hdr"        
    },
    #60
    {
        "image": "https://huggingface.co/glif/anime-blockprint-style/resolve/main/images/glif-block-print-anime-flux-dev-araminta-k-lora-araminta-k-e35k8xqsrb8dtq2qcv4gsr3z.jpg",
        "title": "Block Print",
        "repo": "glif/anime-blockprint-style",
        "weights": "bwmanga.safetensors",
        "trigger_word": "blockprint style"              
    },
    #61
    {
        "image": "https://huggingface.co/renderartist/weirdthingsflux/resolve/main/images/3D__02303_.png",
        "title": "Weird Things Flux",
        "repo": "renderartist/weirdthingsflux",
        "weights": "Weird_Things_Flux_v1_renderartist.safetensors",
        "trigger_word": "w3irdth1ngs, illustration"          
    },
    #62
    {
        "image": "https://replicate.delivery/yhqm/z7f2OBcvga07dCoJ4FeRGZCbE5PvipLhogPhEeU7BazIg5lmA/out-0.webp",
        "title": "Replicate Flux LoRA",
        "repo": "lucataco/ReplicateFluxLoRA",
        "weights": "flux_train_replicate.safetensors",
        "trigger_word": "TOK"       
    },
    #63
    {
        "image": "https://huggingface.co/alvdansen/haunted_linework_flux/resolve/main/images/ComfyUI_00755_.png",
        "title": "Linework",
        "repo": "alvdansen/haunted_linework_flux",
        "weights": "hauntedlinework_flux_araminta_k.safetensors",
        "trigger_word": "hntdlnwrk style"         
    },
    #64
    {
        "image": "https://huggingface.co/fofr/flux-cassette-futurism/resolve/main/images/example_qgry9jnkj.png",
        "title": "Cassette Futurism",
        "repo": "fofr/flux-cassette-futurism",
        "weights": "lora.safetensors",
        "trigger_word": "cassette futurism"   
    },
    #65
    {
        "image": "https://huggingface.co/Wadaka/Mojo_Style_LoRA/resolve/main/Samples/Sample2.png",
        "title": "Mojo Style",
        "repo": "Wadaka/Mojo_Style_LoRA",
        "weights": "Mojo_Style_LoRA.safetensors",
        "trigger_word": "Mojo_Style" 
        
    },
    #66
    {
        "image": "https://huggingface.co/Norod78/JojosoStyle-flux-lora/resolve/main/samples/1725244218477__000004255_1.jpg",
        "title": "Jojoso Style",
        "repo": "Norod78/JojosoStyle-flux-lora",
        "weights": "JojosoStyle_flux_lora.safetensors",
        "trigger_word": "JojosoStyle"        
    },
    #67
    {
        "image": "https://huggingface.co/Chunte/flux-lora-Huggieverse/resolve/main/images/Happy%20star.png",
        "title": "Huggieverse",
        "repo": "Chunte/flux-lora-Huggieverse",
        "weights": "lora.safetensors",
        "trigger_word": "HGGRE"          
    },
    #68
    {
        "image": "https://huggingface.co/diabolic6045/Flux_Wallpaper_Lora/resolve/main/images/example_hjp51et93.png",
        "title": "Wallpaper LoRA",
        "repo": "diabolic6045/Flux_Wallpaper_Lora",
        "weights": "tost-2024-09-20-07-35-44-wallpap3r5.safetensors",
        "trigger_word": "wallpap3r5"        
    },
    #69
    {
        "image": "https://huggingface.co/bingbangboom/flux_geopop/resolve/main/extras/5.png",
        "title": "Geo Pop",
        "repo": "bingbangboom/flux_geopop",
        "weights": "geopop_NWGMTRCPOPV01.safetensors",
        "trigger_word": "illustration in the style of NWGMTRCPOPV01"       
    },
    #70
    {
        "image": "https://huggingface.co/bingbangboom/flux_colorscape/resolve/main/images/4.jpg",
        "title": "Colorscape",
        "repo": "bingbangboom/flux_colorscape",
        "weights": "flux_colorscape.safetensors",
        "trigger_word": "illustration in the style of ASstyle001" 
    },
    #71
    {
        "image": "https://huggingface.co/dvyio/flux-lora-thermal-image/resolve/main/images/WROSaNNU4-Gw0r5QoBRjf_f164ffa4f0804e68bad1d06d30deecfa.jpg",
        "title": "Thermal Image",
        "repo": "dvyio/flux-lora-thermal-image",
        "weights": "79b5004c57ef4c4390dead1c65977bbb_pytorch_lora_weights.safetensors",
        "trigger_word": "thermal image in the style of THRML" 
    },
    #72
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-Clothing-Flux-LoRA/resolve/main/images/333.png",
        "title": "Clothing Flux",
        "repo": "prithivMLmods/Canopus-Clothing-Flux-LoRA",
        "weights": "Canopus-Clothing-Flux-Dev-Florence2-LoRA.safetensors",
        "trigger_word": "Hoodie, Clothes, Shirt, Pant"    
    },
    #73
    {
        "image": "https://huggingface.co/dvyio/flux-lora-stippled-illustration/resolve/main/images/57FPpbu74QTV45w6oNOtZ_26832270585f456c99e4a98b1c073745.jpg",
        "title": "Stippled Illustration",
        "repo": "dvyio/flux-lora-stippled-illustration",
        "weights": "31984be602a04a1fa296d9ccb244fb29_pytorch_lora_weights.safetensors",
        "trigger_word": "stippled illustration in the style of STPPLD"          
    },
    #74
    {
        "image": "https://huggingface.co/wayned/fruitlabels/resolve/main/images/ComfyUI_03969_.png",
        "title": "Fruitlabels",
        "repo": "wayned/fruitlabels",
        "weights": "fruitlabels2.safetensors",
        "trigger_word": "fruit labels"
        
    },
    #75
    {
        "image": "https://huggingface.co/punzel/flux_margot_robbie/resolve/main/images/ComfyUI_Flux_Finetune_00142_.png",
        "title": "Margot Robbie",
        "repo": "punzel/flux_margot_robbie",
        "weights": "flux_margot_robbie.safetensors",
        "trigger_word": ""
    },
    #76
    {
        "image": "https://huggingface.co/diabolic6045/Formula1_Lego_Lora/resolve/main/images/example_502kcuiba.png",
        "title": "Formula 1 Lego",
        "repo": "punzel/flux_margot_robbie",
        "weights": "tost-2024-09-20-09-58-33-f1leg0s.safetensors",
        "trigger_word": "f1leg0s"    
    },
    #77
    {
        "image": "https://huggingface.co/glif/Brain-Melt-Acid-Art/resolve/main/images/IMG_0832.png",
        "title": "Melt Acid",
        "repo": "glif/Brain-Melt-Acid-Art",
        "weights": "Brain_Melt.safetensors",
        "trigger_word": "in an acid surrealism style, maximalism"  
    },
    #78
    {
        "image": "https://huggingface.co/jeremytai/enso-zen/resolve/main/images/example_a0iwdj5lu.png",
        "title": "Enso",
        "repo": "jeremytai/enso-zen",
        "weights": "enso-zen.safetensors",
        "trigger_word": "enso"       
    },
    #79
    {
        "image": "https://huggingface.co/veryVANYA/opus-ascii-flux/resolve/main/31654332.jpeg",
        "title": "Opus Ascii",
        "repo": "veryVANYA/opus-ascii-flux",
        "weights": "flux_opus_ascii.safetensors",
        "trigger_word": "opus_ascii" 
    },
    #80
    {
        "image": "https://huggingface.co/crystantine/cybrpnkz/resolve/main/images/example_plyxk0lej.png",
        "title": "Cybrpnkz",
        "repo": "crystantine/cybrpnkz",
        "weights": "cybrpnkz.safetensors",
        "trigger_word": "architecture style of CYBRPNKZ"
    },
    #81
    {
        "image": "https://huggingface.co/fyp1/pattern_generation/resolve/main/images/1727560066052__000001000_7.jpg",
        "title": "Pattern Generation",
        "repo": "fyp1/pattern_generation",
        "weights": "flux_dev_finetune.safetensors",
        "trigger_word": "pattern"
    },
    #82
    {
        "image": "https://huggingface.co/TheAwakenOne/caricature/resolve/main/sample/caricature_000900_03_20241007143412.png",
        "title": "Caricature",
        "repo": "TheAwakenOne/caricature",
        "weights": "caricature.safetensors",
        "trigger_word": "CCTUR3"      
    },
    #83
    {
        "image": "https://huggingface.co/davidrd123/Flux-MoonLanding76-Replicate/resolve/main/images/example_6adktoq5m.png",
        "title": "MoonLanding 76",
        "repo": "davidrd123/Flux-MoonLanding76-Replicate",
        "weights": "lora.safetensors",
        "trigger_word": "m00nl4nd1ng"          
    },
    #84
    {
        "image": "https://huggingface.co/Purz/neon-sign/resolve/main/33944768.jpeg",
        "title": "Neon",
        "repo": "Purz/neon-sign",
        "weights": "purz-n30n_51gn.safetensors",
        "trigger_word": "n30n_51gn"        
    },
    #85
    {
        "image": "https://huggingface.co/WizWhite/wizard-s-vintage-sardine-tins/resolve/main/27597694.jpeg",
        "title": "Vintage Sardine Tins",
        "repo": "WizWhite/wizard-s-vintage-sardine-tins",
        "weights": "Wiz-SardineTins_Flux.safetensors",
        "trigger_word": "Vintage Sardine Tin, Tinned Fish, vintage xyz tin"    
    },
    #86
    {
        "image": "https://huggingface.co/TheAwakenOne/mtdp-balloon-character/resolve/main/sample/mtdp-balloon-character_000200_01_20241014221110.png",
        "title": "Float Ballon Character",
        "repo": "TheAwakenOne/mtdp-balloon-character",
        "weights": "mtdp-balloon-character.safetensors",
        "trigger_word": "FLOAT"   
    },
    #87
    {
        "image": "https://huggingface.co/glif/golden-haggadah/resolve/main/images/6aca6403-ecd6-4216-a66a-490ae25ff1b2.jpg",
        "title": "Golden Haggadah",
        "repo": "glif/golden-haggadah",
        "weights": "golden_haggadah.safetensors",
        "trigger_word": "golden haggadah style"        
    },
    #88
    {
        "image": "https://huggingface.co/glif-loradex-trainer/usernametaken420__oz_ftw_balaclava/resolve/main/samples/1729278631255__000001500_1.jpg",
        "title": "Ftw Balaclava",
        "repo": "glif-loradex-trainer/usernametaken420__oz_ftw_balaclava",
        "weights": "oz_ftw_balaclava.safetensors",
        "trigger_word": "ftw balaclava"          
    },
    #89
    {
        "image": "https://huggingface.co/AlloReview/flux-lora-undraw/resolve/main/images/Flux%20Lora%20Undraw%20Prediction.webp",
        "title": "Undraw",
        "repo": "AlloReview/flux-lora-undraw",
        "weights": "lora.safetensors",
        "trigger_word": "in the style of UndrawPurple"      
    },
    #90
    {
        "image": "https://huggingface.co/Disra/lora-anime-test-02/resolve/main/assets/image_0_0.png",
        "title": "Anime Test",
        "repo": "Disra/lora-anime-test-02",
        "weights": "pytorch_lora_weights.safetensors",
        "trigger_word": "anime" 
    },
    #91
    {
        "image": "https://huggingface.co/wanghaofan/Black-Myth-Wukong-FLUX-LoRA/resolve/main/images/7d0ac495a4d5e4a3a30df25f08379a3f956ef99e1dc3e252fc1fca3a.jpg",
        "title": "Black Myth Wukong",
        "repo": "wanghaofan/Black-Myth-Wukong-FLUX-LoRA",
        "weights": "pytorch_lora_weights.safetensors",
        "trigger_word": "wukong"         
    },
    #92
    {
        "image": "https://huggingface.co/nerijs/pastelcomic-flux/resolve/main/images/4uZ_vaYg-HQnfa5D9gfli_38bf3f95d8b345e5a9bd42d978a15267.png",
        "title": "Pastelcomic",
        "repo": "nerijs/pastelcomic-flux",
        "weights": "pastelcomic_v1.safetensors",
        "trigger_word": ""            
    },
    #93
    {
        "image": "https://huggingface.co/RareConcepts/Flux.1-dev-LoKr-Moonman/resolve/main/assets/image_6_0.png",
        "title": "Moonman",
        "repo": "RareConcepts/Flux.1-dev-LoKr-Moonman",
        "weights": "pytorch_lora_weights.safetensors",
        "trigger_word": "moonman"            
    },
    #94
    {
        "image": "https://huggingface.co/martintomov/ascii-flux-v1/resolve/main/images/0af53645-ddcc-4803-93c8-f7e43f6fbbd1.jpeg",
        "title": "Ascii Flux",
        "repo": "martintomov/ascii-flux-v1",
        "weights": "ascii-art-v1.safetensors",
        "trigger_word": "ASCII art"          
    },
    #95
    {
        "image": "https://huggingface.co/Omarito2412/Stars-Galaxy-Flux/resolve/main/images/25128409.jpeg",
        "title": "Ascii Flux",
        "repo": "Omarito2412/Stars-Galaxy-Flux",
        "weights": "Stars_Galaxy_Flux.safetensors",
        "trigger_word": "mlkwglx"       
    },
    #96
    {
        "image": "https://huggingface.co/brushpenbob/flux-pencil-v2/resolve/main/26193927.jpeg",
        "title": "Pencil V2",
        "repo": "brushpenbob/flux-pencil-v2",
        "weights": "Flux_Pencil_v2_r1.safetensors",
        "trigger_word": "evang style"           
    },
    #97
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch/resolve/main/images/1f20519208cef367af2fda8d91ddbba674f39b097389d12ee25b4cb1.jpg",
        "title": "Children Simple Sketch",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Children-Simple-Sketch",
        "weights": "FLUX-dev-lora-children-simple-sketch.safetensors",
        "trigger_word": "sketched style"            
    },
    #98
    {
        "image": "https://huggingface.co/victor/contemporarink/resolve/main/images/example_hnqc22urm.png",
        "title": "Contemporarink",
        "repo": "victor/contemporarink",
        "weights": "inky-colors.safetensors",
        "trigger_word": "ECACX"         
    },
    #99
    {
        "image": "https://huggingface.co/wavymulder/OverlordStyleFLUX/resolve/main/imgs/ComfyUI_00668_.png",
        "title": "OverlordStyle",
        "repo": "wavymulder/OverlordStyleFLUX",
        "weights": "ovld_style_overlord_wavymulder.safetensors",
        "trigger_word": "ovld style anime"          
    },
    #100
    {
        "image": "https://huggingface.co/marceloxp/canny-quest/resolve/main/26676266.jpeg",
        "title": "Canny quest",
        "repo": "marceloxp/canny-quest",
        "weights": "Canny_Quest-000004.safetensors",
        "trigger_word": "blonde, silver silk dress, perfectly round sunglasses, pearl necklace"         
    },
    #101
    {
        "image": "https://huggingface.co/busetolunay/building_flux_lora_v1/resolve/main/samples/1725469125185__000001250_2.jpg",
        "title": "Building Flux",
        "repo": "busetolunay/building_flux_lora_v1",
        "weights": "building_flux_lora_v4.safetensors",
        "trigger_word": "a0ce"        
    },
    #102
    {
        "image": "https://huggingface.co/Omarito2412/Tinker-Bell-Flux/resolve/main/images/9e9e7eda-3ddf-467a-a7f8-6d8e3ef80cd0.png",
        "title": "Tinker Bell Flux",
        "repo": "Omarito2412/Tinker-Bell-Flux",
        "weights": "TinkerBellV2-FLUX.safetensors",
        "trigger_word": "TinkerWaifu, blue eyes, single hair bun"        
    },
    #103
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-playful-metropolis/resolve/main/images/3e9265312b3b726c224a955ec9254a0f95c2c8b78ce635929183a075.jpg",
        "title": "Playful Metropolis",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-playful-metropolis",
        "weights": "FLUX-dev-lora-playful_metropolis.safetensors",
        "trigger_word": ""        
    },
    #104
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-Character-Polygon-LoRA/resolve/main/images/1000.webp",
        "title": "Character Polygon",
        "repo": "prithivMLmods/Castor-Character-Polygon-Flux-LoRA",
        "weights": "Castor-Character-Polygon-LoRA.safetensors",
        "trigger_word": "3D Polygon"        
    },
    #105
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-Gta6-Theme-Flux-LoRA/resolve/main/images/gta1.webp",
        "title": "GTA 6 Theme",
        "repo": "prithivMLmods/Castor-Gta6-Theme-Flux-LoRA",
        "weights": "Gta6.safetensors",
        "trigger_word": "GTA 6 Theme, World of GTA 6"        
    },
    #106
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-Concept-Gta6-Character-Design/resolve/main/images/L3.webp",
        "title": "GTA Character Concept",
        "repo": "prithivMLmods/Castor-Flux-Concept-Gta6-Character-Design",
        "weights": "Gta6-Concept-Charecter.safetensors",
        "trigger_word": "Jason, Lucia, GTA 6"          
    },
    #107
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA/resolve/main/images/S1.png",
        "title": "3D Sketchfab",
        "repo": "prithivMLmods/Castor-3D-Sketchfab-Flux-LoRA",
        "weights": "Castor-3D-Sketchfab-Flux-LoRA.safetensors",
        "trigger_word": "3D Sketchfab"    
    },
    #108
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-Collage-Dim-Flux-LoRA/resolve/main/images/C1.webp",
        "title": "In Image Collage",
        "repo": "prithivMLmods/Castor-Collage-Dim-Flux-LoRA",
        "weights": "Castor-Collage-Dim-Flux-LoRA.safetensors",
        "trigger_word": "collage"          
    },
    #109
    {
        "image": "https://huggingface.co/brushpenbob/flux-midjourney-anime/resolve/main/25439344.jpeg",
        "title": "Anime Journey",
        "repo": "brushpenbob/flux-midjourney-anime",
        "weights": "FLUX_MidJourney_Anime.safetensors",
        "trigger_word": "egmid"      
    },
    #110
    {
        "image": "https://huggingface.co/glif-loradex-trainer/maxxd4240_minimalistPastel/resolve/main/samples/1727255690613__000002500_0.jpg",
        "title": "Min Pastel",
        "repo": "glif-loradex-trainer/maxxd4240_minimalistPastel",
        "weights": "minimalistPastel.safetensors",
        "trigger_word": "minimalistPastel"         
    },
    #111
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-Red-Dead-Redemption-2-Flux-LoRA/resolve/main/images/rdr12.webp",
        "title": "RDR2",
        "repo": "prithivMLmods/Castor-Red-Dead-Redemption-2-Flux-LoRA",
        "weights": "Castor-Red-Dead-Redemption-2-Flux-LoRA.safetensors",
        "trigger_word": "Red Dead Redemption 2"     
    },
    #112
    {
        "image": "https://huggingface.co/WizWhite/wizard-s-paper-model-universe/resolve/main/35746354.jpeg",
        "title": "Paper Model",
        "repo": "WizWhite/wizard-s-paper-model-universe",
        "weights": "Wiz-Paper_Model_Universe.safetensors",
        "trigger_word": "A paper model" 
    },
    #113
    {
        "image": "https://huggingface.co/renderartist/retrocomicflux/resolve/main/images/ComfyUI_temp_ipugi_00040_.png",
        "title": "Retrocomic Flux",
        "repo": "renderartist/retrocomicflux",
        "weights": "Retro_Comic_Flux_v1_renderartist.safetensors",
        "trigger_word": "comic book panel"         
    },
    #114
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-Happy-Halloween-Flux-LoRA/resolve/main/images/hw1.webp",
        "title": "Halloween Flux",
        "repo": "prithivMLmods/Castor-Happy-Halloween-Flux-LoRA",
        "weights": "Castor-Happy-Halloween-Flux-LoRA.safetensors",
        "trigger_word": "happy halloween" 
    },
    #115
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-3D-Portrait-Flux-LoRA/resolve/main/images/1.webp",
        "title": "Castor-3D-Portrait",
        "repo": "prithivMLmods/Castor-3D-Portrait-Flux-LoRA",
        "weights": "Castor-3D-Portrait-Flux-LoRA.safetensors",
        "trigger_word": "3D Portrait" 
    },
    #116
    {
        "image": "https://huggingface.co/renderartist/coloringbookflux/resolve/main/images/ComfyUI_09731_.png",
        "title": "Coloring book flux",
        "repo": "renderartist/coloringbookflux",
        "weights": "c0l0ringb00k_Flux_v1_renderartist.safetensors",
        "trigger_word": "c0l0ringb00k, coloring book, coloring book page"        
    },
    #117
    {
        "image": "https://huggingface.co/prithivMLmods/Uncoloured-Polygon-Flux-LoRA/resolve/main/images/1.webp",
        "title": "Uncoloured Polygon",
        "repo": "prithivMLmods/Uncoloured-Polygon-Flux-LoRA",
        "weights": "Uncoloured-3D-Polygon.safetensors",
        "trigger_word": "uncoloured polygon"           
    },
    #118
    {
        "image": "https://huggingface.co/prithivMLmods/Past-Present-Deep-Mix-Flux-LoRA/resolve/main/images/PP3.webp",
        "title": "Past Present Mix",
        "repo": "prithivMLmods/Past-Present-Deep-Mix-Flux-LoRA",
        "weights": "Past-Present-Deep-Mix-Flux-LoRA.safetensors",
        "trigger_word": "Mixing Past and Present"      
    },
    #119
    {
        "image": "https://huggingface.co/gokaygokay/Flux-Double-Exposure-LoRA/resolve/main/images/image3.jpg",
        "title": "Double Exposure",
        "repo": "gokaygokay/Flux-Double-Exposure-LoRA",
        "weights": "double_exposure.safetensors",
        "trigger_word": "dblxpsr"       
    },
    #120
    {
        "image": "https://huggingface.co/gokaygokay/Flux-Seamless-Texture-LoRA/resolve/main/images/image3.jpg",
        "title": "Seamless Texture",
        "repo": "gokaygokay/Flux-Seamless-Texture-LoRA",
        "weights": "seamless_texture.safetensors",
        "trigger_word": "smlstxtr"          
    },
    #121
    {
        "image": "https://huggingface.co/prithivMLmods/Mockup-Texture-Flux-LoRA/resolve/main/images/MU1.webp",
        "title": "Mockup Texture",
        "repo": "prithivMLmods/Mockup-Texture-Flux-LoRA",
        "weights": "Mockup-Texture.safetensors",
        "trigger_word": "Mockup"      
    },
    #122
    {
        "image": "https://huggingface.co/prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA/resolve/main/images/c2.webp",
        "title": "Tarot Cards",
        "repo": "prithivMLmods/Ton618-Tarot-Cards-Flux-LoRA",
        "weights": "Tarot-card.safetensors",
        "trigger_word": "Tarot card"          
    },
    #123
    {
        "image": "https://huggingface.co/prithivMLmods/Ton618-Amxtoon-Flux-LoRA/resolve/main/images/am1.webp",
        "title": "Amxtoon",
        "repo": "prithivMLmods/Ton618-Amxtoon-Flux-LoRA",
        "weights": "Amxtoon.safetensors",
        "trigger_word": "Amxtoon"          
    },
    #124
    {
        "image": "https://huggingface.co/prithivMLmods/Ton618-Epic-Realism-Flux-LoRA/resolve/main/images/ep3.png",
        "title": "Epic Realism",
        "repo": "prithivMLmods/Ton618-Epic-Realism-Flux-LoRA",
        "weights": "Epic-Realism-Unpruned.safetensors",
        "trigger_word": "Epic Realism"         
    },
    #125
    {
        "image": "https://huggingface.co/bingbangboom/flux-mixReality/resolve/main/images/3.jpg",
        "title": "Mixed Reality",
        "repo": "bingbangboom/flux-mixReality",
        "weights": "HLFILSTHLFPHTO_000002500.safetensors",
        "trigger_word": "in the style of HLFILSTHLFPHTO"          
    },
    #126
    {
        "image": "https://huggingface.co/sWizad/pokemon-trainer-sprites-pixelart-flux/resolve/main/26578919.jpeg",
        "title": "Pixelart",
        "repo": "sWizad/pokemon-trainer-sprites-pixelart-flux",
        "weights": "pktrainer_F1-v1-0.safetensors",
        "trigger_word": "pixel image of, pixel art"        
    },
    #127
    {
        "image": "https://huggingface.co/bingbangboom/flux_colorscape/resolve/main/images/2.jpg",
        "title": "Colorscape",
        "repo": "bingbangboom/flux_colorscape",
        "weights": "flux_colorscape.safetensors",
        "trigger_word": "illustration in the style of ASstyle001"         
    },
    #128
    {
        "image": "https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Modern_Pixel_art/resolve/main/images/c363192f-5fa0-4539-8295-b8d9e3e96747.jpeg",
        "title": "Modern Pixel art",
        "repo": "UmeAiRT/FLUX.1-dev-LoRA-Modern_Pixel_art",
        "weights": "ume_modern_pixelart.safetensors",
        "trigger_word": "umempart"             
    },
    #129
    {
        "image": "https://huggingface.co/prithivMLmods/Ton618-Only-Stickers-Flux-LoRA/resolve/main/images/222.png",
        "title": "Sticker",
        "repo": "prithivMLmods/Ton618-Only-Stickers-Flux-LoRA",
        "weights": "only-stickers.safetensors",
        "trigger_word": "Only Sticker"          
    },
    #130
    {
        "image": "https://huggingface.co/prithivMLmods/Ton618-Space-Wallpaper-LoRA/resolve/main/images/222.png",
        "title": "Space Wallpaper",
        "repo": "prithivMLmods/Ton618-Space-Wallpaper-LoRA",
        "weights": "space-wallpaper-xl.safetensor",
        "trigger_word": "Space Wallpaper"        
    },
    #131
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-Pixar-3D-Flux-LoRA/resolve/main/images/11111.png",
        "title": "Pixar 3D",
        "repo": "prithivMLmods/Canopus-Pixar-3D-Flux-LoRA",
        "weights": "Canopus-Pixar-3D-FluxDev-LoRA.safetensors",
        "trigger_word": "Pixar 3D"         
    },
    #132
    {
        "image": "https://huggingface.co/prithivMLmods/EBook-Creative-Cover-Flux-LoRA/resolve/main/images/E2.png",
        "title": "EBook Cover",
        "repo": "prithivMLmods/EBook-Creative-Cover-Flux-LoRA",
        "weights": "EBook-Cover.safetensors",
        "trigger_word": "EBook Cover"         
    },
    #133
    {
        "image": "https://huggingface.co/prithivMLmods/Minimal-Futuristic-Flux-LoRA/resolve/main/images/MF3.png",
        "title": "Minimal Futuristic",
        "repo": "prithivMLmods/Minimal-Futuristic-Flux-LoRA",
        "weights": "Minimal-Futuristic.safetensors",
        "trigger_word": "Minimal Futuristic"         
    },
    #134
    {
        "image": "https://huggingface.co/prithivMLmods/Seamless-Pattern-Design-Flux-LoRA/resolve/main/images/SP1.png",
        "title": "Seamless Pattern",
        "repo": "prithivMLmods/Seamless-Pattern-Design-Flux-LoRA",
        "weights": "Seamless-Pattern-Design.safetensors",
        "trigger_word": "Seamless Pattern Design"          
    },
    #135
    {
        "image": "https://huggingface.co/prithivMLmods/Logo-Design-Flux-LoRA/resolve/main/images/LD1.png",
        "title": "Logo Design",
        "repo": "prithivMLmods/Logo-Design-Flux-LoRA",
        "weights": "Logo-design.safetensors",
        "trigger_word": "Logo Design"         
    },
    #136
    {
        "image": "https://huggingface.co/prithivMLmods/Coloring-Book-Flux-LoRA/resolve/main/images/EB1.png",
        "title": "Coloring Book",
        "repo": "prithivMLmods/Coloring-Book-Flux-LoRA",
        "weights": "coloring-book.safetensors",
        "trigger_word": "Coloring Book"    
    },
    #137
    {
        "image": "https://huggingface.co/prithivMLmods/Intense-Red-Flux-LoRA/resolve/main/images/IR1.png",
        "title": "Intense Red",
        "repo": "prithivMLmods/Intense-Red-Flux-LoRA",
        "weights": "Intense-Red.safetensors",
        "trigger_word": "Intense Red"         
    },
    #138
    {
        "image": "https://huggingface.co/prithivMLmods/Glowing-Body-Flux-LoRA/resolve/main/images/GB3.png",
        "title": "Glowing Body Flux",
        "repo": "prithivMLmods/Glowing-Body-Flux-LoRA",
        "weights": "Glowing-Body.safetensors",
        "trigger_word": "Glowing Body"          
    },
    #139
    {
        "image": "https://huggingface.co/prithivMLmods/Electric-Blue-Flux-LoRA/resolve/main/images/EB3.png",
        "title": "Electric Blue",
        "repo": "prithivMLmods/Electric-Blue-Flux-LoRA",
        "weights": "Electric-Blue.safetensors",
        "trigger_word": "Electric Blue"         
    },
    #140
    {
        "image": "https://huggingface.co/prithivMLmods/Clouds-Illusion-Flux-LoRA/resolve/main/images/CI2.png",
        "title": "Clouds Illusion",
        "repo": "prithivMLmods/Clouds-Illusion-Flux-LoRA",
        "weights": "Clouds-Illusion.safetensors",
        "trigger_word": "Clouds Illusion"        
    },
    #141
    {
        "image": "https://huggingface.co/prithivMLmods/Digital-Yellow-Flux-LoRA/resolve/main/images/DY3.png",
        "title": "Digital Yellow",
        "repo": "prithivMLmods/Digital-Yellow-Flux-LoRA",
        "weights": "Digital-Yellow.safetensors",
        "trigger_word": "Digital Yellow"          
    },
    #142
    {
        "image": "https://huggingface.co/cfahlgren1/flux-qwen-capybara/resolve/main/images/example_72ao6twvk.png",
        "title": "Flux Qwen Capybara",
        "repo": "cfahlgren1/flux-qwen-capybara",
        "weights": "flux-qwen-capybara.safetensors",
        "trigger_word": "QWENCAPY"          
    },
    #143
    {
        "image": "https://huggingface.co/dasdsff/PleinAirArt/resolve/main/images/e7499ccc-7504-4086-842f-275a5428ef0e.jpg",
        "title": "Plein Air Art ",
        "repo": "dasdsff/PleinAirArt",
        "weights": "PleinAir_000002500.safetensors",
        "trigger_word": "P1e!n"        
    },
    #144
    {
        "image": "https://huggingface.co/prithivMLmods/Orange-Chroma-Flux-LoRA/resolve/main/images/OC1.png",
        "title": "Orange Chroma",
        "repo": "prithivMLmods/Orange-Chroma-Flux-LoRA",
        "weights": "Orange-Chroma.safetensors",
        "trigger_word": "Orange Chroma"        
    },
    #145
    {
        "image": "https://huggingface.co/prithivMLmods/Lime-Green-Flux-LoRA/resolve/main/images/LM1.png",
        "title": "Lime Green",
        "repo": "prithivMLmods/Lime-Green-Flux-LoRA",
        "weights": "Lime-Green.safetensors",
        "trigger_word": "Lime Green"         
    },
    #146
    {
        "image": "https://huggingface.co/prithivMLmods/Fractured-Line-Flare/resolve/main/images/FS1.png",
        "title": "Line Flare",
        "repo": "prithivMLmods/Fractured-Line-Flare",
        "weights": "Fractured-Line-Flare.safetensors",
        "trigger_word": "Fractured Line Flare"        
    },
    #147
    {
        "image": "https://huggingface.co/prithivMLmods/Golden-Dust-Flux-LoRA/resolve/main/images/GD2.png",
        "title": "Golden Dust",
        "repo": "prithivMLmods/Golden-Dust-Flux-LoRA",
        "weights": "Golden-Dust.safetensors",
        "trigger_word": "Golden Dust"         
    },
    #148
    {
        "image": "https://huggingface.co/prithivMLmods/Castor-Dramatic-Neon-Flux-LoRA/resolve/main/images/DN2.webp",
        "title": "Dramatic Neon",
        "repo": "prithivMLmods/Castor-Dramatic-Neon-Flux-LoRA",
        "weights": "Dramatic-Neon-Flux-LoRA.safetensors",
        "trigger_word": "Dramatic Neon"         
    },
    #149
    {
        "image": "https://huggingface.co/tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator/resolve/main/images/sample7.jpeg",
        "title": "Outfit Generator",
        "repo": "tryonlabs/FLUX.1-dev-LoRA-Outfit-Generator",
        "weights": "outfit-generator.safetensors",
        "trigger_word": "Outfit"         
    },
    #150
    {
        "image": "https://huggingface.co/davisbro/half_illustration/resolve/main/images/example1.webp",
        "title": "Half Illustration",
        "repo": "davisbro/half_illustration",
        "weights": "flux_train_replicate.safetensors",
        "trigger_word": "in the style of TOK"           
    },
    #151
    {
        "image": "https://huggingface.co/bingbangboom/flux_oilscape/resolve/main/extras/3.jpg",
        "title": "Oilscape",
        "repo": "bingbangboom/flux_oilscape",
        "weights": "flux_Oilstyle.safetensors",
        "trigger_word": "in the style of Oilstyle002"           
    },
    #152
    {
        "image": "https://huggingface.co/prithivMLmods/Red-Undersea-Flux-LoRA/resolve/main/images/RU1.png",
        "title": "Red Undersea Flux",
        "repo": "prithivMLmods/Red-Undersea-Flux-LoRA",
        "weights": "Red-Undersea.safetensors",
        "trigger_word": "Red Undersea"          
    },
    #153
    {
        "image": "https://huggingface.co/prithivMLmods/3D-Render-Flux-LoRA/resolve/main/images/3D2.png",
        "title": "3D Render Flux LoRA",
        "repo": "prithivMLmods/3D-Render-Flux-LoRA",
        "weights": "3D_Portrait.safetensors",
        "trigger_word": "3D Portrait, 3d render"        
    },
    #154
    {
        "image": "https://huggingface.co/prithivMLmods/Yellow-Pop-Flux-Dev-LoRA/resolve/main/images/YP1.png",
        "title": "Yellow Pop Flux",
        "repo": "prithivMLmods/Yellow-Pop-Flux-Dev-LoRA",
        "weights": "Yellow_Pop.safetensors",
        "trigger_word": "Yellow Pop"           
    },
    #155
    {
        "image": "https://huggingface.co/prithivMLmods/Purple-Grid-Flux-LoRA/resolve/main/images/PG2.png",
        "title": "Purple Grid Flux",
        "repo": "prithivMLmods/Purple-Grid-Flux-LoRA",
        "weights": "Purple_Grid.safetensors",
        "trigger_word": "Purple Grid"           
    },
    #156
    {
        "image": "https://huggingface.co/prithivMLmods/Dark-Thing-Flux-LoRA/resolve/main/images/DT2.png",
        "title": "Dark Thing Flux",
        "repo": "prithivMLmods/Dark-Thing-Flux-LoRA",
        "weights": "Dark_Creature.safetensors",
        "trigger_word": "Dark Creature"          
    },
    #157
    {
        "image": "https://huggingface.co/prithivMLmods/Shadow-Projection-Flux-LoRA/resolve/main/images/SP2.png",
        "title": "Shadow Projection",
        "repo": "prithivMLmods/Shadow-Projection-Flux-LoRA",
        "weights": "Shadow-Projection.safetensors",
        "trigger_word": "Shadow Projection"           
    },
    #158
    {
        "image": "https://huggingface.co/prithivMLmods/Street-Bokeh-Flux-LoRA/resolve/main/images/SB2.png",
        "title": "Street Bokeh",
        "repo": "prithivMLmods/Street-Bokeh-Flux-LoRA",
        "weights": "Street_Bokeh.safetensors",
        "trigger_word": "Street Bokeh"           
    },
    #159
    {
        "image": "https://huggingface.co/prithivMLmods/Abstract-Cartoon-Flux-LoRA/resolve/main/images/AC2.png",
        "title": "Abstract Cartoon",
        "repo": "prithivMLmods/Abstract-Cartoon-Flux-LoRA",
        "weights": "Abstract-Cartoon.safetensors",
        "trigger_word": "Abstract Cartoon"         
    },
    #160
    {
        "image": "https://huggingface.co/Norod78/CartoonStyle-flux-lora/resolve/main/samples/1725344450635__000003800_1.jpg",
        "title": "Cartoon Style Flux",
        "repo": "Norod78/CartoonStyle-flux-lora",
        "weights": "CartoonStyle_flux_lora.safetensors",
        "trigger_word": ""          
    },
    #161
    {
        "image": "https://huggingface.co/prithivMLmods/Digital-Chaos-Flux-LoRA/resolve/main/images/HDRDC3.webp",
        "title": "HDR Digital Chaos",
        "repo": "prithivMLmods/Digital-Chaos-Flux-LoRA",
        "weights": "HDR-Digital-Chaos.safetensors",
        "trigger_word": "Digital Chaos"         
    },
    #162
    {
        "image": "https://huggingface.co/prithivMLmods/Yellow-Laser-Flux-LoRA/resolve/main/images/YL1.png",
        "title": "Yellow Laser",
        "repo": "prithivMLmods/Yellow-Laser-Flux-LoRA",
        "weights": "Yellow-Laser.safetensors",
        "trigger_word": "Yellow Lasers"        
    },
    #163
    {
        "image": "https://huggingface.co/prithivMLmods/Bold-Shadows-Flux-LoRA/resolve/main/images/BS1.png",
        "title": "Bold Shadows",
        "repo": "prithivMLmods/Bold-Shadows-Flux-LoRA",
        "weights": "Bold-Shadows.safetensors",
        "trigger_word": "Bold Shadows"         
    },
    #164
    {
        "image": "https://huggingface.co/prithivMLmods/Knitted-Character-Flux-LoRA/resolve/main/images/KC1.png",
        "title": "Knitted Character",
        "repo": "prithivMLmods/Knitted-Character-Flux-LoRA",
        "weights": "Knitted-Character.safetensors",
        "trigger_word": "Knitted Character"          
    },
    #165
    {
        "image": "https://huggingface.co/alvdansen/frosting_lane_flux/resolve/main/images/content%20-%202024-08-11T010011.238.jpeg",
        "title": "Frosting Lane",
        "repo": "alvdansen/frosting_lane_flux",
        "trigger_word": "frstingln illustration"
    },
    #166
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Realism-FineDetailed/resolve/main/images/FD2.png",
        "title": "Fine Detailed Character",
        "repo": "prithivMLmods/Flux-Realism-FineDetailed",
        "weights": "Flux-Realism-FineDetailed.safetensors",
        "trigger_word": "Fine Detailed"           
    },
    #167
    {
        "image": "https://huggingface.co/prithivMLmods/Aura-9999/resolve/main/images/A3.png",
        "title": "Aura 9999+",
        "repo": "prithivMLmods/Aura-9999",
        "weights": "Aura-9999.safetensors",
        "trigger_word": "Aura 9999"          
    },
    #168
    {
        "image": "https://huggingface.co/prithivMLmods/Pastel-BG-Flux-LoRA/resolve/main/images/PB2.png",
        "title": "Pastel BG",
        "repo": "prithivMLmods/Pastel-BG-Flux-LoRA",
        "weights": "Pastel-BG.safetensors",
        "trigger_word": "Pastel BG"         
    },
    #169
    {
        "image": "https://huggingface.co/prithivMLmods/Green-Cartoon-Flux-LoRA/resolve/main/images/GC1.png",
        "title": "Green Cartoon",
        "repo": "prithivMLmods/Green-Cartoon-Flux-LoRA",
        "weights": "Green-Cartoon.safetensors",
        "trigger_word": "Green Cartoon"        
    },
    #170
    {
        "image": "https://huggingface.co/prithivMLmods/Retro-Pixel-Flux-LoRA/resolve/main/images/RP1.png",
        "title": "Retro Pixel",
        "repo": "prithivMLmods/Retro-Pixel-Flux-LoRA",
        "weights": "Retro-Pixel.safetensors",
        "trigger_word": "Retro Pixel"           
    },
    #171
    {
        "image": "https://huggingface.co/prithivMLmods/Teen-Outfit/resolve/main/images/TO2.png",
        "title": "Teen Outfit",
        "repo": "prithivMLmods/Teen-Outfit",
        "weights": "Teen-Outfit.safetensors",
        "trigger_word": "Teen Outfit"        
    },
    #172
    {
        "image": "https://huggingface.co/prithivMLmods/CAnime-LoRA/resolve/main/images/CA3.png",
        "title": "CAnime",
        "repo": "prithivMLmods/CAnime-LoRA",
        "weights": "CAnime.safetensors",
        "trigger_word": "CAnime"             
    },
    #173
    {
        "image": "https://huggingface.co/prithivMLmods/Super-Pencil-Flux-LoRA/resolve/main/images/SP1.png",
        "title": "Simple Pencil",
        "repo": "prithivMLmods/Super-Pencil-Flux-LoRA",
        "weights": "Pencil.safetensors",
        "trigger_word": "Simple Pencil"        
    },
    #174
    {
        "image": "https://huggingface.co/martintomov/retrofuturism-flux/resolve/main/images/2e40deba-858e-454f-ae1c-d1ba2adb6a65.jpeg",
        "title": "Retro futurism",
        "repo": "martintomov/retrofuturism-flux",
        "weights": "retrofuturism_flux_lora_martintomov_v1.safetensors",
        "trigger_word": "retrofuturism"             
    },
    #175
    {
        "image": "https://huggingface.co/Bootoshi/retroanime/resolve/main/images/9f21dffe-c4da-46c0-b0a6-e06257cf98d6.webp",
        "title": "Retro Anime",
        "repo": "Bootoshi/retroanime",
        "weights": "RetroAnimeFluxV1.safetensors",
        "trigger_word": "retro anime"   
    },
    #176
    {
        "image": "https://huggingface.co/alvdansen/plushy-world-flux/resolve/main/images/ComfyUI_00666_%20(1).png",
        "title": "Plushy world",
        "repo": "alvdansen/plushy-world-flux",
        "weights": "plushy_world_flux_araminta_k.safetensors",
        "trigger_word": "3dcndylnd style" 
    },
    #177
    {
        "image": "https://huggingface.co/renderartist/ROYGBIVFlux/resolve/main/images/ComfyUI_temp_qpxhm_00154_.png",
        "title": "ROYGBIVFlux",
        "repo": "renderartist/ROYGBIVFlux",
        "weights": "ROYGBIV_Flux_v1_renderartist.safetensors",
        "trigger_word": "r0ygb1v, digital illustration, textured"         
    },
    #178
    {
        "image": "https://huggingface.co/alvdansen/sonny-anime-flex/resolve/main/images/GLuFasaLyEoBaAUQMREVf_20b5cf5b178a404296978e360a9ac435.png",
        "title": "sonny anime",
        "repo": "alvdansen/sonny-anime-flex",
        "weights": "araminta_k_sonnyanime_fluxd_flex.safetensors",
        "trigger_word": "nm22 [style] style"            
    },
    #179
    {
        "image": "https://huggingface.co/bingbangboom/flux_whimscape/resolve/main/images/2.png",
        "title": "flux whimscape",
        "repo": "bingbangboom/flux_whimscape",
        "weights": "WHMSCPE001.safetensors",
        "trigger_word": "illustration in the style of WHMSCPE001"  
    },
    #180
    {
        "image": "https://huggingface.co/glif-loradex-trainer/AP123_movie_shots_ic_lora_experiment_v1/resolve/main/samples/1730923312010__000000500_1.jpg",
        "title": "movie shots ic lora",
        "repo": "glif-loradex-trainer/AP123_movie_shots_ic_lora_experiment_v1",
        "weights": "movie_shots_ic_lora_experiment_v1.safetensors",
        "trigger_word": "MOVIE-SHOTS"        
    },
    #181
    {
        "image": "https://huggingface.co/glif/LiDAR-Vision/resolve/main/images/f8f1995e-c583-425b-b73a-f3e873ce1005.png",
        "title": "LiDAR",
        "repo": "glif/LiDAR-Vision",
        "weights": "Lidar.safetensors",
        "trigger_word": "L1d4r"        
    },
    #182
    {
        "image": "https://huggingface.co/prithivMLmods/Canopus-Flux-LoRA-Hoodies/resolve/main/images/XXX.png",
        "title": "Hoodies",
        "repo": "prithivMLmods/Canopus-Flux-LoRA-Hoodies",
        "weights": "Canopus-Flux-LoRA-Hoodies.safetensors",
        "trigger_word": "Hoodie"          
    },
    #183
    {
        "image": "https://huggingface.co/dvyio/flux-lora-rdr2/resolve/main/images/RTqPoC9s0M1wNhago27OV_dda06f47ee764202aa5e55efa923b94e.jpg",
        "title": "World of RDR",
        "repo": "dvyio/flux-lora-rdr2",
        "weights": "eb79a593332f40458ea36fe0782f01a4_pytorch_lora_weights.safetensors",
        "trigger_word": "in the style of RDRGM"           
    },
    #184
    {
        "image": "https://huggingface.co/Fihade/Retro-Collage-Art-Flux-Dev/resolve/main/images/005.jpeg",
        "title": "Retro Collage Art",
        "repo": "Fihade/Retro-Collage-Art-Flux-Dev",
        "weights": "flux_dev_ff_collage_artstyle.safetensors",
        "trigger_word": "ff-collage"         
    },
    #185
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Quote-LoRA/resolve/main/images/QQ2.png",
        "title": "Quote",
        "repo": "prithivMLmods/Flux.1-Dev-Quote-LoRA",
        "weights": "quoter001.safetensors",
        "trigger_word": "quoter" 
    },
    #186
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Stamp-Art-LoRA/resolve/main/images/SS2.png",
        "title": "Stamp",
        "repo": "prithivMLmods/Flux.1-Dev-Stamp-Art-LoRA",
        "weights": "stam9.safetensors",
        "trigger_word": "stam9"         
    },
    #187
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Hand-Sticky-LoRA/resolve/main/images/H3.png",
        "title": "Hand Sticky",
        "repo": "prithivMLmods/Flux.1-Dev-Hand-Sticky-LoRA",
        "weights": "handstick69.safetensors",
        "trigger_word": "handstick69"        
    },
    #188
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Poster-HQ-LoRA/resolve/main/images/PP2.png",
        "title": "Poster Foss",
        "repo": "prithivMLmods/Flux.1-Dev-Poster-HQ-LoRA",
        "weights": "poster-foss.safetensors",
        "trigger_word": "poster foss"          
    },
    #189
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Ctoon-LoRA/resolve/main/images/C3.png",
        "title": "Ctoon",
        "repo": "prithivMLmods/Flux.1-Dev-Ctoon-LoRA",
        "weights": "ctoon.safetensors",
        "trigger_word": "ctoon"         
    },
    #190
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-C33-Design-LoRA/resolve/main/images/3.png",
        "title": "C33 Design",
        "repo": "prithivMLmods/Flux-C33-Design-LoRA",
        "weights": "C33.safetensors",
        "trigger_word": "C33 Design"           
    },
    #191
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Indo-Realism-LoRA/resolve/main/images/333.png",
        "title": "Indo Realism",
        "repo": "prithivMLmods/Flux.1-Dev-Indo-Realism-LoRA",
        "weights": "indo-realism.safetensors",
        "trigger_word": "indo-realism"        
    },
    #192
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Sketch-Card-LoRA/resolve/main/images/SC2.png",
        "title": "Sketch Card",
        "repo": "prithivMLmods/Flux.1-Dev-Sketch-Card-LoRA",
        "weights": "sketchcard.safetensors",
        "trigger_word": "sketch card"            
    },
    #193
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Movie-Boards-LoRA/resolve/main/images/MB1.png",
        "title": "Movie Board",
        "repo": "prithivMLmods/Flux.1-Dev-Movie-Boards-LoRA",
        "weights": "movieboard.safetensors",
        "trigger_word": "movieboard"        
    },
    #194
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Pov-DoorEye-LoRA/resolve/main/images/L4.png",
        "title": "Door Eye View",
        "repo": "prithivMLmods/Flux.1-Dev-Pov-DoorEye-LoRA",
        "weights": "look-in-2.safetensors",
        "trigger_word": "look in 2"         
    },
    #195
    {
        "image": "https://huggingface.co/alvdansen/enna-sketch-style/resolve/main/images/out-0%20(23).webp",
        "title": "Enna Sketch",
        "repo": "alvdansen/enna-sketch-style",
        "weights": "enna_sketch_style_araminta_k.safetensors",
        "trigger_word": "sketch illustration style"           
    },
    #196
    {
        "image": "https://huggingface.co/jbilcke-hf/flux-dev-panorama-lora-2/resolve/main/samples/HDRI%20panoramic%20view%20of%20TOK%2C%20visiting%20an%20amusement%20park%20about%20harry%20potter.webp",
        "title": "Panorama",
        "repo": "jbilcke-hf/flux-dev-panorama-lora-2",
        "weights": "flux_train_replicate.safetensors",
        "trigger_word": "HDRI panoramic view of TOK"         
    },
    #197
    {
        "image": "https://huggingface.co/Shakker-Labs/FLUX.1-dev-LoRA-Micro-landscape-on-Mobile-Phone/resolve/main/images/a29b8763a8f733dea09c1ab07a42263ef6e304cb81be3f5c97fbf8f6.jpg",
        "title": "Micro Landscape",
        "repo": "Shakker-Labs/FLUX.1-dev-LoRA-Micro-landscape-on-Mobile-Phone",
        "weights": "FLUX-dev-lora-micro-landscape.safetensors",
        "trigger_word": "miniature stereoscopic scene"           
    },
    #198
    {
        "image": "https://huggingface.co/glif-loradex-trainer/goldenark__Ancient_Greece_Watercolor_Sketch_Style/resolve/main/samples/1727152322975__000002000_0.jpg",
        "title": "Ancient Greece Watercolor",
        "repo": "glif-loradex-trainer/goldenark__Ancient_Greece_Watercolor_Sketch_Style",
        "weights": "Ancient_Greece_Watercolor_Sketch_Style.safetensors",
        "trigger_word": "AncientWaterColorStyle"          
    },
    #199
    {
        "image": "https://huggingface.co/glif-loradex-trainer/i12bp8_appelsiensam_mii_v1/resolve/main/samples/1731918886531__000003000_0.jpg",
        "title": "M11 PPLSNSM",
        "repo": "glif-loradex-trainer/i12bp8_appelsiensam_mii_v1",
        "weights": "appelsiensam_mii_v1.safetensors",
        "trigger_word": "M11_PPLSNSM"         
    },
    #200
    {
        "image": "https://huggingface.co/glif-loradex-trainer/an303042_RisographPrint_v1/resolve/main/samples/1731852835625__000003000_5.jpg",
        "title": "RisographPrint",
        "repo": "glif-loradex-trainer/an303042_RisographPrint_v1",
        "weights": "RisographPrint_v1.safetensors",
        "trigger_word": "rsgrf , risograph"          
    },
    #201
    {
        "image": "https://huggingface.co/gokaygokay/Flux-White-Background-LoRA/resolve/main/images/example_mtojzmerf.png",
        "title": "White Background",
        "repo": "gokaygokay/Flux-White-Background-LoRA",
        "weights": "80cfbf52faf541d49c6abfe1ac571112_lora.safetensors",
        "trigger_word": "in the middle ,white background"        
    },
    #202
    {
        "image": "https://huggingface.co/glif/Gesture-Draw/resolve/main/images/cf8697fb-d6b4-4684-8a1d-82beb9d266ed.jpg",
        "title": "Gesture Draw",
        "repo": "glif/Gesture-Draw",
        "weights": "Gesture_Draw_v1.safetensors",
        "trigger_word": "gstdrw style"         
    },
    #203
    {
        "image": "https://huggingface.co/MuninStudio/FLUX.1-dev-LoRA-Hard-Flash/resolve/main/images/02.jpg",
        "title": "Hard Flash",
        "repo": "MuninStudio/FLUX.1-dev-LoRA-Hard-Flash",
        "weights": "HRDFLS.safetensors",
        "trigger_word": "HRDFLS"          
    },
    #204
    {
        "image": "https://huggingface.co/dvyio/flux-lora-the-sims/resolve/main/images/dunBAVBsALOepaE_dsWFI_6b0fef6b0fc4472aa07d00edea7c75b3.jpg",
        "title": "SIMS",
        "repo": "dvyio/flux-lora-the-sims",
        "weights": "011ed14848b3408c8d70d3ecfa14f122_lora.safetensors",
        "trigger_word": "video game screenshot in the style of THSMS"        
    },
    #205
    {
        "image": "https://huggingface.co/UmeAiRT/FLUX.1-dev-LoRA-Ume_Sky/resolve/main/images/flux_00171_.png",
        "title": "Umesky",
        "repo": "UmeAiRT/FLUX.1-dev-LoRA-Ume_Sky",
        "weights": "ume_sky_v2.safetensors",
        "trigger_word": "umesky"          
    },
    #206
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Realtime-Toon-Mix/resolve/main/images/T1.png",
        "title": "Realtime Toon Mix",
        "repo": "prithivMLmods/Flux.1-Dev-Realtime-Toon-Mix",
        "weights": "toon-mix.safetensors",
        "trigger_word": "toon mix"           
    },
    #207
    {
        "image": "https://huggingface.co/oshtz/flux-pointcrayonstyle/resolve/main/images/pointcrayonstyle%20illustration%2C%20at%20a%20lighthouse%2C%20the%20ambiance%20is%20exotic%20and%20mysterious.png",
        "title": "Pointcrayonstyle",
        "repo": "oshtz/flux-pointcrayonstyle",
        "weights": "flux-pointcrayonstyle.safetensors",
        "trigger_word": "pointcrayonstyle"              
    },
    #208
    {
        "image": "https://huggingface.co/Purz/vhs-box/resolve/main/33727059.jpeg",
        "title": "VSH Box",
        "repo": "Purz/vhs-box",
        "weights": "purz-vhs_box.safetensors",
        "trigger_word": "vhs_box"          
    },
    #209
    {
        "image": "https://huggingface.co/nerijs/flux_prettyshot_v1/resolve/main/images/image_5.png",
        "title": "Prettyshot",
        "repo": "nerijs/flux_prettyshot_v1",
        "weights": "flux_prettyshot_v1.safetensors",
        "trigger_word": "pr3ttysh0t"         
    },
    #210
    {
        "image": "https://huggingface.co/glif-loradex-trainer/insectagon_pipo_hippo1/resolve/main/samples/1729839793051__000001500_1.jpg",
        "title": "Insectagon pipo",
        "repo": "glif-loradex-trainer/insectagon_pipo_hippo1",
        "weights": "pipo_hippo1.safetensors",
        "trigger_word": "pipo_meme"          
    },
    
    #----------------------------------------------------------Phase2-----------------------------------------------------------------------#

    #211
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Polaroid-Plus/resolve/main/images/P1.png",
        "title": "Polaroid Plus",
        "repo": "prithivMLmods/Flux-Polaroid-Plus",
        "weights": "polaroid-plus.safetensors",
        "trigger_word": "Polaroid Collage"          
    },
    #212
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Product-Ad-Backdrop/resolve/main/images/PA1.png",
        "title": "Product Ad",
        "repo": "prithivMLmods/Flux-Product-Ad-Backdrop",
        "weights": "Prod-Ad.safetensors",
        "trigger_word": "Product Ad"         
    },
    #213
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Art-Nightmare-99/resolve/main/images/NM4.png",
        "title": "Nightmare 99",
        "repo": "prithivMLmods/Flux-Art-Nightmare-99",
        "weights": "nm99.safetensors",
        "trigger_word": "nm99"         
    },
    #214
    {
        "image": "https://huggingface.co/prithivMLmods/Flux.1-Dev-Frosted-Container-LoRA/resolve/main/images/FG1.png",
        "title": "Frosted Container",
        "repo": "prithivMLmods/Flux.1-Dev-Frosted-Container-LoRA",
        "weights": "frosted-gc.safetensors",
        "trigger_word": "frosted GC"          
    },
    #215
    {
        "image": "https://huggingface.co/glif-loradex-trainer/swap_magenta_kuki_roblox/resolve/main/samples/1731068408259__000001500_4.jpg",
        "title": "Magenta Kuki Roblox",
        "repo": "glif-loradex-trainer/swap_magenta_kuki_roblox",
        "weights": "magenta_kuki_roblox.safetensors",
        "trigger_word": "kuki_magenta, roblox"         
    },
    #216
    {
        "image": "https://huggingface.co/glif-loradex-trainer/maxxd4240_PleinAir/resolve/main/samples/1730493095161__000003000_4.jpg",
        "title": "Plein Air",
        "repo": "glif-loradex-trainer/maxxd4240_PleinAir",
        "weights": "PleinAir.safetensors",
        "trigger_word": "P1e!n"           
    },
    #217
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-GArt-LoRA/resolve/main/images/G2.png",
        "title": "GArt",
        "repo": "prithivMLmods/Flux-GArt-LoRA",
        "weights": "GArt.safetensors",
        "trigger_word": "GArt"         
    },
    #218
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-Super-Capybara-HF/resolve/main/images/C2.png",
        "title": "Capybara HF",
        "repo": "strangerzonehf/Flux-Super-Capybara-HF",
        "weights": "capybara-hf.safetensors",
        "trigger_word": "capybara hf"        
    },
    #219
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Fine-Detail-LoRA/resolve/main/images/SR1.png",
        "title": "Fine Detail",
        "repo": "prithivMLmods/Flux-Fine-Detail-LoRA",
        "weights": "Fine-Detail.safetensors",
        "trigger_word": "Super Detail"          
    },
    #220
    {
        "image": "https://huggingface.co/gokaygokay/Flux-Digital-Backgrounds-LoRA/resolve/main/images/image1.jpg",
        "title": "Digital Backgrounds",
        "repo": "gokaygokay/Flux-Digital-Backgrounds-LoRA",
        "weights": "digital_background_lora.safetensors",
        "trigger_word": "dgtlbg"         
    },
    #221
    {
        "image": "https://huggingface.co/gokaygokay/Flux-Realistic-Backgrounds-LoRA/resolve/main/images/image1.jpg",
        "title": "Realistic Backgrounds",
        "repo": "gokaygokay/Flux-Realistic-Backgrounds-LoRA",
        "weights": "realistic_background_lora.safetensors",
        "trigger_word": "rlstcbg"         
    },
    #222
    {
        "image": "https://huggingface.co/prithivMLmods/Flux-Lego-Ref-LoRA/resolve/main/images/2.png",
        "title": "LEGO",
        "repo": "prithivMLmods/Flux-Lego-Ref-LoRA",
        "weights": "Lego.safetensors",
        "trigger_word": "lego --fref --89890"          
    },
    #223
    #3DXL
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-3DXL-Partfile-0001/resolve/main/images/4.png",
        "title": "3DXLP1",
        "repo": "strangerzonehf/Flux-3DXL-Partfile-0001",
        "weights": "3DXLP1.safetensors",
        "trigger_word": "3DXLP1"          
    },
    #224
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-3DXL-Partfile-0002/resolve/main/images/44.png",
        "title": "3DXLP2",
        "repo": "strangerzonehf/Flux-3DXL-Partfile-0002",
        "weights": "3DXLP2.safetensors",
        "trigger_word": "3DXLP2"         
    },
    #225
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-3DXL-Partfile-0003/resolve/main/images/222.png",
        "title": "3DXLP3",
        "repo": "strangerzonehf/Flux-3DXL-Partfile-0003",
        "weights": "3DXLP3.safetensors",
        "trigger_word": "3DXLP3"           
    },
    #226
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-3DXL-Partfile-0004/resolve/main/images/3333.png",
        "title": "3DXLP4",
        "repo": "strangerzonehf/Flux-3DXL-Partfile-0004",
        "weights": "3DXLP4.safetensors",
        "trigger_word": "3DXLP4"         
    },
    #227
    {
        "image": "https://huggingface.co/strangerzonehf/Flux-3DXL-Partfile-C0001/resolve/main/images/C3.png",
        "title": "3DXLC1",
        "repo": "strangerzonehf/Flux-3DXL-Partfile-C0001",
        "weights": "3DXLC1.safetensors",
        "trigger_word": "3DXLC1"         
    }
    #add new
]

#--------------------------------------------------Model Initialization-----------------------------------------------------------------------------------------#

dtype = torch.bfloat16
device = "cuda" if torch.cuda.is_available() else "cpu"
base_model = "black-forest-labs/FLUX.1-dev"

#TAEF1 is very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE. FLUX.1 is useful for real-time previewing of the FLUX.1 generation process.#
taef1 = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=dtype).to(device)
good_vae = AutoencoderKL.from_pretrained(base_model, subfolder="vae", torch_dtype=dtype).to(device)
pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype, vae=taef1).to(device)
pipe_i2i = AutoPipelineForImage2Image.from_pretrained(base_model,
                                                      vae=good_vae,
                                                      transformer=pipe.transformer,
                                                      text_encoder=pipe.text_encoder,
                                                      tokenizer=pipe.tokenizer,
                                                      text_encoder_2=pipe.text_encoder_2,
                                                      tokenizer_2=pipe.tokenizer_2,
                                                      torch_dtype=dtype
                                                     )

MAX_SEED = 2**32-1

pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)

class calculateDuration:
    def __init__(self, activity_name=""):
        self.activity_name = activity_name

    def __enter__(self):
        self.start_time = time.time()
        return self
    
    def __exit__(self, exc_type, exc_value, traceback):
        self.end_time = time.time()
        self.elapsed_time = self.end_time - self.start_time
        if self.activity_name:
            print(f"Elapsed time for {self.activity_name}: {self.elapsed_time:.6f} seconds")
        else:
            print(f"Elapsed time: {self.elapsed_time:.6f} seconds")

def update_selection(evt: gr.SelectData, width, height):
    selected_lora = loras[evt.index]
    new_placeholder = f"Type a prompt for {selected_lora['title']}"
    lora_repo = selected_lora["repo"]
    updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✅"
    if "aspect" in selected_lora:
        if selected_lora["aspect"] == "portrait":
            width = 768
            height = 1024
        elif selected_lora["aspect"] == "landscape":
            width = 1024
            height = 768
        else:
            width = 1024
            height = 1024
    return (
        gr.update(placeholder=new_placeholder),
        updated_text,
        evt.index,
        width,
        height,
    )

@spaces.GPU(duration=100)
def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
    pipe.to("cuda")
    generator = torch.Generator(device="cuda").manual_seed(seed)
    with calculateDuration("Generating image"):
        # Generate image
        for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
            prompt=prompt_mash,
            num_inference_steps=steps,
            guidance_scale=cfg_scale,
            width=width,
            height=height,
            generator=generator,
            joint_attention_kwargs={"scale": lora_scale},
            output_type="pil",
            good_vae=good_vae,
        ):
            yield img

def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps, cfg_scale, width, height, lora_scale, seed):
    generator = torch.Generator(device="cuda").manual_seed(seed)
    pipe_i2i.to("cuda")
    image_input = load_image(image_input_path)
    final_image = pipe_i2i(
        prompt=prompt_mash,
        image=image_input,
        strength=image_strength,
        num_inference_steps=steps,
        guidance_scale=cfg_scale,
        width=width,
        height=height,
        generator=generator,
        joint_attention_kwargs={"scale": lora_scale},
        output_type="pil",
    ).images[0]
    return final_image 

@spaces.GPU(duration=100)
def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
    if selected_index is None:
        raise gr.Error("You must select a LoRA before proceeding.🧨")
    selected_lora = loras[selected_index]
    lora_path = selected_lora["repo"]
    trigger_word = selected_lora["trigger_word"]
    if(trigger_word):
        if "trigger_position" in selected_lora:
            if selected_lora["trigger_position"] == "prepend":
                prompt_mash = f"{trigger_word} {prompt}"
            else:
                prompt_mash = f"{prompt} {trigger_word}"
        else:
            prompt_mash = f"{trigger_word} {prompt}"
    else:
        prompt_mash = prompt

    with calculateDuration("Unloading LoRA"):
        pipe.unload_lora_weights()
        pipe_i2i.unload_lora_weights()
        
    #LoRA weights flow
    with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
        pipe_to_use = pipe_i2i if image_input is not None else pipe
        weight_name = selected_lora.get("weights", None)
        
        pipe_to_use.load_lora_weights(
            lora_path, 
            weight_name=weight_name, 
            low_cpu_mem_usage=True
        )
            
    with calculateDuration("Randomizing seed"):
        if randomize_seed:
            seed = random.randint(0, MAX_SEED)
            
    if(image_input is not None):
        
        final_image = generate_image_to_image(prompt_mash, image_input, image_strength, steps, cfg_scale, width, height, lora_scale, seed)
        yield final_image, seed, gr.update(visible=False)
    else:
        image_generator = generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress)
    
        final_image = None
        step_counter = 0
        for image in image_generator:
            step_counter+=1
            final_image = image
            progress_bar = f'<div class="progress-container"><div class="progress-bar" style="--current: {step_counter}; --total: {steps};"></div></div>'
            yield image, seed, gr.update(value=progress_bar, visible=True)
            
        yield final_image, seed, gr.update(value=progress_bar, visible=False)
        
def get_huggingface_safetensors(link):
  split_link = link.split("/")
  if(len(split_link) == 2):
            model_card = ModelCard.load(link)
            base_model = model_card.data.get("base_model")
            print(base_model)
      
            #Allows Both
            if((base_model != "black-forest-labs/FLUX.1-dev") and (base_model != "black-forest-labs/FLUX.1-schnell")):
                raise Exception("Flux LoRA Not Found!")
                
            # Only allow "black-forest-labs/FLUX.1-dev"
            #if base_model != "black-forest-labs/FLUX.1-dev":
                #raise Exception("Only FLUX.1-dev is supported, other LoRA models are not allowed!")
                
            image_path = model_card.data.get("widget", [{}])[0].get("output", {}).get("url", None)
            trigger_word = model_card.data.get("instance_prompt", "")
            image_url = f"https://huggingface.co/{link}/resolve/main/{image_path}" if image_path else None
            fs = HfFileSystem()
            try:
                list_of_files = fs.ls(link, detail=False)
                for file in list_of_files:
                    if(file.endswith(".safetensors")):
                        safetensors_name = file.split("/")[-1]
                    if (not image_url and file.lower().endswith((".jpg", ".jpeg", ".png", ".webp"))):
                      image_elements = file.split("/")
                      image_url = f"https://huggingface.co/{link}/resolve/main/{image_elements[-1]}"
            except Exception as e:
              print(e)
              gr.Warning(f"You didn't include a link neither a valid Hugging Face repository with a *.safetensors LoRA")
              raise Exception(f"You didn't include a link neither a valid Hugging Face repository with a *.safetensors LoRA")
            return split_link[1], link, safetensors_name, trigger_word, image_url

def check_custom_model(link):
    if(link.startswith("https://")):
        if(link.startswith("https://huggingface.co") or link.startswith("https://www.huggingface.co")):
            link_split = link.split("huggingface.co/")
            return get_huggingface_safetensors(link_split[1])
    else: 
        return get_huggingface_safetensors(link)

def add_custom_lora(custom_lora):
    global loras
    if(custom_lora):
        try:
            title, repo, path, trigger_word, image = check_custom_model(custom_lora)
            print(f"Loaded custom LoRA: {repo}")
            card = f'''
            <div class="custom_lora_card">
              <span>Loaded custom LoRA:</span>
              <div class="card_internal">
                <img src="{image}" />
                <div>
                    <h3>{title}</h3>
                    <small>{"Using: <code><b>"+trigger_word+"</code></b> as the trigger word" if trigger_word else "No trigger word found. If there's a trigger word, include it in your prompt"}<br></small>
                </div>
              </div>
            </div>
            '''
            existing_item_index = next((index for (index, item) in enumerate(loras) if item['repo'] == repo), None)
            if(not existing_item_index):
                new_item = {
                    "image": image,
                    "title": title,
                    "repo": repo,
                    "weights": path,
                    "trigger_word": trigger_word
                }
                print(new_item)
                existing_item_index = len(loras)
                loras.append(new_item)
        
            return gr.update(visible=True, value=card), gr.update(visible=True), gr.Gallery(selected_index=None), f"Custom: {path}", existing_item_index, trigger_word
        except Exception as e:
            gr.Warning(f"Invalid LoRA: either you entered an invalid link, or a non-FLUX LoRA")
            return gr.update(visible=True, value=f"Invalid LoRA: either you entered an invalid link, a non-FLUX LoRA"), gr.update(visible=False), gr.update(), "", None, ""
    else:
        return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""

def remove_custom_lora():
    return gr.update(visible=False), gr.update(visible=False), gr.update(), "", None, ""

run_lora.zerogpu = True

css = '''
#gen_btn{height: 100%}
#gen_column{align-self: stretch}
#title{text-align: center}
#title h1{font-size: 3em; display:inline-flex; align-items:center}
#title img{width: 100px; margin-right: 0.5em}
#gallery .grid-wrap{height: 10vh}
#lora_list{background: var(--block-background-fill);padding: 0 1em .3em; font-size: 90%}
.card_internal{display: flex;height: 100px;margin-top: .5em}
.card_internal img{margin-right: 1em}
.styler{--form-gap-width: 0px !important}
#progress{height:30px}
#progress .generating{display:none}
.progress-container {width: 100%;height: 30px;background-color: #f0f0f0;border-radius: 15px;overflow: hidden;margin-bottom: 20px}
.progress-bar {height: 100%;background-color: #4f46e5;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
'''

with gr.Blocks(theme="prithivMLmods/Minecraft-Theme", css=css, delete_cache=(60, 60)) as app:
    title = gr.HTML(
        """<h1>FLUX LoRA DLC🥳</h1>""",
        elem_id="title",
    )
    selected_index = gr.State(None)
    with gr.Row():
        with gr.Column(scale=3):
            prompt = gr.Textbox(label="Prompt", lines=1, placeholder=":/ choose the LoRA and type the prompt ")
        with gr.Column(scale=1, elem_id="gen_column"):
            generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
    with gr.Row():
        with gr.Column():
            selected_info = gr.Markdown("")
            gallery = gr.Gallery(
                [(item["image"], item["title"]) for item in loras],
                label="LoRA DLC's",
                allow_preview=False,
                columns=3,
                elem_id="gallery",
                show_share_button=False
            )
            with gr.Group():
                custom_lora = gr.Textbox(label="Enter Custom LoRA", placeholder="prithivMLmods/Canopus-LoRA-Flux-Anime")
                gr.Markdown("[Check the list of FLUX LoRA's](https://huggingface.co/models?other=base_model:adapter:black-forest-labs/FLUX.1-dev)", elem_id="lora_list")
            custom_lora_info = gr.HTML(visible=False)
            custom_lora_button = gr.Button("Remove custom LoRA", visible=False)
        with gr.Column():
            progress_bar = gr.Markdown(elem_id="progress",visible=False)
            result = gr.Image(label="Generated Image")

    with gr.Row():
        with gr.Accordion("Advanced Settings", open=False):
            with gr.Row():
                input_image = gr.Image(label="Input image", type="filepath")
                image_strength = gr.Slider(label="Denoise Strength", info="Lower means more image influence", minimum=0.1, maximum=1.0, step=0.01, value=0.75)
            with gr.Column():
                with gr.Row():
                    cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=3.5)
                    steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=28)
                
                with gr.Row():
                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
                    height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024)
                
                with gr.Row():
                    randomize_seed = gr.Checkbox(True, label="Randomize seed")
                    seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
                    lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=0.95)

    gallery.select(
        update_selection,
        inputs=[width, height],
        outputs=[prompt, selected_info, selected_index, width, height]
    )
    custom_lora.input(
        add_custom_lora,
        inputs=[custom_lora],
        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, prompt]
    )
    custom_lora_button.click(
        remove_custom_lora,
        outputs=[custom_lora_info, custom_lora_button, gallery, selected_info, selected_index, custom_lora]
    )
    gr.on(
        triggers=[generate_button.click, prompt.submit],
        fn=run_lora,
        inputs=[prompt, input_image, image_strength, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
        outputs=[result, seed, progress_bar]
    )

app.queue()
app.launch()