add error handling

myapp/app.py

@@ -9,13 +9,12 @@ from huggingface_hub import InferenceClient
 from PIL import Image
 from qrcode_artistic import write_artistic
 
-if gr.NO_RELOAD:
-    try:
-        import dotenv
+try:
+    import dotenv
 
-        dotenv.load_dotenv()
-    except ImportError:
-        pass
+    dotenv.load_dotenv()
+except ImportError:
+    pass
 
 client = InferenceClient(model="black-forest-labs/FLUX.1-schnell")
 
@@ -35,10 +34,16 @@ with gr.Blocks() as demo:
     @gr.on(
         triggers=[button.click, prompt.submit],
         inputs=prompt,
-        outputs=[background, output],
+        outputs=[button, background],
     )
     def generate_background(prompt):
-        return client.text_to_image(prompt, width=400, height=400), None
+        yield gr.update(interactive=False), None
+        response = None
+
+        try:
+            response = client.text_to_image(prompt, width=400, height=400)
+        finally:
+            yield gr.update(interactive=True), response
 
     @gr.on(
         triggers=[text.submit, background.change, scale.change],