support meta json file download

app.py

@@ -232,9 +232,6 @@ def SRT_quat_to_matrix(scale, quat, translation):
 
 def write_output(primitives, name):
     out_json = {}
-    out_json['operation'] = 0
-    out_json['type'] = 1
-    out_json['scene_id'] = None
 
     new_group = []
     model_scene = trimesh.Scene()
@@ -255,7 +252,6 @@ def write_output(primitives, name):
         new_block['data']['location'] = translation.tolist()
         new_block['data']['rotation'] = euler_to_quat(rotation).tolist()
         new_block['data']['scale'] = scale.tolist()
-        new_block['data']['color'] = ['808080']
         new_group.append(new_block)
 
         trans = SRT_quat_to_matrix(scale, euler_to_quat(rotation), translation)
@@ -344,7 +340,13 @@ def process_3d_model(input_3d, dilated_offset, do_marching_cubes, postprocess_me
         do_marching_cubes=do_marching_cubes,
         postprocess=postprocess_method
     )
-    return output_model_obj
+    
+    # Save JSON to a file
+    json_path = os.path.join(LOG_PATH, f'output_{os.path.basename(input_3d)[:-4]}.json')
+    with open(json_path, 'w') as f:
+        json.dump(output_model_json, f, indent=4)
+    
+    return output_model_obj, json_path
     # except Exception as e:
     #     return f"Error processing file: {str(e)}"
 
@@ -394,12 +396,13 @@ with gr.Blocks(title="PrimitiveAnything: Human-Crafted 3D Primitive Assembly Gen
             
         with gr.Column():
             # Output components
-            output = gr.Model3D(label="Primitive Assembly Predition")
+            output_3d = gr.Model3D(label="Primitive Assembly Prediction")
+            output_json = gr.File(label="Download JSON File")
             
     submit_btn.click(
         fn=process_3d_model,
         inputs=[input_3d, dilated_offset, do_marching_cubes],
-        outputs=output
+        outputs=[output_3d, output_json]
     )