Update app.py

app.py

@@ -6,7 +6,6 @@ import os
 import time
 from PIL import Image
 from deep_translator import GoogleTranslator
-import json
 
 # Project by Nymbo
 
@@ -16,7 +15,7 @@ headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 100
 
 def query(prompt, is_negative=False, steps=30, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7):
-    if prompt == "" or prompt == None:
+    if prompt == "" or prompt is None:
         return None, None  # Return None for both image and file
 
     key = random.randint(0, 999)
@@ -91,21 +90,17 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
         with gr.Row():
             download_button = gr.File(label="Download Image", visible=False, elem_id="download-button")
         
-        # Update the download button visibility and file when the image is generated
+        # Update the download button when the image is generated
         def update_download_button(image, file):
             if image is not None:
                 return gr.File.update(value=file, visible=True)
             return gr.File.update(visible=False)
 
+        # Trigger the query function and update the image and download button
         text_button.click(
             query,
             inputs=[text_prompt, negative_prompt, steps, cfg, method, seed, strength],
             outputs=[image_output, download_button]
         )
-        image_output.change(
-            update_download_button,
-            inputs=[image_output, download_button],
-            outputs=[download_button]
-        )
 
 app.launch(show_api=False, share=True)