Add Generation Detail

app.py

@@ -26,7 +26,7 @@ def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_s
         seed = random.randint(0, 99999999)
     if use_defaults:
         prompt = f"{prompt}, masterpiece, best quality"
-        negative_prompt = f"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}"
+        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}"
     generator = torch.manual_seed(seed)
     
     def callback(step, timestep, latents):
@@ -53,10 +53,18 @@ def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_s
 # 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)
-    return image, seed, gr.update(value=seed)
+    
+    # Create the generation details string
+    details = f"""
+    {prompt}
+    Negative prompt: {negative_prompt}
+    Steps: {num_inference_steps}, CFG scale: {guidance_scale}, Seed: {seed}, Size: {resolution}
+    """
+    
+    return image, seed, gr.update(value=seed), details
 
 def reset_inputs():
-    return gr.update(value=''), gr.update(value='realistic, 3d,'), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True)
+    return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True)
 
 with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
     gr.HTML(
@@ -66,7 +74,7 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
     with gr.Row():
         with gr.Column():
             prompt_input = gr.Textbox(lines=2, placeholder="Enter prompt here", label="Prompt")
-            negative_prompt_input = gr.Textbox(lines=2, placeholder="Enter negative prompt here", label="Negative Prompt", value="realistic, 3d,")
+            negative_prompt_input = gr.Textbox(lines=2, placeholder="Enter negative prompt here", label="Negative Prompt")
             use_defaults_input = gr.Checkbox(label="Use Default Quality Tags and Negative Prompt", value=True)
             resolution_input = gr.Radio(
                 choices=[
@@ -85,6 +93,8 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
 
         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)
             gr.Markdown(
                 """
                 ### Recommended prompt formatting:
@@ -103,7 +113,7 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
         inputs=[
             prompt_input, negative_prompt_input, use_defaults_input, resolution_input, guidance_scale_input, num_inference_steps_input, seed_input, randomize_seed_input
         ],
-        outputs=[output_image, seed_input]
+        outputs=[output_image, seed_input, seed_input, generation_info]
     )
     
     reset_button.click(