Update app.py

app.py

@@ -1,4 +1,4 @@
-
+import os
 import spaces
 import gradio as gr
 import torch
@@ -8,9 +8,13 @@ import random
 import uuid
 from typing import Tuple
 import numpy as np
+from huggingface_hub import login
 
-
-
+# Authenticate with Hugging Face once at startup
+HF_TOKEN = os.environ.get("HF_TOKEN")
+if HF_TOKEN:
+    login(HF_TOKEN)
+    print("Authenticated with Hugging Face.")
 
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"
@@ -23,21 +27,21 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     return seed
 
 MAX_SEED = np.iinfo(np.int32).max
-# Define DESCRIPTIONz before using it
-DESCRIPTIONz = "" # Initialize as an empty string
 
+# Description for the UI
+DESCRIPTIONz = ""
 if not torch.cuda.is_available():
-    DESCRIPTIONz += "\n<p>⚠️Running on CPU, This may not work on CPU.</p>"
+    DESCRIPTIONz += "\n<p>⚠️ Running on CPU. This may not work as expected.</p>"
 
+# Load base model and LoRA weights using the HF token
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
-
+pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16, use_auth_token=HF_TOKEN)
 lora_repo = "strangerzonehf/Flux-Super-Realism-LoRA"
 trigger_word = "Super Realism"  # Leave trigger_word blank if not used.
-
 pipe.load_lora_weights(lora_repo)
 pipe.to("cuda")
 
+# Define style list for prompt enhancements
 style_list = [
     {
         "name": "3840 x 2160",
@@ -56,9 +60,7 @@ style_list = [
         "prompt": "{prompt}",
     },
 ]
-
 styles = {k["name"]: k["prompt"] for k in style_list}
-
 DEFAULT_STYLE_NAME = "3840 x 2160"
 STYLE_NAMES = list(styles.keys())
 
@@ -77,12 +79,11 @@ def generate(
     progress=gr.Progress(track_tqdm=True),
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
-
     positive_prompt = apply_style(style_name, prompt)
     
     if trigger_word:
         positive_prompt = f"{trigger_word} {positive_prompt}"
-
+    
     images = pipe(
         prompt=positive_prompt,
         width=width,
@@ -100,25 +101,46 @@ examples = [
     "Woman in a red jacket, snowy, in the style of hyper-realistic portraiture, caninecore, mountainous vistas, timeless beauty, palewave, iconic, distinctive noses --ar 72:101 --stylize 750 --v 6",
     "Super Realism, Headshot of handsome young man, wearing dark gray sweater with buttons and big shawl collar, brown hair and short beard, serious look on his face, black background, soft studio lighting, portrait photography --ar 85:128 --v 6.0 --style",
     "Super Realism, High-resolution photograph, woman, UHD, photorealistic, shot on a Sony A7III --chaos 20 --ar 1:2 --style raw --stylize 250",
-    "Super-realism, Purple Dreamy, a medium-angle shot of a young woman with long brown hair, wearing a pair of eye-level glasses, stands in front of a backdrop of purple and white lights. The womans eyes are closed, her lips are slightly parted, as if she is looking up at the sky. Her hair is cascading over her shoulders, framing her face. She is wearing a sleeveless top, adorned with tiny white dots, and a gold chain necklace around her neck. Her left earrings are dangling from her ears, adding a pop of color to the scene."
+    "Super-realism, Purple Dreamy, a medium-angle shot of a young woman with long brown hair, wearing a pair of eye-level glasses, stands in front of a backdrop of purple and white lights. The woman's eyes are closed, her lips slightly parted, as if she is looking up at the sky. Her hair cascades over her shoulders, framing her face. She wears a sleeveless top adorned with tiny white dots and a gold chain necklace around her neck. Her left earrings add a pop of color to the scene."
 ]
 
+# Updated CSS for a violet-themed, improved UI
 css = '''
-.gradio-container{max-width: 888px !important}
-h1{text-align:center}
+.gradio-container {
+    max-width: 888px !important;
+    background-color: #f3f0ff;
+    padding: 20px;
+    border-radius: 12px;
+    box-shadow: 0 4px 12px rgba(0,0,0,0.1);
+}
+h1 {
+    text-align: center;
+    color: #4b0082;
+    margin-bottom: 10px;
+}
 footer {
-    visibility: hidden
+    visibility: hidden;
 }
 .submit-btn {
-    background-color: #e34949 !important;
+    background-color: #8A2BE2 !important;
     color: white !important;
+    border-radius: 8px;
+    padding: 10px 20px;
+    font-weight: bold;
 }
 .submit-btn:hover {
-    background-color: #ff3b3b !important;
+    background-color: #6a0dad !important;
+}
+.accordion-header {
+    background-color: #e6e0ff;
+    color: #4b0082;
 }
 '''
 
-with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
+with gr.Blocks(css=css, theme="default") as demo:
+    gr.Markdown("<h1>FLUX Image Generator</h1>")
+    gr.Markdown(DESCRIPTIONz)
+    
     with gr.Row():
         with gr.Column(scale=1):
             prompt = gr.Text(
@@ -128,9 +150,9 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 placeholder="Enter your prompt",
                 container=False,
             )
-            run_button = gr.Button("Generate as ( 768 x 1024 )🤗", scale=0, elem_classes="submit-btn")
+            run_button = gr.Button("Generate 🤗", elem_classes="submit-btn")
             
-            with gr.Accordion("Advanced options", open=True, visible=True):
+            with gr.Accordion("Advanced options", open=True):
                 seed = gr.Slider(
                     label="Seed",
                     minimum=0,
@@ -141,7 +163,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                 )
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
                 
-                with gr.Row(visible=True):
+                with gr.Row():
                     width = gr.Slider(
                         label="Width",
                         minimum=512,
@@ -181,10 +203,8 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                     value=DEFAULT_STYLE_NAME,
                     label="Quality Style",
                 )
-        
         with gr.Column(scale=2):
             result = gr.Gallery(label="Result", columns=1, show_label=False)
-            
             gr.Examples(
                 examples=examples,
                 inputs=prompt,
@@ -213,4 +233,4 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=40).launch()
+    demo.queue(max_size=40).launch()