Update app.py

app.py

@@ -46,7 +46,8 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         generator=generator
     ).images[0]
 
-    return image, seed
+    # Return image, seed, and the used prompts
+    return image, seed, f"Prompt used: {final_prompt}\nNegative prompt used: {negative_prompt}"
 
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
@@ -59,6 +60,9 @@ css = """
     margin: 0 auto;
     max-width: 640px;
 }
+#run-button {
+    width: 100%;
+}
 """
 
 with gr.Blocks(css=css) as demo:
@@ -71,6 +75,9 @@ with gr.Blocks(css=css) as demo:
         # Display result image at the top
         result = gr.Image(label="Result", show_label=False)
 
+        # Add a textbox to display the prompts used for generation
+        prompt_info = gr.Textbox(label="Prompts Used", lines=3, interactive=False)
+
         # Tabbed interface to select either Prompt or Tags
         with gr.Tabs() as tabs:
             with gr.TabItem("Prompt Input"):
@@ -92,7 +99,7 @@ with gr.Blocks(css=css) as demo:
                 use_tags = gr.State(True)
 
         # Full-width "Run" button
-        run_button = gr.Button("Run", scale=0)
+        run_button = gr.Button("Run", scale=0, elem_id="run-button")
 
         with gr.Accordion("Advanced Settings", open=False):
             negative_prompt = gr.Text(
@@ -146,10 +153,10 @@ with gr.Blocks(css=css) as demo:
                     value=35,
                 )
 
-        gr.Examples(
-            examples=examples,
-            inputs=[prompt]
-        )
+        # gr.Examples(
+        #     examples=examples,
+        #     inputs=[prompt]
+        # )
 
     def check_tab(prompt, tag_selection_1, tag_selection_2, selected_tab):
         return selected_tab == "Tag Selection"
@@ -160,7 +167,7 @@ with gr.Blocks(css=css) as demo:
         triggers=[run_button.click, prompt.submit],
         fn=infer,
         inputs=[prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, tag_selection_1, tag_selection_2, use_tags],
-        outputs=[result, seed]
+        outputs=[result, seed, prompt_info]  # Include prompt_info in the outputs
     )
 
 demo.queue().launch()