bug fixes

app.py

@@ -212,7 +212,10 @@ with gr.Blocks() as demo:
             gr.Markdown("Note: Ensure that the input image is correctly pre-processed into a grey background, otherwise the results will be unpredictable.")
 
     inputs = [
-        processed_image,
+        image_input,
+        background_choice,
+        foreground_ratio,
+        back_groud_color,
         seed,
         guidance_scale,
         step,
@@ -221,17 +224,31 @@ with gr.Blocks() as demo:
         image_output,
         xyz_ouput,
         output_model,
-        # output_obj,
     ]
 
-
-    text_button.click(fn=check_input_image, inputs=[image_input]).success(
-        fn=preprocess_image,
-        inputs=[image_input, background_choice, foreground_ratio, back_groud_color],
-        outputs=[processed_image],
-    ).success(
-        fn=gen_image,
+    text_button.click(
+        fn=process_and_generate,
         inputs=inputs,
         outputs=outputs,
     )
+    
     demo.queue().launch()
+
+def process_and_generate(input_image, background_choice, foreground_ratio, backgroud_color, seed, scale, step):
+    """Process the input image and generate the 3D model in a single function"""
+    if input_image is None:
+        raise gr.Error("No image uploaded!")
+    
+    # Preprocess the image
+    processed = preprocess_image(input_image, background_choice, foreground_ratio, backgroud_color)
+    
+    # Generate the 3D model
+    pipeline.set_seed(seed)
+    rt_dict = pipeline(processed, scale=scale, step=step)
+    stage1_images = rt_dict["stage1_images"]
+    stage2_images = rt_dict["stage2_images"]
+    np_imgs = np.concatenate(stage1_images, 1)
+    np_xyzs = np.concatenate(stage2_images, 1)
+
+    glb_path = generate3d(model, np_imgs, np_xyzs, args.device)
+    return Image.fromarray(np_imgs), Image.fromarray(np_xyzs), glb_path