Update app.py

app.py

@@ -70,7 +70,7 @@ def save_ply_from_array(verts):
     return tmpf.name
 
 @spaces.GPU(duration=120)
-def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,top_p,temperature,video_path):
+def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,top_p,temperature,video_path,simplify,texture_size):
     torch.manual_seed(seed)
     chat_query = _chatbot[-1][0]
     query = task_history[-1][0]
@@ -175,8 +175,8 @@ def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,t
                 glb = postprocessing_utils.to_glb(
                     outputs['gaussian'][0],
                     outputs['mesh'][0],
-                    simplify=0.95,          
-                    texture_size=1024,    
+                    simplify=simplify,          
+                    texture_size=texture_size,    
                     verbose=False  
                 )
                 glb.export(f"{TMP_DIR}/{trial_id}.glb")
@@ -202,8 +202,8 @@ def predict(_chatbot,task_history,viewer_voxel,viewer_mesh,task_new,seed,top_k,t
                 glb = postprocessing_utils.to_glb(
                     outputs['gaussian'][0],
                     outputs['mesh'][0],
-                    simplify=0.95,          
-                    texture_size=1024,    
+                    simplify=simplify,          
+                    texture_size=texture_size,    
                     verbose=False  
                 )
                 glb.export(f"{TMP_DIR}/{trial_id}.glb")
@@ -467,6 +467,9 @@ with gr.Blocks() as demo:
                 top_k       = gr.Slider(label="top_k",minimum=1024,maximum=8194,value=1024,step=10)
                 top_p       = gr.Slider(label="top_p",minimum=0.1,maximum=1.0,value=0.1,step=0.05)
                 temperature = gr.Slider(label="temperature",minimum=0.1,maximum=1.0,value=0.1,step=0.05)
+            with gr.Accordion(label="GLB Extraction Settings", open=False):
+                mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
+                texture_size = gr.Slider(512, 2048, label="Texture Size", value=1024, step=512)
 
             query = gr.Textbox(lines=2, label='Input')
             image_input = gr.Image(visible=False, type="filepath", label="Image Input")
@@ -481,7 +484,6 @@ with gr.Blocks() as demo:
             task_new     = gr.State([])
         with gr.Column():
             viewer_plot  = gr.Plot(label="Voxel Visual",scale=0.5)
-            #viewer_mesh  = gr.Model3D(label="Mesh Visual", height=200,scale=1.0)
             video_output = gr.Video(label="Generated 3D Asset", autoplay=True, loop=True, height=300)
             viewer_mesh  = LitModel3D(label="Extracted GLB", exposure=20.0, height=300)
 
@@ -521,7 +523,7 @@ with gr.Blocks() as demo:
 
     submit_btn.click(add_text, [chatbot, task_history, query,task_new],\
                                [chatbot, task_history,task_new]).then(
-        predict, [chatbot, task_history,viewer_plot,viewer_mesh,task_new,seed,top_k,top_p,temperature,video_output],\
+        predict, [chatbot, task_history,viewer_plot,viewer_mesh,task_new,seed,top_k,top_p,temperature,video_output,simplify,texture_size],\
                  [chatbot,viewer_plot,viewer_mesh,task_new,video_output], show_progress=True
     )
     submit_btn.click(reset_user_input, [], [query])