Update app.py

app.py

@@ -42,26 +42,27 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
         headers = {"Authorization": f"Bearer {huggingface_api_key}"}
 
     if prompt == "" or prompt is None:
-        return None
+        return None, None, None
 
     key = random.randint(0, 999)
 
     prompt = GoogleTranslator(source='ru', target='en').translate(prompt)
     print(f'\033[1mGeneration {key} translation:\033[0m {prompt}')
 
+    original_prompt = prompt
     if enhance_prompt_option:
         prompt = enhance_prompt(prompt)
         print(f'\033[1mGeneration {key} enhanced prompt:\033[0m {prompt}')
 
-    prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
-    print(f'\033[1mGeneration {key}:\033[0m {prompt}')
+    final_prompt = f"{prompt} | ultra detail, ultra elaboration, ultra quality, perfect."
+    print(f'\033[1mGeneration {key}:\033[0m {final_prompt}')
 
     # If seed is -1, generate a random seed and use it
     if seed == -1:
         seed = random.randint(1, 1000000000)
 
     payload = {
-        "inputs": prompt,
+        "inputs": final_prompt,
         "is_negative": is_negative,
         "steps": steps,
         "cfg_scale": cfg_scale,
@@ -80,16 +81,16 @@ def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Ka
     try:
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
-        print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
+        print(f'\033[1mGeneration {key} completed!\033[0m ({final_prompt})')
 
         # Save the image to a file and return the file path and seed
         output_path = f"./output_{key}.png"
         image.save(output_path)
         
-        return output_path, seed
+        return output_path, seed, prompt if enhance_prompt_option else original_prompt
     except Exception as e:
         print(f"Error when trying to open the image: {e}")
-        return None, None
+        return None, None, None
 
 css = """
 #app-container {
@@ -144,9 +145,11 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
             text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
         with gr.Row():
             image_output = gr.Image(type="pil", label="Image Output", elem_id="gallery")
+        with gr.Row():
             seed_output = gr.Textbox(label="Seed Used", elem_id="seed-output")
+            final_prompt_output = gr.Textbox(label="Final Prompt", elem_id="final-prompt-output")
         
         # Adjust the click function to include the API key, use_dev, and enhance_prompt_option as inputs
-        text_button.click(query, inputs=[text_prompt, negative_prompt, steps, cfg, method, seed, strength, huggingface_api_key, use_dev, enhance_prompt_option], outputs=[image_output, seed_output])
+        text_button.click(query, inputs=[text_prompt, negative_prompt, steps, cfg, method, seed, strength, huggingface_api_key, use_dev, enhance_prompt_option], outputs=[image_output, seed_output, final_prompt_output])
 
-app.launch(show_api=True, share=False,show_error=True)
+app.launch(show_api=True, share=False)