Update app.py

app.py

@@ -18,12 +18,16 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
 # Function to generate an image
-@spaces.GPU  # Adjust the duration as needed
+@spaces.GPU
 def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress()):
-    pipe.to('cuda')  # Move the model to GPU when the function is called
+    pipe.to('cuda')
     
     if randomize_seed:
         seed = random.randint(0, 99999999)
+    
+    original_prompt = prompt
+    original_negative_prompt = negative_prompt
+    
     if use_defaults:
         prompt = f"{prompt}, masterpiece, best quality"
         negative_prompt = f"nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name, {negative_prompt}"
@@ -48,18 +52,26 @@ def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_s
 
     torch.cuda.empty_cache()
 
-    return image, seed
+    return image, seed, original_prompt, original_negative_prompt
 
 # Define Gradio interface
 def interface_fn(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress()):
-    image, seed = generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress)
+    image, seed, original_prompt, original_negative_prompt = generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress)
     
-    # Create the generation details string
-    details = f"""
-    {prompt}
-    Negative prompt: {negative_prompt}
-    Steps: {num_inference_steps}, CFG scale: {guidance_scale}, Seed: {seed}, Size: {resolution}
-    """
+    details = f"""Prompt: {original_prompt}
+
+Negative prompt: {original_negative_prompt}
+
+Steps: {num_inference_steps}, CFG scale: {guidance_scale}, Seed: {seed}, Size: {resolution}
+
+Default quality tags: {"Enabled" if use_defaults else "Disabled"}"""
+
+    if use_defaults:
+        details += f"""
+
+Default prompt addition: , masterpiece, best quality
+
+Default negative prompt addition: nsfw, lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, artist name"""
     
     return image, seed, gr.update(value=seed), details
 
@@ -92,9 +104,15 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
             reset_button = gr.Button("Reset")
 
         with gr.Column():
-            output_image = gr.Image(type="pil", label="Generated Image")
-            with gr.Accordion("Generation Details", open=False):
-                generation_info = gr.Textbox(label="", max_lines=10, interactive=False)
+        output_image = gr.Image(type="pil", label="Generated Image")
+        with gr.Accordion("Generation Details", open=False):
+            generation_info = gr.Textbox(
+                label="", 
+                max_lines=15, 
+                interactive=False,
+                elem_id="generation_info",
+                show_copy_button=True
+            )
             gr.Markdown(
                 """
                 ### Recommended prompt formatting: