Upload app.py with huggingface_hub

app.py

@@ -328,12 +328,17 @@ def end_session(req: gr.Request):
 
 
 def remove_background(input: Image.Image) -> Image.Image:
-    with tempfile.NamedTemporaryFile(suffix='.png') as f:
-        input = input.convert('RGB')
-        input.save(f.name)
-        output = rmbg_client.predict(handle_file(f.name), api_name="/image")[0][0]
-        output = Image.open(output)
-        return output
+    try:
+        with tempfile.NamedTemporaryFile(suffix='.png') as f:
+            input = input.convert('RGB')
+            input.save(f.name)
+            output = rmbg_client.predict(handle_file(f.name), api_name="/image")[0][0]
+            output = Image.open(output)
+            return output
+    except Exception as e:
+        print(f"[WARNING] Background removal failed: {e}. Using image as-is.")
+        # Return original image with a full-white alpha channel
+        return input.convert('RGBA')
 
 
 def preprocess_image(input: Image.Image) -> Image.Image:
@@ -547,19 +552,14 @@ def image_to_3d(
     )
     mesh = outputs[0]
     mesh.simplify(16777216)  # nvdiffrast limit
-    render_images = render_utils.render_snapshot(mesh, resolution=1024, r=2, fov=36, nviews=STEPS, envmap=envmap)
+    images = render_utils.render_snapshot(mesh, resolution=1024, r=2, fov=36, nviews=STEPS, envmap=envmap)
     state = pack_state(latents)
     torch.cuda.empty_cache()
-    return state, render_images
 
-
-def build_preview_html(state, render_images):
-    """Encode rendered images to base64 and build the HTML preview.
-    No GPU needed — pure CPU work."""
-    # Encode 48 images in parallel
+    # --- HTML Construction ---
     def encode_preview_image(args):
         m_idx, s_idx, render_key = args
-        img_base64 = image_to_base64(Image.fromarray(render_images[render_key][s_idx]))
+        img_base64 = image_to_base64(Image.fromarray(images[render_key][s_idx]))
         return (m_idx, s_idx, img_base64)
 
     encode_tasks = [
@@ -571,7 +571,6 @@ def build_preview_html(state, render_images):
     with ThreadPoolExecutor(max_workers=8) as executor:
         encoded_results = list(executor.map(encode_preview_image, encode_tasks))
 
-    # Build HTML from encoded results
     encoded_map = {(m, s): b64 for m, s, b64 in encoded_results}
     images_html = ""
     for m_idx, mode in enumerate(MODES):
@@ -588,7 +587,6 @@ def build_preview_html(state, render_images):
                      loading="eager">
             """
 
-    # Button Row HTML
     btns_html = ""
     for idx, mode in enumerate(MODES):
         active_class = "active" if idx == DEFAULT_MODE else ""
@@ -599,7 +597,6 @@ def build_preview_html(state, render_images):
                  title="{mode['name']}">
         """
 
-    # Assemble the full component
     full_html = f"""
     <div class="previewer-container">
         <div class="tips-wrapper">
@@ -754,7 +751,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", neutral_hue="slate"),
 
     output_buf = gr.State()
     processed_buf = gr.State()
-    render_buf = gr.State()
     selected_img_idx = gr.State(value=None)
 
 
@@ -803,10 +799,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="orange", neutral_hue="slate"),
             tex_slat_guidance_strength, tex_slat_guidance_rescale, tex_slat_sampling_steps, tex_slat_rescale_t,
             multiimage_algo, tex_multiimage_algo
         ],
-        outputs=[output_buf, render_buf],
-    ).then(
-        build_preview_html,
-        inputs=[output_buf, render_buf],
         outputs=[output_buf, preview_output],
     )