test for api generate

app.py

@@ -280,20 +280,38 @@ def quick_generate_glb(
     texture_size: int,
     req: gr.Request,
 ) -> Tuple[str, str]:
-    seed = get_seed(True, 0)
     state, _ = image_to_3d(
         image=image,
-        multiimages=[],
-        is_multiimage=False,
+        multiimages=multiimages,
+        is_multiimage=is_multiimage,
         seed=seed,
-        ss_guidance_strength=7.5,
-        ss_sampling_steps=12,
-        slat_guidance_strength=3.0,
-        slat_sampling_steps=12,
-        multiimage_algo="stochastic",
-        req=None
+        ss_guidance_strength=ss_guidance_strength,
+        ss_sampling_steps=ss_sampling_steps,
+        slat_guidance_strength=slat_guidance_strength,
+        slat_sampling_steps=slat_sampling_steps,
+        multiimage_algo=multiimage_algo,
+        req=req
     )
-    return extract_glb(state, mesh_simplify=0.95, texture_size=1024, req=None)
+    return extract_glb(state, mesh_simplify=mesh_simplify, texture_size=texture_size, req=req)
+
+@spaces.GPU(api_name="quick_glb_dl")
+def quick_glb_dl(image: Image.Image, req: gr.Request) -> str:
+    seed = get_seed(True, 0)
+    state, _ = image_to_3d(image, [], False, seed, 7.5, 12, 3.0, 12, "stochastic", req)
+    glb_path, _ = extract_glb(state, 0.95, 1024, req)
+    return glb_path
+
+def test_for_api_gen(image: Image.Image) -> Image.Image:
+    """
+    bilibili .
+
+    Args:
+        image (Image.Image): The input imagein hererererer.
+
+    Returns:
+        Image.Image: The preprocessed image no processs.
+    """
+    return image
     
 
 
@@ -343,6 +361,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                 extract_gs_btn = gr.Button("Extract Gaussian", interactive=False)
                 generate_glb_btn = gr.Button("Quick Generate GLB in layout")
                 quick_generate_glb_btn = gr.Button("Quick Generate GLB with download")
+                quick_generate_glb_easy_btn = gr.Button("Easy Quick Generate GLB with download")
             gr.Markdown("""
                         *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
                         """)
@@ -474,24 +493,29 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         outputs=[download_glb]
     )
 
-    # quick_generate_glb_btn.click(
-    #     fn=quick_generate_glb,
-    #     inputs=[
-    #         image_input,
-    #         gr.State([]),  # multiimages
-    #         gr.State(False),  # is_multiimage
-    #         gr.State(0),  # seed
-    #         gr.State(7.5),  # ss_guidance_strength
-    #         gr.State(12),  # ss_sampling_steps
-    #         gr.State(3.0),  # slat_guidance_strength
-    #         gr.State(12),  # slat_sampling_steps
-    #         gr.State("stochastic"),  # multiimage_algo
-    #         gr.State(0.95),  # mesh_simplify
-    #         gr.State(1024),  # texture_size
-    #     ],
-    #     outputs=[model_output, glb_download],
-    #     api_name="quick_generate_glb"
-    # )
+    quick_generate_glb_btn.click(
+        fn=quick_generate_glb,
+        inputs=[
+            image_prompt,
+            multiimage_prompt,
+            is_multiimage,
+            seed,
+            ss_guidance_strength,
+            ss_sampling_steps,
+            slat_guidance_strength,
+            slat_sampling_steps,
+            multiimage_algo,
+            mesh_simplify,
+            texture_size,
+        ],
+        outputs=[model_output, download_glb],
+    )
+
+    quick_generate_glb_easy_btn.click(
+        fn=quick_glb_dl,
+        inputs=[image_prompt],
+        outputs=[download_glb],
+    )