Update gradio_app.py

gradio_app.py

@@ -1,5 +1,4 @@
 import os
-import base64
 import tempfile
 from typing import Any
 import torch
@@ -60,13 +59,11 @@ def create_batch(input_image: Image) -> dict[str, Any]:
     }
     return batch
 
-def process_image(image_base64: str) -> str:
-    """Process image and return GLB as base64."""
+def process_image(image_path: str) -> str:
+    """Process image and return path to GLB file."""
     try:
-        # Decode base64 image
-        image_data = base64.b64decode(image_base64)
-        input_image = Image.open(tempfile.SpooledTemporaryFile(suffix='.png'))
-        input_image.frombytes(image_data)
+        # Load image
+        input_image = Image.open(image_path)
         
         # Remove background if needed
         if input_image.mode != 'RGBA':
@@ -100,14 +97,7 @@ def process_image(image_base64: str) -> str:
         temp_file = tempfile.NamedTemporaryFile(suffix='.glb', delete=False)
         trimesh_mesh.export(temp_file.name, file_type="glb", include_normals=True)
         
-        # Convert to base64
-        with open(temp_file.name, 'rb') as f:
-            glb_base64 = base64.b64encode(f.read()).decode('utf-8')
-            
-        # Cleanup
-        os.unlink(temp_file.name)
-        
-        return glb_base64
+        return temp_file.name
         
     except Exception as e:
         return str(e)
@@ -115,11 +105,17 @@ def process_image(image_base64: str) -> str:
 # Create Gradio interface
 demo = gr.Interface(
     fn=process_image,
-    inputs=gr.Text(label="Base64 Image"),
-    outputs=gr.Text(label="Base64 GLB"),
+    inputs=gr.File(
+        label="Upload Image",
+        file_types=["image"],
+    ),
+    outputs=gr.File(
+        label="Download GLB",
+        file_types=[".glb"],
+    ),
     title="SPAR3D Image to GLB Converter",
-    description="Upload a base64-encoded image and get back a base64-encoded GLB file"
+    description="Upload an image (JPG, PNG, or WebP) and get back a 3D model in GLB format",
 )
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch()