Update app.py

app.py

@@ -6,16 +6,22 @@ import os
 import base64
 from PIL import Image, PngImagePlugin
 
+url = os.environ['url']
+status_code = requests.get(url).status_code
+
+
 with gr.Blocks() as block:
     with gr.Row():
         txt_prompt = gr.Text(label="Prompt",value="1girl",lines=2)
     with gr.Row():
-        btn_generate = gr.Button("Generate")
+        if status_code == requests.codes.ok:
+            btn_generate = gr.Button("Generate",variant="primary")
+        else:
+            btn_generate = gr.Button("Sorry, not working",variant="secondary")
     with gr.Row():
         image = gr.Image()
 
     def generate(prompt):
-        url = os.environ['url']
         payload = {
             "seed": -1,
             "sampler_name": "Euler",
@@ -43,5 +49,4 @@ with gr.Blocks() as block:
         outputs=[image]
     )
 
-block.queue().launch()
-
+block.queue().launch()