add quick_generate_gs

app.py

@@ -294,12 +294,33 @@ def quick_generate_glb(
     )
     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
+
+@spaces.GPU(api_name="quick_generate_gs")
+def quick_generate_gs(
+    image: Image.Image,
+    multiimages: List[Tuple[Image.Image, str]],
+    is_multiimage: bool,
+    seed: int,
+    ss_guidance_strength: float,
+    ss_sampling_steps: int,
+    slat_guidance_strength: float,
+    slat_sampling_steps: int,
+    multiimage_algo: Literal["multidiffusion", "stochastic"],
+    req: gr.Request,
+) -> Tuple[str, str]:
+    state, _ = image_to_3d(
+        image=image,
+        multiimages=multiimages,
+        is_multiimage=is_multiimage,
+        seed=seed,
+        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_gaussian(state, req=req)
 
 def test_for_api_gen(image: Image.Image) -> Image.Image:
     """
@@ -318,10 +339,26 @@ def test_for_api_gen(image: Image.Image) -> Image.Image:
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
     ## Image to 3D Asset with [TRELLIS](https://trellis3d.github.io/)
-    * Upload an image and click "Generate" to create a 3D asset. If the image has alpha channel, it be used as the mask. Otherwise, we use `rembg` to remove the background.
-    * If you find the generated 3D asset satisfactory, click "Extract GLB" to extract the GLB file and download it.
-    
-    ✨New: 1) Experimental multi-image support. 2) Gaussian file extraction.
+
+    Thanks to the incredible work of [JeffreyXiang/TRELLIS-image-large](https://huggingface.co/JeffreyXiang/TRELLIS-image-large) for providing such a stunning implementation of the TRELLIS 3D pipeline.
+
+    During my usage, I noticed that many users had questions regarding API access. I've spent some time refactoring the `image_to_3d` pipeline and adding two new endpoints:
+
+    - 🔁 `quick_generate_glb`: Directly generate and download a `.glb` 3D asset.
+    - 🌐 `quick_generate_gs`: Directly generate and download the Gaussian `.ply` file.
+    - 🧩 Both functions are exposed as Hugging Face API endpoints and can be called via `gradio_client` or any HTTP client.
+
+    ### How to Use:
+    - Upload an image and click **"Generate"** to create a 3D asset. If the image has an alpha channel, it will be used as a mask. Otherwise, `rembg` will automatically remove the background.
+    - If you're satisfied with the result, click **"Extract GLB"** or **"Extract Gaussian"** to download the 3D file.
+
+    ### Features:
+    - ✅ Single-image and experimental multi-image generation
+    - ✅ `.glb` extraction with mesh simplification and texturing
+    - ✅ `.ply` (Gaussian) extraction
+    - ✅ Public API endpoints for one-click asset generation and download
+
+    Feel free to try it out and send feedback — I'm happy to keep improving it based on your suggestions!
     """)
     
     with gr.Row():
@@ -360,8 +397,11 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
                 extract_glb_btn = gr.Button("Extract GLB", interactive=False)
                 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")
+            
+            with gr.Row():
+                quick_generate_glb_btn = gr.Button("Quick Generate GLB")
+                quick_generate_gs_btn = gr.Button("Quick Generate Gaussian")
+                
             gr.Markdown("""
                         *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
                         """)
@@ -511,14 +551,22 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
         outputs=[model_output, download_glb],
     )
 
-    quick_generate_glb_easy_btn.click(
-        fn=quick_glb_dl,
-        inputs=[image_prompt],
-        outputs=[download_glb],
+    quick_generate_gs_btn.click(
+        fn=quick_generate_gs,
+        inputs=[
+            image_prompt,
+            multiimage_prompt,
+            is_multiimage,
+            seed,
+            ss_guidance_strength,
+            ss_sampling_steps,
+            slat_guidance_strength,
+            slat_sampling_steps,
+            multiimage_algo,
+        ],
+        outputs=[model_output, download_gs],
     )
 
-
-
     
 
 # Launch the Gradio app