Refactor gen_image function in app.py to create a temporary GLB file with a proper extension and return its path. This change enhances file management by ensuring the generated file is correctly handled and simplifies the cleanup process.

app.py

@@ -108,13 +108,19 @@ def gen_image(input_image, seed, scale, step):
     
     glb_path = generate3d(model, np_imgs, np_xyzs, args.device)
     
-    # Read the GLB file and encode it in base64
-    with open(glb_path, 'rb') as f:
-        glb_bytes = f.read()
-    encoded_glb = 'data:model/gltf-binary;base64,' + base64.b64encode(glb_bytes).decode('utf-8')
+    # Create a temporary file with a proper name for the GLB data
+    import tempfile
+    import shutil
     
-    # Return images and the encoded GLB data
-    return Image.fromarray(np_imgs), Image.fromarray(np_xyzs), encoded_glb
+    # Create a temporary file with a proper extension
+    temp_glb = tempfile.NamedTemporaryFile(suffix='.glb', delete=False)
+    temp_glb.close()
+    
+    # Copy the generated GLB file to our temporary file
+    shutil.copy2(glb_path, temp_glb.name)
+    
+    # Return images and the path to the temporary GLB file
+    return Image.fromarray(np_imgs), Image.fromarray(np_xyzs), temp_glb.name
 
 
 parser = argparse.ArgumentParser()
@@ -243,4 +249,4 @@ with gr.Blocks() as demo:
         inputs=inputs,
         outputs=outputs,
     )
-    demo.queue().launch()
+    demo.queue().launch()