CSS

app.py

@@ -11,6 +11,7 @@ import torch
 from PIL import Image
 import gradio as gr
 
+
 from diffusers import (
     DiffusionPipeline,
     AutoencoderTiny,
@@ -59,7 +60,7 @@ def retrieve_timesteps(
     **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")
+        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
@@ -2148,20 +2149,19 @@ dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 base_model = "black-forest-labs/FLUX.1-dev"
 
-# TAEF1 is a very tiny autoencoder which uses the same "latent API" as FLUX.1's VAE.
+#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
-)
+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
 
@@ -2250,7 +2250,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
     selected_lora = loras[selected_index]
     lora_path = selected_lora["repo"]
     trigger_word = selected_lora["trigger_word"]
-    if trigger_word:
+    if(trigger_word):
         if "trigger_position" in selected_lora:
             if selected_lora["trigger_position"] == "prepend":
                 prompt_mash = f"{trigger_word} {prompt}"
@@ -2265,7 +2265,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         pipe.unload_lora_weights()
         pipe_i2i.unload_lora_weights()
         
-    # LoRA weights flow
+    #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)
@@ -2280,7 +2280,8 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         if randomize_seed:
             seed = random.randint(0, MAX_SEED)
             
-    if image_input is not None:
+    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:
@@ -2289,7 +2290,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         final_image = None
         step_counter = 0
         for image in image_generator:
-            step_counter += 1
+            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)
@@ -2298,15 +2299,19 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
         
 def get_huggingface_safetensors(link):
   split_link = link.split("/")
-  if len(split_link) == 2:
+  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"):
+            #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
@@ -2314,20 +2319,20 @@ def get_huggingface_safetensors(link):
             try:
                 list_of_files = fs.ls(link, detail=False)
                 for file in list_of_files:
-                    if file.endswith(".safetensors"):
+                    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("You didn't include a link neither a valid Hugging Face repository with a *.safetensors LoRA")
-              raise Exception("You didn't include a link neither a valid Hugging Face repository with a *.safetensors LoRA")
+              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"):
+    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: 
@@ -2335,7 +2340,7 @@ def check_custom_model(link):
 
 def add_custom_lora(custom_lora):
     global loras
-    if custom_lora:
+    if(custom_lora):
         try:
             title, repo, path, trigger_word, image = check_custom_model(custom_lora)
             print(f"Loaded custom LoRA: {repo}")
@@ -2352,7 +2357,7 @@ def add_custom_lora(custom_lora):
             </div>
             '''
             existing_item_index = next((index for (index, item) in enumerate(loras) if item['repo'] == repo), None)
-            if not existing_item_index:
+            if(not existing_item_index):
                 new_item = {
                     "image": image,
                     "title": title,
@@ -2366,8 +2371,8 @@ def add_custom_lora(custom_lora):
         
             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("Invalid LoRA: either you entered an invalid link, or a non-FLUX LoRA")
-            return gr.update(visible=True, value="Invalid LoRA: either you entered an invalid link, or a non-FLUX LoRA"), gr.update(visible=False), gr.update(), "", None, ""
+            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, ""
 
@@ -2376,73 +2381,22 @@ def remove_custom_lora():
 
 run_lora.zerogpu = True
 
-# ─── UPDATED CSS WITH YOUR SITE COLORS ─────────────────────────────────────────────────────────────
 css = '''
-/* Force our background with !important */
-body {
-    background: #f4ecd8 url('https://www.transparenttextures.com/patterns/purty-wood.png') repeat !important;
-    font-family: 'Cinzel', serif;
-    color: #543d29;
-    margin: 0;
-    padding: 0;
-}
-
-/* Button & Layout */
-#gen_btn { height: 100%; }
-#gen_column { align-self: stretch; }
-
-/* Title / Header */
-#title { text-align: center; margin-bottom: 20px; }
-#title h1 {
-    font-size: 3em;
-    display: inline-flex;
-    align-items: center;
-    color: #795548; /* medium brown */
-    text-shadow: 2px 2px 4px #543d29; /* dark brown */
-}
-#title img { width: 100px; margin-right: 0.5em; }
-
-/* Gallery & LoRA List */
-#gallery .grid-wrap { height: 10vh; }
-#lora_list {
-    background: #f0e6d2;
-    padding: 0 1em 0.3em;
-    font-size: 90%;
-    color: #795548;
-    border: 1px solid #543d29;
-}
-.card_internal {
-    display: flex;
-    height: 100px;
-    margin-top: 0.5em;
-    border: 2px solid #543d29;
-    background: #f0e6d2;
-    border-radius: 8px;
-    padding: 5px;
-}
-.card_internal img {
-    margin-right: 1em;
-    border-radius: 5px;
-    border: 1px solid #543d29;
-}
-
-/* Progress Bar */
-#progress { height: 30px; margin-top: 10px; }
-#progress .generating { display: none; }
-.progress-container {
-    width: 100%;
-    height: 30px;
-    background-color: #795548;
-    border-radius: 15px;
-    overflow: hidden;
-    margin-bottom: 20px;
-}
-.progress-bar {
-    height: 100%;
-    background-color: #543d29;
-    width: calc(var(--current) / var(--total) * 100%);
-    transition: width 0.5s ease-in-out;
-}
+#gen_btn{height: 100%}
+#gen_column{align-self: stretch}
+#title{text-align: center}
+#title h1{font-size: 3em; display:inline-flex; align-items:center; color: #ffd700;}
+#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%; color: #ffd700;}
+.card_internal{display: flex;height: 100px;margin-top: .5em; border: 2px solid #8b4513; background: #3a3a3a;}
+.card_internal img{margin-right: 1em; border-radius: 5px;}
+.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: #d2691e;width: calc(var(--current) / var(--total) * 100%);transition: width 0.5s ease-in-out}
+body {background: url('https://www.transparenttextures.com/patterns/black-linen.png') center; font-family: 'Cinzel', serif; color: #ffd700;}
 '''
 
 with gr.Blocks(theme="YTheme/Minecraft", css=css, delete_cache=(60, 60)) as app:
@@ -2451,10 +2405,10 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css, delete_cache=(60, 60)) as app:
     <div id="title">
         <h1>⚔️ ChatDnD.net ⚔️</h1>
         <p>
-            <strong>Forge Your Destiny!</strong> Create legendary heroes, intricate maps, epic quests, and awe‐inspiring battle scenes.
-            Whether you’re an adventurer or a Dungeon Master, let your imagination run wild. <br>
-            <a href="https://chatdnd.net" target="_blank">Visit Our Keep</a> | 
-            <a href="https://buymeacoffee.com/watchoutformike" target="_blank">Support the Guild</a>
+            <strong>Unleash Your Imagination!</strong> Create heroes, maps, quests, and epic scenes to bring your campaigns to life. 
+            Tailored for adventurers seeking inspiration or Dungeon Masters constructing their next grand story. <br>
+            <a href="https://chatdnd.net" target="_blank">Visit Our Website</a> | 
+            <a href="https://buymeacoffee.com/watchoutformike" target="_blank">Support Us</a>
         </p>
     </div>
     """,
@@ -2465,9 +2419,9 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css, delete_cache=(60, 60)) as app:
     with gr.Row():
         with gr.Column(scale=3):
             prompt = gr.Textbox(
-                label="🎲 Your Epic Prompt",
+                label="🎲 Your Legendary Prompt",
                 lines=1,
-                placeholder="Describe your valiant hero, fearsome villain, or an epic clash of forces..."
+                placeholder="Describe your hero, villain, or epic battle scene..."
             )
         with gr.Column(scale=1, elem_id="gen_column"):
             generate_button = gr.Button("Forge Your Vision", variant="primary", elem_id="gen_btn")
@@ -2476,7 +2430,7 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css, delete_cache=(60, 60)) as app:
             selected_info = gr.Markdown("")
             gallery = gr.Gallery(
                 [(item["image"], item["title"]) for item in loras],
-                label="🛡️ LoRA Artifacts",
+                label="🛡️ LoRA Artifacts 🛡️",
                 allow_preview=False,
                 columns=3,
                 elem_id="gallery",
@@ -2488,7 +2442,7 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css, delete_cache=(60, 60)) as app:
                     placeholder="prithivMLmods/Canopus-LoRA-Flux-Anime"
                 )
                 gr.Markdown(
-                    """[Explore our FLUX Artifacts Collection 📜](https://huggingface.co/models?other=base_model:adapter:black-forest-labs/FLUX.1-dev)""",
+                    """[Explore FLUX Artifacts Collection 📜](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)
@@ -2513,7 +2467,7 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css, delete_cache=(60, 60)) as app:
                         minimum=1, maximum=20, step=0.5, value=3.5
                     )
                     steps = gr.Slider(
-                        label="⏱️ Generation Steps",
+                        label="⏱️ Steps for Generation",
                         minimum=1, maximum=50, step=1, value=28
                     )