Update app.py

app.py

@@ -6,7 +6,6 @@ import spaces
 from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 from diffusers.utils import load_image
-import pandas as pd
 import random
 import time
 
@@ -17,10 +16,6 @@ KRYPTO_LORA = {
     "adapter_name": "krypt0"
 }
 
-# Load prompts for the randomize button
-df = pd.read_csv('prompts.csv', header=None)
-prompt_values = df.values.flatten()
-
 # Get access token from Space secrets
 HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
@@ -44,7 +39,8 @@ print(f"Loading on-board LoRA: {KRYPTO_LORA['repo']}")
 pipe.load_lora_weights(
     KRYPTO_LORA['repo'],
     low_cpu_mem_usage=True,
-    adapter_name=KRYPTO_LORA['adapter_name']
+    adapter_name=KRYPTO_LORA['adapter_name'],
+    token=HF_TOKEN
 )
 print("LoRA loaded successfully.")
 
@@ -55,7 +51,6 @@ pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_ret
 
 
 def calculate_dimensions(aspect_ratio, resolution):
-    """Calculates width and height based on aspect ratio and base resolution."""
     resolution = int(resolution)
     if aspect_ratio == "Square (1:1)":
         width, height = resolution, resolution
@@ -65,16 +60,13 @@ def calculate_dimensions(aspect_ratio, resolution):
         width, height = resolution, int(resolution * 9 / 16)
     elif aspect_ratio == "Ultrawide (21:9)":
         width, height = resolution, int(resolution * 9 / 21)
-    else: # Fallback
+    else:
         width, height = resolution, resolution
-    # Ensure dimensions are multiples of 64 for optimal performance
     width = (width // 64) * 64
     height = (height // 64) * 64
     return width, height
 
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
-    """Generator function for text-to-image with live preview."""
-    # The parent @spaces.GPU function has already allocated a GPU
     pipe.to(device) 
     generator = torch.Generator(device=device).manual_seed(seed)
     
@@ -89,7 +81,6 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
         output_type="pil",
         good_vae=good_vae,
     )
-    # Yield previews and the final image
     final_image = None
     for i, image in enumerate(image_generator):
         final_image = image
@@ -98,8 +89,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
     yield final_image, gr.update(visible=False)
 
 def update_history(new_image, history):
-    """Adds the new image to the history gallery."""
-    if new_image is None: # Don't add empty images on error
+    if new_image is None:
         return history
     if history is None:
         history = []
@@ -111,10 +101,17 @@ def run_generation(prompt, lora_scale, cfg_scale, steps, randomize_seed, seed, a
     if not prompt:
         raise gr.Error("Prompt cannot be empty.")
 
-    prompt_mash = f"{KRYPTO_LORA['trigger']}, {prompt}"
-    print("Final prompt:", prompt_mash)
+    # --- NOUVELLE LOGIQUE DE PROMPT ---
+    # Définition des parties fixes du prompt
+    prefix_prompt = f"{KRYPTO_LORA['trigger']}, Krypt0 the white scruffy superdog with a red cape,"
+    suffix_prompt = ", This is a cinematic, ultra-high-detail, photorealistic still"
+    
+    # Construction du prompt final
+    user_prompt = prompt # Le prompt entré par l'utilisateur
+    prompt_mash = f"{prefix_prompt} {user_prompt}{suffix_prompt}"
+
+    print("Final prompt sent to model:", prompt_mash)
     
-    # Activate the LoRA adapter with the slider's weight
     pipe.set_adapters([KRYPTO_LORA['adapter_name']], adapter_weights=[lora_scale])
     print(f"Adapter '{KRYPTO_LORA['adapter_name']}' activated with weight {lora_scale}.")
     
@@ -124,7 +121,6 @@ def run_generation(prompt, lora_scale, cfg_scale, steps, randomize_seed, seed, a
     width, height = calculate_dimensions(aspect_ratio, base_resolution)
     print(f"Generating a {width}x{height} image.")
 
-    # The function now only handles text-to-image
     for image, progress_update in generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
         yield image, seed, progress_update
 
@@ -132,41 +128,58 @@ run_generation.zerogpu = True
 
 # --- User Interface (Gradio) ---
 css = '''
-#title{text-align: center}
-#title h1{font-size: 3em; display:inline-flex; align-items:center}
-#title img{width: 80px; margin-right: 0.25em}
+#title_container { text-align: center; margin-bottom: 1em; }
+#title_line { display: flex; justify-content: center; align-items: center; }
+#title_line h1 { font-size: 2.5em; margin: 0; }
+#subtitle { font-size: 1.1em; color: #57606a; margin-top: 0.3em; }
 .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.1s ease-in-out}
-#random_prompt_btn{max-width: 2.5em; min-width: 2.5em !important; height: 100% !important;}
 '''
 
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
     # --- Header ---
-    with gr.Row():
-        gr.HTML(
-            """<div id='title'><h1><img src="https://huggingface.co/Econogoat/KRYPTO_LORA/resolve/main/krypt0.png" alt="LoRA"> Krypt0 Image Generator</h1><br><span>Generate images with the Krypt0 artistic style</span></div>"""
-        )
+    gr.HTML(
+        """
+        <div id="title_container">
+            <div id="title_line">
+                <h1>Krypto Image Generator - beta v1</h1>
+            </div>
+            <div id="subtitle">
+                Powered by $Krypto | @Kryptocoinonsol
+            </div>
+        </div>
+        """
+    )
     
     with gr.Row():
         # --- LEFT COLUMN: CONTROLS ---
         with gr.Column(scale=3):
-            # Prompt and Style Controls
+            # Prompt Controls (Simplifié)
             with gr.Group():
-                with gr.Row():
-                    random_prompt_btn = gr.Button("🎲", elem_id="random_prompt_btn")
-                    prompt = gr.Textbox(label="Prompt", lines=2, placeholder="e.g., a portrait of a warrior queen", scale=8)
-                
-                lora_scale = gr.Slider(label="Krypt0 Style Strength", minimum=0, maximum=2, step=0.05, value=0.9)
-            
-            # Image Shape Controls
-            aspect_ratio = gr.Radio(
-                label="Aspect Ratio",
-                choices=["Square (1:1)", "Portrait (9:16)", "Landscape (16:9)", "Ultrawide (21:9)"],
-                value="Square (1:1)"
-            )
+                prompt = gr.Textbox(
+                    label="Prompt",
+                    lines=3,
+                    placeholder="Krypto the superdog sits in the snow, with snow on his muzzle, looking innocent. It's a medium shot of the dog, and the image creates a friendly atmosphere."
+                )
             
+            # Image Shape and Style Controls
+            with gr.Group():
+                aspect_ratio = gr.Radio(
+                    label="Aspect Ratio",
+                    choices=["Square (1:1)", "Portrait (9:16)", "Landscape (16:9)", "Ultrawide (21:9)"],
+                    value="Square (1:1)"
+                )
+                lora_scale = gr.Slider(
+                    label="Krypt0 Style Strength",
+                    minimum=0,
+                    maximum=2,
+                    step=0.05,
+                    value=0.9,
+                    info="Controls how strongly the artistic style is applied. Higher values mean a more stylized image."
+                )
+
             # Advanced Settings
-            with gr.Accordion("Advanced Settings", open=True):
+            with gr.Accordion("Advanced Settings", open=False):
                 base_resolution = gr.Slider(label="Resolution (longest side)", minimum=768, maximum=1408, step=64, value=1024)
                 steps = gr.Slider(label="Generation Steps", minimum=4, maximum=50, step=1, value=20)
                 cfg_scale = gr.Slider(label="Guidance (CFG Scale)", minimum=1, maximum=10, step=0.5, value=3.5)
@@ -185,15 +198,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
                 history_gallery = gr.Gallery(label="History", columns=4, object_fit="contain", interactive=False)
     
     # --- Event Logic ---
-    def get_random_prompt():
-        return random.choice(prompt_values)
-        
-    random_prompt_btn.click(
-        fn=get_random_prompt,
-        inputs=[],
-        outputs=[prompt]
-    )
-
     generation_event = gr.on(
         triggers=[generate_button.click, prompt.submit],
         fn=run_generation,