SaveFig.py

import os
import CheckCreateDirectory

from PyQt5.QtWidgets import (
    QFileDialog,
    QMessageBox,
)


def save_fig(self, fig, name):
        """
        Saves the current figure as a PNG image.

        This method prompts the user to select a file name and location to save the
        figure as a PNG image.
        """
        path = ".\\Results"
        CheckCreateDirectory.check_create_dir(path)

        file_name, _ = QFileDialog.getSaveFileName(self, 'Save Figure', os.path.join(os.getcwd(), 'Results', name), 'PNG (*.png)')

        if file_name:
            try:
                fig.savefig(file_name)
                QMessageBox.information(self, 'Success', 'Figure saved successfully.')
            except Exception as e:
                QMessageBox.warning(self, 'Error', f'An error occurred while saving the figure: {e}')