Remove text output, display only slice visualization

app.py

@@ -7,7 +7,7 @@ matplotlib.use('Agg')
 
 def visualize_nifti(file, volume_idx=0, slice_idx=0):
     if file is None:
-        return None, "No file uploaded."
+        return None
     
     try:
         # Load the NIfTI file
@@ -31,7 +31,7 @@ def visualize_nifti(file, volume_idx=0, slice_idx=0):
             plt.title(f'Axial Slice: {slice_idx}/{max_slice}')
             plt.tight_layout()
             
-            return fig, f'3D Volume - Displaying Slice {slice_idx}/{max_slice}'
+            return fig
             
         elif len(shape) == 4:
             # 4D data
@@ -50,12 +50,12 @@ def visualize_nifti(file, volume_idx=0, slice_idx=0):
             plt.title(f'Volume: {volume_idx}/{max_volume}, Slice: {slice_idx}/{max_slice}')
             plt.tight_layout()
             
-            return fig, f'4D Volume - Displaying Volume {volume_idx}/{max_volume}, Slice {slice_idx}/{max_slice}'
+            return fig
         else:
-            return None, f"Unsupported dimensions: {len(shape)}D"
+            return None
             
     except Exception as e:
-        return None, f"Error loading file: {str(e)}"
+        return None
 
 # Create Gradio interface with Blocks for dynamic controls
 with gr.Blocks() as demo:
@@ -75,13 +75,10 @@ with gr.Blocks() as demo:
     with gr.Row():
         output_plot = gr.Plot(label="Slice Visualization")
     
-    with gr.Row():
-        output_text = gr.Textbox(label="Info", lines=2)
-    
     visualize_btn.click(
         fn=visualize_nifti,
         inputs=[file_input, volume_slider, slice_slider],
-        outputs=[output_plot, output_text]
+        outputs=[output_plot]
     )
 
 if __name__ == "__main__":