Update app.py

app.py

@@ -73,14 +73,14 @@ def process(
 ):
     if not input_text:
         gr.Info("Please enter a text prompt.")
-        return None, None
+        return None, None, None, None
     input_text = "A military COR3 "+input_text
     image = input_image_editor['background']
     mask = input_image_editor['layers'][0]
 
     if not image:
         gr.Info("Please upload an image.")
-        return None, None
+        return None, None, None, None
         
     width, height = resize_image_dimensions(original_resolution_wh=image.size)
     resized_image = image.resize((width, height), Image.LANCZOS)
@@ -96,13 +96,14 @@ def process(
         image=resized_image,
         width=width,
         height=height,
+        num_images_per_prompt =4,
         strength=strength_slider,
         generator=generator,
         joint_attention_kwargs={"scale": 1.2},
         num_inference_steps=num_inference_steps_slider
-        ).images[0]
+        )
     print('INFERENCE DONE')
-    return result, None
+    return result.images[0], result.images[1], result.images[2], result.images[3]
 
      #resized_mask = mask.resize((width, height), Image.LANCZOS)
 
@@ -128,10 +129,9 @@ with gr.Blocks() as demo:
             input_image_editor_component = gr.ImageEditor(
                 label='Image',
                 type='pil',
-                sources=["upload", "webcam"],
+                sources=["upload"],
                 image_mode='RGB',
-                layers=False,
-                brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
+                layers=False
 
             with gr.Row():
                 input_text_component = gr.Text(
@@ -180,9 +180,13 @@ with gr.Blocks() as demo:
         with gr.Column():
             output_image_component = gr.Image(
                 type='pil', image_mode='RGB', label='Generated image', format="png")
-            with gr.Accordion("Debug", open=False):
-                output_mask_component = gr.Image(
-                    type='pil', image_mode='RGB', label='Input mask', format="png")
+             output_image_component2 = gr.Image(
+                type='pil', image_mode='RGB', label='Generated image', format="png")
+             output_image_component3 = gr.Image(
+                type='pil', image_mode='RGB', label='Generated image', format="png")
+             output_image_component4 = gr.Image(
+                type='pil', image_mode='RGB', label='Generated image', format="png")
+
 
     submit_button_component.click(
         fn=process,
@@ -196,7 +200,9 @@ with gr.Blocks() as demo:
         ],
         outputs=[
             output_image_component,
-            output_mask_component
+            output_image_component2,
+            output_image_component3,
+            output_image_component4
         ]
     )