Update app.py

app.py

@@ -220,7 +220,7 @@ def run_triposg(image_path: str,
     This function takes a single 2D image as input and produces a set of part-based 3D meshes,
     using compositional latent diffusion with attention to structure and part separation.
     Optionally removes the background using a pretrained background removal model (RMBG),
-    and outputs a merged object mesh, a split preview (exploded view), and a downloadable ZIP of all parts.
+    and outputs a merged object mesh, a split preview (exploded view).
 
     Args:
         image_path (str): Path to the input image file on disk.
@@ -239,7 +239,6 @@ def run_triposg(image_path: str,
             - `merged_path` (str): File path to the merged full object mesh (`object.glb`).
             - `split_preview_path` (str): File path to the exploded-view mesh (`split.glb`).
             - `export_dir` (str): Directory where all generated meshes were saved.
-            - `zip_path` (str): Path to the ZIP file containing all individual part meshes.
 
     Notes:
         - This function utilizes HuggingFace pretrained weights for both part generation and background removal.
@@ -293,12 +292,6 @@ def run_triposg(image_path: str,
         part = os.path.join(export_dir, f"part_{idx:02}.glb")
         mesh.export(part)
         parts.append(part)
-        
-    zip_path = os.path.join(os.environ["PARTCRAFTER_PROCESSED"], f"{session_id}.zip")
-    
-    # shutil.make_archive wants the base name without extension:
-    base_name = zip_path[:-4]  # strip off '.zip'
-    shutil.make_archive(base_name, 'zip', export_dir)
     
     # Merge and color
     merged = get_colored_mesh_composition(outputs)
@@ -310,7 +303,7 @@ def run_triposg(image_path: str,
     split_preview_path = os.path.join(export_dir, "split.glb")
     split_mesh.export(split_preview_path)
  
-    return merged_path, split_preview_path, export_dir, zip_path
+    return merged_path, split_preview_path, export_dir
 
 def cleanup(request: gr.Request):
 
@@ -379,7 +372,6 @@ def build_demo():
                         output_model = gr.Model3D(label="Merged 3D Object", height=512)
                         split_model = gr.Model3D(label="Split Preview", height=512)
                         output_dir = gr.Textbox(label="Export Directory", visible=False)
-                        download_zip = gr.File(label="Download All Parts (zip)", visible=False)
             with gr.Row():
                 with gr.Column():
                     examples = gr.Examples(
@@ -404,7 +396,7 @@ def build_demo():
                             
                         ],
                         inputs=[input_image, num_parts],
-                        outputs=[output_model, split_model, output_dir, download_zip],
+                        outputs=[output_model, split_model, output_dir],
                         fn=run_triposg,
                         cache_examples=True,
                     )
@@ -412,7 +404,7 @@ def build_demo():
             run_button.click(fn=run_triposg,
                              inputs=[input_image, num_parts, seed, num_tokens, num_steps,
                                      guidance, flash_decoder, remove_bg, session_state],
-                             outputs=[output_model, split_model, output_dir, download_zip])
+                             outputs=[output_model, split_model, output_dir])
         return demo
 
 if __name__ == "__main__":