Update app.py Added Make3d with API Endpoint to upload iamges

app.py

@@ -180,6 +180,8 @@ model = model.to(device)
 
 print('Loading Finished!')
 
+def generate_mesh_from_mvs(mvs):
+    return make3d(mvs)
 
 def check_input_image(input_image):
     if input_image is None:
@@ -332,6 +334,7 @@ If you have any questions, feel free to open a discussion or contact us at <b>bl
 
 with gr.Blocks() as demo:
     gr.Markdown(_HEADER_)
+
     with gr.Row(variant="panel"):
         with gr.Column():
             with gr.Row():
@@ -339,46 +342,55 @@ with gr.Blocks() as demo:
                     label="Input Image",
                     image_mode="RGBA",
                     sources="upload",
-                    #width=256,
-                    #height=256,
                     type="pil",
                     elem_id="content_image",
                 )
+
                 processed_image = gr.Image(
-                    label="Processed Image", 
-                    image_mode="RGBA", 
-                    #width=256,
-                    #height=256,
-                    type="pil", 
-                    interactive=False
+                    label="Processed Image",
+                    image_mode="RGBA",
+                    type="pil",
+                    interactive=False,
                 )
+
             with gr.Row():
                 with gr.Group():
                     do_remove_background = gr.Checkbox(
-                        label="Remove Background", value=True
+                        label="Remove Background",
+                        value=True,
+                    )
+
+                    sample_seed = gr.Number(
+                        value=42,
+                        label="Seed Value",
+                        precision=0,
                     )
-                    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
+                        step=5,
                     )
 
             with gr.Row():
-                submit = gr.Button("Generate", elem_id="generate", variant="primary")
+                submit = gr.Button(
+                    "Generate",
+                    elem_id="generate",
+                    variant="primary",
+                )
 
             with gr.Row(variant="panel"):
                 gr.Examples(
                     examples=[
-                        os.path.join("examples", img_name) for img_name in sorted(os.listdir("examples"))
+                        os.path.join("examples", img_name)
+                        for img_name in sorted(os.listdir("examples"))
                     ],
                     inputs=[input_image],
                     label="Examples",
                     cache_examples=False,
-                    examples_per_page=16
+                    examples_per_page=16,
                 )
 
         with gr.Column():
@@ -390,51 +402,77 @@ with gr.Blocks() as demo:
                         label="Generated Multi-views",
                         type="pil",
                         width=379,
-                        interactive=False
+                        interactive=False,
                     )
 
-                # with gr.Column():
-                #     output_video = gr.Video(
-                #         label="video", format="mp4",
-                #         width=379,
-                #         autoplay=True,
-                #         interactive=False
-                #     )
-
             with gr.Row():
                 with gr.Tab("OBJ"):
                     output_model_obj = gr.Model3D(
                         label="Output Model (OBJ Format)",
                         interactive=False,
                     )
-                    gr.Markdown("Note: Downloaded .obj model will be flipped. Export .glb instead or manually flip it before usage.")
+
+                    gr.Markdown(
+                        "Note: Downloaded .obj model will be flipped. Export .glb instead or manually flip it before usage."
+                    )
+
                 with gr.Tab("GLB"):
                     output_model_glb = gr.Model3D(
                         label="Output Model (GLB Format)",
                         interactive=False,
                     )
-                    gr.Markdown("Note: The model shown here has a darker appearance. Download to get correct results.")
+
+                    gr.Markdown(
+                        "Note: The model shown here has a darker appearance. Download to get correct results."
+                    )
 
             with gr.Row():
-                gr.Markdown('''Try a different <b>seed value</b> if the result is unsatisfying (Default: 42).''')
+                gr.Markdown(
+                    "Try a different <b>seed value</b> if the result is unsatisfying (Default: 42)."
+                )
 
     gr.Markdown(_CITE_)
 
     mv_images = gr.State()
 
-    submit.click(fn=check_input_image, inputs=[input_image]).success(
+    # -----------------------------
+    # Standard Workflow
+    # -----------------------------
+    submit.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]
-        
+        outputs=[mv_images, mv_show_images],
     ).success(
         fn=make3d,
         inputs=[mv_images],
-        outputs=[output_model_obj, output_model_glb]
+        outputs=[output_model_obj, output_model_glb],
+        api_name="generate_mesh",
+    )
+
+    # -----------------------------
+    # API: fertiges MVS -> Mesh
+    # -----------------------------
+    mvs_input = gr.Image(
+        type="pil",
+        visible=False,
+    )
+
+    api_btn = gr.Button(
+        visible=False,
+    )
+
+    api_btn.click(
+        fn=make3d,
+        inputs=[mvs_input],
+        outputs=[output_model_obj, output_model_glb],
+        api_name="make3d_from_mvs",
     )
 
 demo.launch()