opt quick glb

app.py

@@ -265,6 +265,36 @@ def split_image(image: Image.Image) -> List[Image.Image]:
         images.append(Image.fromarray(image[:, s:e+1]))
     return [preprocess_image(image) for image in images]
 
+def quick_generate_glb(
+    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"],
+    mesh_simplify: float,
+    texture_size: int,
+    req: gr.Request,
+) -> Tuple[str, str]:
+    seed = get_seed(True, 0)
+    state, _ = image_to_3d(
+        image=image,
+        multiimages=[],
+        is_multiimage=False,
+        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
+    )
+    return extract_glb(state, mesh_simplify=0.95, texture_size=1024, req=None)
+    
+
 
 with gr.Blocks(delete_cache=(600, 600)) as demo:
     gr.Markdown("""
@@ -310,6 +340,7 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             with gr.Row():
                 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")
             gr.Markdown("""
                         *NOTE: Gaussian file can be very large (~50MB), it will take a while to display and download.*
                         """)
@@ -321,11 +352,6 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
             with gr.Row():
                 download_glb = gr.DownloadButton(label="Download GLB", interactive=False)
                 download_gs = gr.DownloadButton(label="Download Gaussian", interactive=False)
-
-            with gr.Accordion("Quick GLB from Image", open=False):
-                generate_glb_btn = gr.Button("Upload and Generate GLB Automatically")
-                quick_video = gr.Video(label="Quick 3D Preview", autoplay=True, loop=True)
-                quick_glb_download = gr.DownloadButton(label="Download GLB", interactive=False)  
     
     is_multiimage = gr.State(False)
     output_buf = gr.State()
@@ -424,23 +450,26 @@ with gr.Blocks(delete_cache=(600, 600)) as demo:
 
     generate_glb_btn.click(
         lambda: get_seed(True, 0),
-        outputs=[seed]
+        outputs=[]
     ).then(
         image_to_3d,
         inputs=[
             image_prompt,
             gr.State([]),
             gr.State(False),
-            seed,
+            gr.State(0),
             gr.State(7.5), gr.State(12),
             gr.State(3.0), gr.State(12),
             gr.State("stochastic")
         ],
-        outputs=[output_buf, quick_video],
+        outputs=[output_buf, video_output],
     ).then(
         extract_glb,
-        inputs=[output_buf, mesh_simplify, texture_size],
-        outputs=[model_output, quick_glb_download]
+        inputs=[output_buf, gr.State(0.95), gr.State(1024)],
+        outputs=[model_output, glb_download]
+    ).then(
+        lambda: gr.Button(interactive=True),
+        outputs=[glb_download]
     )