fixes to interface

app.py

@@ -441,7 +441,12 @@ def generate_splat(selected_image, progress=gr.Progress(track_tqdm=True)):
         register_temp_file(str(ply_path))
 
         status_msg = f"✅ Generated {num_gaussians:,} gaussians | PLY: {ply_path.stat().st_size/1024:.1f}KB"
-        return str(ply_path), status_msg
+        return (
+            gr.update(visible=True, value=str(ply_path)),  # ply_download
+            str(ply_path),  # ply_path_state
+            gr.update(visible=True),  # convert_splat_btn
+            status_msg,  # splat_status
+        )
 
     except gr.Error:
         raise
@@ -452,6 +457,26 @@ def generate_splat(selected_image, progress=gr.Progress(track_tqdm=True)):
         raise gr.Error(f"Failed to generate 3D splat: {str(e)}")
 
 
+def convert_and_save_splat(ply_path):
+    """Convert PLY to SPLAT format and save to temp file."""
+    if not ply_path or not os.path.exists(ply_path):
+        raise gr.Error("PLY file not found. Please generate a 3D splat first.")
+
+    try:
+        splat_data = convert_ply_to_splat(ply_path)
+        splat_file = tempfile.NamedTemporaryFile(suffix=".splat", delete=False)
+        splat_file.write(splat_data)
+        splat_file.close()
+        register_temp_file(splat_file.name)
+
+        size_kb = os.path.getsize(splat_file.name) / 1024
+        status_msg = f"✅ SPLAT file created | Size: {size_kb:.1f}KB"
+        return status_msg, gr.update(visible=True, value=splat_file.name)
+    except Exception as e:
+        print(f"Error converting to SPLAT: {e}")
+        raise gr.Error(f"Failed to convert to SPLAT: {str(e)}")
+
+
 @spaces.GPU
 def generate(
     prompt: str,
@@ -639,8 +664,9 @@ with gr.Blocks(title="Z-Image Demo") as demo:
 a 3D Gaussian splat model. You can then download PLY files or SPLAT in an additional step."""
                 )
 
-                # State to hold selected image
+                # State to hold selected image and PLY path
                 selected_image_state = gr.State(value=None)
+                ply_path_state = gr.State(value=None)
 
                 with gr.Row():
                     generate_splat_btn = gr.Button(
@@ -648,6 +674,11 @@ a 3D Gaussian splat model. You can then download PLY files or SPLAT in an additi
                         variant="secondary",
                         interactive=False,
                     )
+                    convert_splat_btn = gr.Button(
+                        "Convert to SPLAT",
+                        variant="secondary",
+                        visible=False,
+                    )
 
                 splat_status = gr.Textbox(
                     label="Status",
@@ -659,7 +690,11 @@ a 3D Gaussian splat model. You can then download PLY files or SPLAT in an additi
                 # Download buttons
                 with gr.Row():
                     ply_download = gr.File(
-                        label="Download PLY File",
+                        label="Download PLY",
+                        visible=False,
+                    )
+                    splat_download = gr.File(
+                        label="Download SPLAT",
                         visible=False,
                     )
 
@@ -696,11 +731,14 @@ a 3D Gaussian splat model. You can then download PLY files or SPLAT in an additi
     generate_splat_btn.click(
         generate_splat,
         inputs=[selected_image_state],
-        outputs=[ply_download, splat_status],
-    ).success(
-        lambda ply_path, status: gr.update(visible=True, value=ply_path) if ply_path and os.path.exists(ply_path) else gr.update(visible=False),
-        inputs=[ply_download, splat_status],
-        outputs=[ply_download],
+        outputs=[ply_download, ply_path_state, convert_splat_btn, splat_status],
+    )
+
+    # SPLAT conversion handler
+    convert_splat_btn.click(
+        convert_and_save_splat,
+        inputs=[ply_path_state],
+        outputs=[splat_status, splat_download],
     )
 
     def update_res_choices(res_cat_value):