chore: Update Gradio UI layout for image generation and processing

app.py

@@ -192,13 +192,14 @@ print('Loading Finished!')
 
 with gr.Blocks() as demo:
     with gr.Group():
-        with gr.Row():
-            prompt = gr.Textbox(label='Enter your prompt (English)', scale=2)
-            submit_prompt = gr.Button(
-                'Generate Image', scale=1, variant='primary')
-            img = gr.Image()
+        with gr.Tab("Generate Image and Remove Background"):
+            with gr.Row():
+                prompt = gr.Textbox(
+                    label='Enter your prompt (English)', scale=2)
+                submit_prompt = gr.Button(
+                    'Generate Image', scale=1, variant='primary')
+                img = gr.Image()
 
-        with gr.Tab("Image Processing"):
             with gr.Row(variant="panel"):
                 with gr.Column():
                     with gr.Row():
@@ -219,13 +220,9 @@ with gr.Blocks() as demo:
                         with gr.Group():
                             do_remove_background = gr.Checkbox(
                                 label="Remove Background", value=True)
-                            sample_seed = gr.Number(
-                                value=42, label="Seed Value", precision=0)
-                            sample_steps = gr.Slider(
-                                label="Sample Steps", minimum=30, maximum=75, value=75, step=5)
                     with gr.Row():
-                        submit_mesh = gr.Button(
-                            "Generate", elem_id="generate", variant="primary")
+                        submit_process = gr.Button(
+                            "Process Image", elem_id="process", variant="primary")
                     with gr.Row(variant="panel"):
                         gr.Examples(
                             examples=[os.path.join("examples", img_name) for img_name in sorted(
@@ -236,7 +233,7 @@ with gr.Blocks() as demo:
                             examples_per_page=16
                         )
 
-        with gr.Tab("3D Model"):
+        with gr.Tab("Generate 3D Model"):
             with gr.Column():
                 with gr.Row():
                     with gr.Column():
@@ -246,6 +243,15 @@ with gr.Blocks() as demo:
                             width=379,
                             interactive=False
                         )
+                with gr.Row():
+                    with gr.Group():
+                        sample_seed = gr.Number(
+                            value=42, label="Seed Value", precision=0)
+                        sample_steps = gr.Slider(
+                            label="Sample Steps", minimum=30, maximum=75, value=75, step=5)
+                with gr.Row():
+                    submit_mesh = gr.Button(
+                        "Generate 3D Model", elem_id="generate", variant="primary")
                 with gr.Row():
                     with gr.Tab("OBJ"):
                         output_model_obj = gr.Model3D(
@@ -268,13 +274,11 @@ with gr.Blocks() as demo:
     mv_images = gr.State()
 
     submit_prompt.click(fn=generate_image, inputs=[prompt], outputs=img)
-    submit_mesh.click(fn=check_input_image, inputs=[input_image]).success(
+    submit_process.click(fn=check_input_image, inputs=[input_image]).success(
         fn=preprocess, inputs=[input_image,
                                do_remove_background], outputs=[processed_image],
-    ).success(
-        fn=generate_mvs, inputs=[processed_image, sample_steps, sample_seed], outputs=[
-            mv_images, mv_show_images]
-    ).success(
+    )
+    submit_mesh.click(fn=generate_mvs, inputs=[processed_image, sample_steps, sample_seed], outputs=[mv_images, mv_show_images]).success(
         fn=make3d, inputs=[mv_images], outputs=[
             output_model_obj, output_model_glb]
     )