chore: improve app page structure

app.py

@@ -382,9 +382,9 @@ with demo:
     )
     
     gr.Markdown("### Step 3. Generate the private key.")
-    with gr.Row():
-        keygen_button = gr.Button("Generate the private key.")
+    keygen_button = gr.Button("Generate the private key.")
 
+    with gr.Row():
         keygen_checkbox = gr.Checkbox(label="Private key generated:", interactive=False)
 
         private_key_size = gr.Number(
@@ -394,9 +394,9 @@ with demo:
     user_id = gr.Textbox(label="", max_lines=2, interactive=False, visible=False)
     
     gr.Markdown("### Step 4. Encrypt the image using FHE.")
-    with gr.Row():
-        encrypt_button = gr.Button("Encrypt the image using FHE.")
+    encrypt_button = gr.Button("Encrypt the image using FHE.")
 
+    with gr.Row():
         # FIXME: Display an image representation here
         # Currently not really possible as Concrete-Numpy does not give the ability to decrypt
         # an encrypted input 
@@ -418,18 +418,14 @@ with demo:
     )
 
     gr.Markdown("### Step 5. Send the encrypted image to the server.")
-    with gr.Row():
-        send_input_button = gr.Button("Send the encrypted image to the server.")
-
-        send_input_checkbox = gr.Checkbox(label="Encrypted image sent.", interactive=False)
+    send_input_button = gr.Button("Send the encrypted image to the server.")
+    send_input_checkbox = gr.Checkbox(label="Encrypted image sent.", interactive=False)
 
     gr.Markdown("### Step 6. Run FHE execution.")
-    with gr.Row():
-        execute_fhe_button = gr.Button("Run FHE execution.")
-
-        fhe_execution_time = gr.Textbox(
-            label="Total FHE execution time (in seconds):", max_lines=1, interactive=False
-        )
+    execute_fhe_button = gr.Button("Run FHE execution.")
+    fhe_execution_time = gr.Textbox(
+        label="Total FHE execution time (in seconds):", max_lines=1, interactive=False
+    )
 
     gr.Markdown("### Step 7. Receive the encrypted output image from the server.")
     gr.Markdown(
@@ -437,9 +433,9 @@ with demo:
         "decrypted using a different private key. This is only used to visually represent an "
         "encrypted image."
     )
-    with gr.Row():
-        get_output_button = gr.Button("Receive the encrypted output image from the server.")
+    get_output_button = gr.Button("Receive the encrypted output image from the server.")
 
+    with gr.Row():
         encrypted_output_representation = gr.Image(
             label=f"Encrypted output representation ({INPUT_SHAPE[0]}x{INPUT_SHAPE[1]}):", interactive=False
         )