Updated app.py

app.py

@@ -64,7 +64,7 @@ def run_nnunet_predict(nifti_file):
 
     # Extract the original filename without the extension
     original_filename = os.path.basename(nifti_file.name)
-    base_filename = original_filename.replace(".nii.gz", "").replace("_0000", "")
+    base_filename = original_filename.replace(".nii.gz", "")
     
     # Save the uploaded file to the input directory
     input_path = os.path.join(INPUT_DIR, "image_0000.nii.gz")
@@ -105,7 +105,7 @@ def run_nnunet_predict(nifti_file):
             extract_middle_slice(new_output_file, output_slice_path)
 
             # Return paths for the Gradio interface
-            return input_slice_path, output_slice_path, new_output_file
+            return new_output_file, input_slice_path, output_slice_path
         else:
             return "Error: Output file not found."
     except subprocess.CalledProcessError as e: