change simple

app.py

@@ -45,30 +45,28 @@ def process_img2img(image,mask_image,prompt="a person",strength=0.75,seed=0,num_
 @spaces.GPU(duration=180)
 def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
     print("start process_images")
-    try:
-        # I'm not sure when this happen
-        if not isinstance(image, dict):
-            if image2 == None:
-                print("empty mask")
-                return image
-            else:
-                image = dict({'background': image, 'layers': [image2]})
-
-        if image2!=None:
-            #print("use image2")
-            mask = image2
+    
+    # I'm not sure when this happen
+    if not isinstance(image, dict):
+        if image2 == None:
+            print("empty mask")
+            return image
         else:
-            if len(image['layers']) == 0:
-                print("empty mask")
-                return image
-            print("use layer")
-            mask = image['layers'][0]
+            image = dict({'background': image, 'layers': [image2]})
+
+    if image2!=None:
+        #print("use image2")
+        mask = image2
+    else:
+        if len(image['layers']) == 0:
+            print("empty mask")
+            return image
+        print("use layer")
+        mask = image['layers'][0]
 
 
-        output = process_img2img(image["background"],mask,prompt,strength,seed)
-    except Exception as e:
-        print(f"An error occurred: {e}")
-        gr.Error(e)
+    output = process_img2img(image["background"],mask,prompt,strength,seed)
+   
     print("end process_images")
     return output
     
@@ -99,8 +97,8 @@ css="""
     max-width: 640px;
 }
 """
-demo_blocks = gr.Blocks(css=css, elem_id="demo-container")
-with demo_blocks as demo:
+
+with gr.Blocks(css=css, elem_id="demo-container") as demo:
     with gr.Column():
         gr.HTML(read_file("demo_header.html"))
     with gr.Row():
@@ -110,7 +108,7 @@ with demo_blocks as demo:
                         with gr.Row():
                             prompt = gr.Textbox(label="Prompt",value="a eyes closed girl,shut eyes",placeholder="Your prompt (what you want in place of what is erased)", elem_id="prompt")
                             
-                    btn = gr.Button("Img2Img", elem_id="run_button",variant="primary")
+                    btn = gr.Button("Img2Img", elem_id="run_button",scale=0)
                     
                     image_mask = gr.Image(sources=['upload','clipboard'],  elem_id="mask_upload", type="pil", label="Mask_Upload",height=400, value=None)
                     with gr.Accordion(label="Advanced Settings", open=False):
@@ -127,7 +125,7 @@ with demo_blocks as demo:
                     
             
 
-    btn.click(fn=process_images, inputs=[image, image_mask,prompt,strength,seed], outputs =image_out, api_name='infer')
+    #btn.click(fn=process_images, inputs=[image, image_mask,prompt,strength,seed], outputs =image_out, api_name='infer')
     gr.Examples(
                examples=[
                     #["images/00547245_99.jpg", "images/00547245_99_mask.jpg","a beautiful girl,eyes closed",0.8,"images/00547245.jpg"],
@@ -145,10 +143,13 @@ with demo_blocks as demo:
             
         """
     )
+    gr.on(
+        triggers=[btn.click, prompt.submit],
+        fn = process_images,
+        inputs = [image, image_mask, prompt, strength, seed],
+        outputs = [image_out]
+    )
+
+    demo.launch()
 
-try:
-    demo_blocks.queue(max_size=25).launch(share=False,debug=True)
-except Exception as e:
-    print(f"An launch error occurred: {e}")
-    gr.Error(e)