Update app.py

app.py

@@ -220,18 +220,25 @@ def callback(image, coordinates, prompt):
     # Retourner les images résultantes pour l'affichage
     return segmented_image, grid_image
 
+
 with gr.Blocks() as demo:
     with gr.Row():
-        image_input = gr.Image(type="numpy", label="Upload Image", interactive=True)
-        coordinates_output = gr.Textbox(label="Coordinates")
-    with gr.Row():
-        prompt_input = gr.Textbox(label="What do you want to change?")
-        submit_button = gr.Button("Submit")
-    with gr.Row():
-        segmented_image_output = gr.Image(type="numpy", label="Segmented Image")
-        grid_image_output = gr.Image(type="pil", label="Generated Image Grid")
+        with gr.Column():
+            image_input = gr.Image(type="numpy", label="Upload Image", interactive=True)
+            coordinates_output = gr.Textbox(label="Coordinates", interactive=False)
+            prompt_input = gr.Textbox(label="What do you want to change?")
+            process_button = gr.Button("Process")
+
+        with gr.Column():
+            segmented_image_output = gr.Image(type="pil", label="Segmented Image")
+            grid_image_output = gr.Image(type="pil", label="Generated Image Grid")
 
     image_input.select(onclick, inputs=[image_input], outputs=coordinates_output)
-    submit_button.click(fn=callback, inputs=[image_input, coordinates_output, prompt_input], outputs=[segmented_image_output, grid_image_output])
+
+    process_button.click(
+        fn=callback,
+        inputs=[image_input, coordinates_output, prompt_input],
+        outputs=[segmented_image_output, grid_image_output]
+    )
 
 demo.launch(debug=True)