Restored the background removal file to the original state

gradio_app.py

@@ -923,7 +923,7 @@ def build_app():
 #                    with gr.Tab('Text Prompt', id='tab_txt_prompt', visible=HAS_T2I and not MV_MODE) as tab_tp:
 #                        caption = gr.Textbox(label='Text Prompt',
 #                                             placeholder='HunyuanDiT will be used to generate image.',
-#                                             info='Example: A 3D model of a cute cat, white background')
+#                                             info='Example: A 3D model of a cute cat, white background.')
                     with gr.Tab('MultiView Prompt', visible=True) as tab_mv:
                         # gr.Label('Please upload at least one front image.')
                         with gr.Row():

hy3dshape/hy3dshape/rembg.py

@@ -12,127 +12,14 @@
 # fine-tuning enabling code and other elements of the foregoing made publicly available
 # by Tencent in accordance with TENCENT HUNYUAN COMMUNITY LICENSE AGREEMENT.
 
-import os
-import io
-import traceback
-import numpy as np
-
-# Set comprehensive OpenMP environment variables to prevent ANY threading/fork conflicts
-# This prevents both our code and the rembg library from causing OpenMP fork() errors
-os.environ["OMP_NUM_THREADS"] = "1"  # Limit OpenMP to single thread
-os.environ["MKL_NUM_THREADS"] = "1"  # Intel MKL threading
-os.environ["NUMEXPR_NUM_THREADS"] = "1"  # NumExpr threading
-os.environ["OPENBLAS_NUM_THREADS"] = "1"  # OpenBLAS threading
-os.environ["VECLIB_MAXIMUM_THREADS"] = "1"  # Apple Accelerate threading
-os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"  # Avoid Intel MKL errors
-os.environ["OMP_THREAD_LIMIT"] = "1"  # Limit total number of OpenMP threads
-os.environ["OMP_DISPLAY_ENV"] = "FALSE"  # Don't display OpenMP environment
-os.environ["KMP_WARNINGS"] = "FALSE"  # Suppress KMP warnings
-os.environ["PYTHONFAULTHANDLER"] = "1"  # Better error reporting
-# Additional settings to prevent subprocess/multiprocessing issues in rembg
-os.environ["TOKENIZERS_PARALLELISM"] = "false"  # Disable tokenizer parallelism
-os.environ["OMP_WAIT_POLICY"] = "PASSIVE"  # Use passive waiting
-os.environ["KMP_INIT_AT_FORK"] = "FALSE"  # Don't initialize OpenMP at fork
-
 from PIL import Image
 from rembg import remove, new_session
 
 
-def _remove_bg_in_process(image_bytes):
-    """Function to run in a separate process to avoid OpenMP issues"""
-    try:
-        # Create a new session in this process
-        session = new_session()
-        
-        # Convert bytes back to PIL Image
-        image = Image.open(io.BytesIO(image_bytes))
-        
-        # Standard background removal
-        output = remove(
-            image,
-            session=session,
-            bgcolor=[255, 255, 255, 0],
-        )
-        
-        # Convert back to bytes
-        output_bytes = io.BytesIO()
-        output.save(output_bytes, format='PNG')
-        return output_bytes.getvalue()
-    
-    except Exception as e:
-        print(f"Background removal process error: {e}")
-        traceback.print_exc()
-        return None
-
-
 class BackgroundRemover():
     def __init__(self):
-        # Initialize background remover with direct processing
-        # OpenMP conflicts are prevented by environment variables set at module level
-        print("Initializing BackgroundRemover with direct processing (no multiprocessing needed)")
-        
-        # Pre-initialize session to download models and avoid runtime downloads
-        # This prevents subprocess calls during actual background removal
-        try:
-            print("Pre-loading rembg models to prevent runtime downloads...")
-            self.session = new_session('u2net')  # Pre-load the default model
-            print("rembg models pre-loaded successfully!")
-        except Exception as e:
-            print(f"Warning: Could not pre-load rembg models: {e}")
-            print("Models will be downloaded on first use (may cause OpenMP issues)")
-            self.session = None
+        self.session = new_session()
 
     def __call__(self, image: Image.Image):
-        if not isinstance(image, Image.Image):
-            print(f"Warning: Expected PIL Image, got {type(image)}. Attempting conversion...")
-            try:
-                if hasattr(image, 'convert'):
-                    image = image.convert('RGBA')
-                else:
-                    # Try to create PIL Image from input
-                    image = Image.fromarray(np.asarray(image))
-            except Exception as e:
-                print(f"Could not convert input to PIL Image: {e}")
-                return image
-        
-        try:
-            # Ensure image is in RGBA mode
-            if image.mode != 'RGBA':
-                image = image.convert('RGBA')
-            
-            print("Starting background removal (OpenMP conflicts handled by environment variables)...")
-            
-            # Standard background removal
-            try:
-                # Use pre-loaded session if available, otherwise create new one
-                session = self.session if self.session is not None else new_session('u2net')
-                output = remove(
-                    image,
-                    session=session,
-                    bgcolor=[255, 255, 255, 0],
-                )
-                print("Background removal successful!")
-                return output
-            except Exception as e:
-                error_msg = str(e)
-                print(f"Background removal failed: {e}")
-                
-                # Check for specific OpenMP fork errors
-                if "fork()" in error_msg and "OpenMP" in error_msg:
-                    print("DETECTED: OpenMP fork() error in rembg library!")
-                    print("The rembg library itself has internal multiprocessing that conflicts with OpenMP.")
-                    print("This is a known issue with the rembg library in daemon environments.")
-                else:
-                    import traceback
-                    traceback.print_exc()
-            
-            # If all else fails, return the original image
-            print("Background removal completely failed, returning original image")
-            return image
-            
-        except Exception as e:
-            print(f"Background removal error in main process: {e}")
-            traceback.print_exc()
-            # If all else fails, just return the original image
-            # This ensures the pipeline doesn't break completely
-            return image
+        output = remove(image, session=self.session, bgcolor=[255, 255, 255, 0])
+        return output