Upload gradio_app.py

gradio_app.py

@@ -9,6 +9,7 @@ import gradio as gr
 import trimesh
 from transparent_background import Remover
 from pathlib import Path
+import uuid
 import subprocess
 
 def install_cuda_toolkit():
@@ -210,14 +211,14 @@ 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')
-        output_path = os.path.join(OUTPUT_DIR, 'output.glb')
-        model_dl.append(output_path)
+        unique_id = str(uuid.uuid4())
+        filename = f'model_{unique_id}.glb'
+        output_path = os.path.join(OUTPUT_DIR, filename)
+        public_url = f"https://john6666-image-to-3d-test.hf.space/file={output_path}"
         
         trimesh_mesh.export(output_path, file_type="glb", include_normals=True)
         
-        return output_path
+        return output_path, public_url
         
     except Exception as e:
         print(f"Error during generation: {str(e)}")
@@ -238,9 +239,10 @@ with gr.Blocks() as demo:
         model_output = gr.Model3D(
             label="Generated .GLB model",
             clear_color=[0.0, 0.0, 0.0, 0.0],
+            visible=False
         )
-        model_dl = gr.File(label="Download generated .GLB model", value=[], interactive=False)
-    
+        output_url = gr.Textbox(label="Output URL", value="", lines=1, interactive=False)
+
     # Event handler
     input_img.upload(
         fn=generate_and_process_3d,