Update app.py

app.py

@@ -42,6 +42,8 @@ def process(data, api, api_key):
         # Initial check without blocking, useful for quick tasks
         status = run_request.status()
         print(f"Initial job status: {status}")
+        if status=="IN_QUEUE":
+            raise gr.Info("Queued 🚶🚶🚶🚶!", duration=15)
 
         if status != "COMPLETED":
             # Polling with timeout for long-running tasks
@@ -51,6 +53,11 @@ def process(data, api, api_key):
         print(f"Job output: {output}")
     except Exception as e:
         print(f"An error occurred: {e}")
+        status = run_request.status()
+        if status=="FAILED":
+            raise gr.Error("An error occured 💥!", duration=5)
+        if status=="TIMED_OUT":
+            raise gr.Error("Sorry we could not secure a worker for you ⏳! Try again", duration=5)
         
 
     image_data = output['image']
@@ -63,7 +70,13 @@ def process(data, api, api_key):
 
 def process_generate(fore, prompt, image_width, image_height, intensity, mode, refprompt):    
         
-
+    size = fore.size
+    if size[0]*size[1]<=(768*768):
+        gr.Warning("ℹ️ The input image resolution is low, it might lead to some deformation!")
+        
+    if size[0]*size[1]>(1500*1500):
+        gr.Warning("ℹ️ The input image size is too big, I will lower it!")
+    
     forestr = image_to_base64(fore.convert("RGBA"))
 
     
@@ -155,7 +168,7 @@ with block:
             image_height.change(fn=update_value, inputs=image_height, outputs=image_height)
             
     with gr.Row():
-        intensity = gr.Slider(label="Refiner Strength", minimum=1.0, maximum=7.0, value=3.0, step=0.1)
+        intensity = gr.Slider(label="Refiner Strength", minimum=1.0, maximum=7.0, value=3.0, step=0.5)
         intensity.change(fn=update_value, inputs=intensity, outputs=intensity)
         generate_button = gr.Button(value="Generate")