Update app.py

app.py

@@ -89,22 +89,18 @@ def run_example(text_prompt, do_remove_background, foreground_ratio, mc_resoluti
     
     return preprocessed, mesh_name_obj, mesh_name_glb
 
+
 with gr.Blocks() as demo:
     gr.Markdown(HEADER)
     with gr.Row(variant="panel"):
         with gr.Column():
             with gr.Row():
-                # text_prompt = gr.Textbox(
-                #     label="Text Prompt",
-                #     placeholder="Enter a text prompt for image generation"
-                # )
                 input_image = gr.Image(
-                    label="Generated Image",
+                    label="Input Image",
                     image_mode="RGBA",
                     sources="upload",
                     type="pil",
                     elem_id="content_image",
-                    visible=False  # Hidden since we generate the image from text
                 )
                 processed_image = gr.Image(label="Processed Image", interactive=False)
             with gr.Row():
@@ -149,15 +145,19 @@ with gr.Blocks() as demo:
             inputs=[input_image],
             outputs=[processed_image, output_model_obj, output_model_glb],
             cache_examples=True,
-            fn=partial(run_example, do_remove_background=True, foreground_ratio=0.85, mc_resolution=256),
+            fn=partial(run_example),
             label="Examples",
             examples_per_page=20
         )
-    submit.click(fn=check_input_image, inputs=[text_prompt]).success(
-        fn=run_example,
-        inputs=[input_image, do_remove_background, foreground_ratio, mc_resolution],
-        outputs=[processed_image, output_model_obj, output_model_glb],
+    submit.click(fn=check_input_image, inputs=[input_image]).success(
+        fn=preprocess,
+        inputs=[input_image, do_remove_background, foreground_ratio],
+        outputs=[processed_image],
+    ).success(
+        fn=generate,
+        inputs=[processed_image, mc_resolution],
+        outputs=[output_model_obj, output_model_glb],
     )
 
 demo.queue(max_size=10)
-demo.launch()
+demo.launch()