Upload gradio_app.py

gradio_app.py

@@ -8,7 +8,7 @@ from PIL import Image
 import gradio as gr
 import trimesh
 from transparent_background import Remover
-
+from pathlib import Path
 import subprocess
 
 def install_cuda_toolkit():
@@ -41,6 +41,8 @@ COND_HEIGHT = 512
 COND_DISTANCE = 2.2
 COND_FOVY = 0.591627
 BACKGROUND_COLOR = [0.5, 0.5, 0.5]
+OUTPUT_DIR = "output"
+os.makedirs(OUTPUT_DIR, exist_ok=True)
 
 # Initialize models
 device = spar3d_utils.get_device()
@@ -208,8 +210,9 @@ def generate_and_process_3d(image: Image.Image) -> tuple[str | None, Image.Image
                 trimesh_mesh = trimesh_mesh[0]
 
         # Export to GLB
-        temp_dir = tempfile.mkdtemp()
-        output_path = os.path.join(temp_dir, 'output.glb')
+        #temp_dir = tempfile.mkdtemp()
+        #output_path = os.path.join(temp_dir, 'output.glb')
+        output_path = os.path.join(OUTPUT_DIR, 'output.glb')
         
         trimesh_mesh.export(output_path, file_type="glb", include_normals=True)
         
@@ -245,4 +248,4 @@ with gr.Blocks() as demo:
     )
     
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch(ssr_mode=False, allowed_paths=[Path(OUTPUT_DIR).resolve()])