Update app.py

app.py

@@ -77,7 +77,7 @@ def query(prompt, is_negative=False, steps=35, cfg_scale=7, sampler="DPM++ 2M Ka
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
         print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
-        return image
+        return image, seed
     except Exception as e:
         print(f"Error when trying to open the image: {e}")
         return None
@@ -178,21 +178,23 @@ with gr.Blocks(theme=theme, css=css) as app:
                 # Add a text input for the main prompt
                 with gr.Row():
                     with gr.Column(elem_id="prompt-container"):
-                        with gr.Row():
-                            text_prompt = gr.Textbox(label="Image Prompt ✍️", placeholder="Enter a prompt here", lines=2, show_copy_button = True, elem_id="prompt-text-input")
+                        with gr.Group():
+                            with gr.Row():
+                                text_prompt = gr.Textbox(label="Image Prompt ✍️", placeholder="Enter a prompt here", lines=2, show_copy_button = True, elem_id="prompt-text-input")
+                            
+                            # Accordion for advanced settings
+                            with gr.Row():
+                                with gr.Accordion("Advanced Settings", open=False):
+                                    negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
+                                    with gr.Row():
+                                        width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=32)
+                                        height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1216, step=32)
+                                    steps = gr.Slider(label="Sampling steps", value=35, minimum=1, maximum=100, step=1)
+                                    cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=1)
+                                    strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
+                                    seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1) # Setting the seed to -1 will make it random
+                                    method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ 2S a Karras", "DPM2 a Karras", "DPM2 Karras", "DPM++ SDE Karras", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM2 Ancestral", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "dpmpp_2s_ancestral", "Euler", "Euler CFG PP", "Euler a", "Euler Ancestral", "Euler+beta", "Heun", "Heun PP2", "DDIM", "LMS Karras", "PLMS", "UniPC", "UniPC BH2"])
                         
-                        # Accordion for advanced settings
-                        with gr.Row():
-                            with gr.Accordion("Advanced Settings", open=False):
-                                negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="What should not be in the image", value="(deformed, distorted, disfigured), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, (mutated hands and fingers), disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation, misspellings, typos", lines=3, elem_id="negative-prompt-text-input")
-                                with gr.Row():
-                                    width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=32)
-                                    height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1216, step=32)
-                                steps = gr.Slider(label="Sampling steps", value=35, minimum=1, maximum=100, step=1)
-                                cfg = gr.Slider(label="CFG Scale", value=7, minimum=1, maximum=20, step=1)
-                                strength = gr.Slider(label="Strength", value=0.7, minimum=0, maximum=1, step=0.001)
-                                seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1) # Setting the seed to -1 will make it random
-                                method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ 2S a Karras", "DPM2 a Karras", "DPM2 Karras", "DPM++ SDE Karras", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM2 Ancestral", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "dpmpp_2s_ancestral", "Euler", "Euler CFG PP", "Euler a", "Euler Ancestral", "Euler+beta", "Heun", "Heun PP2", "DDIM", "LMS Karras", "PLMS", "UniPC", "UniPC BH2"])
                             with gr.Row():
                                 with gr.Accordion("🫘Seed", open=False):
                                     seed_output = gr.Textbox(label="Seed Used", elem_id="seed-output")