CSVNMR_To_MATLABFile

import tkinter as tk
from tkinter import filedialog, messagebox
import pandas as pd
import numpy as np
import scipy.io as sio
import os

class NMRDataProcessor:
    def __init__(self, root):
        self.root = root
        self.root.title("MRTools - NMR Data Processor")
        self.root.geometry("500x300")

        self.file_path = tk.StringVar()
        self.save_path = tk.StringVar()

        # Interface
        self.create_widgets()

    def create_widgets(self):

        main_frame = tk.Frame(self.root, padx=20, pady=20)
        main_frame.pack(expand=True, fill=tk.BOTH)

        # Botão para selecionar arquivo
        tk.Label(main_frame, text="Select your data files (Excel/CSV):").pack(anchor=tk.W)
        file_frame = tk.Frame(main_frame)
        file_frame.pack(fill=tk.X, pady=5)
        tk.Entry(file_frame, textvariable=self.file_path, width=40).pack(side=tk.LEFT, fill=tk.X, expand=True)
        tk.Button(file_frame, text="Search", command=self.select_file).pack(side=tk.LEFT, padx=5)

        # Botão para selecionar local de salvamento
        tk.Label(main_frame, text="Save your file .mat as:").pack(anchor=tk.W, pady=(15,0))
        save_frame = tk.Frame(main_frame)
        save_frame.pack(fill=tk.X, pady=5)
        tk.Entry(save_frame, textvariable=self.save_path, width=40).pack(side=tk.LEFT, fill=tk.X, expand=True)
        tk.Button(save_frame, text="Search", command=self.select_save_location).pack(side=tk.LEFT, padx=5)


        tk.Button(main_frame, text="Process and Save", command=self.process_data,
                 bg='#4CAF50', fg='white').pack(pady=20)

        # Status
        self.status_label = tk.Label(main_frame, text="", fg='gray')
        self.status_label.pack()

    def select_file(self):
        filetypes = (
            ('Excel/CSV files', '*.xlsx *.csv'),
            ('All files', '*.*')
        )
        filename = filedialog.askopenfilename(
            title='Select you data file',
            filetypes=filetypes
        )
        if filename:
            self.file_path.set(filename)

            base_name = os.path.splitext(os.path.basename(filename))[0] + '_nmrdata.mat'
            self.save_path.set(os.path.join(os.path.dirname(filename), base_name))

    def select_save_location(self):
        filename = filedialog.asksaveasfilename(
            title='Save your file .mat as',
            defaultextension='.mat',
            filetypes=(('MATLAB Data', '*.mat'),)
        )
        if filename:
            self.save_path.set(filename)

    def process_data(self):
        if not self.file_path.get():
            messagebox.showerror("Error", "No file selected!")
            return

        if not self.save_path.get():
            messagebox.showerror("Error", "You must select a save location!")
            return

        try:
            self.status_label.config(text="Processing...", fg='blue')
            self.root.update()


            file_path = self.file_path.get()
            ext = os.path.splitext(file_path)[1].lower()

            if ext == '.xlsx':

                df = pd.read_excel(file_path, header=0)
                headers = df.columns.tolist()
                data = df.values
            elif ext == '.csv':
                # Lê CSV
                df = pd.read_csv(file_path, header=0)
                headers = df.columns.tolist()
                data = df.values
            else:
                raise ValueError("Wrong file type!")

            ppm_mean = data[:, 0].T
            sample_data = data[:, 1:].T
            sample_names = headers[1:]

            nmrdata = {
                'data': sample_data,
                'ppm_mean': ppm_mean,
                'subjects': np.array(sample_names, dtype='object')
            }

            sio.savemat(self.save_path.get(), {'nmrdata': nmrdata})

            self.status_label.config(text="File saved!", fg='green')
            messagebox.showinfo("Sucess", "Processed data and saved!")

        except Exception as e:
            self.status_label.config(text="Process failure", fg='red')
            messagebox.showerror("Error", f"Something is wrong!:\n{str(e)}")

if __name__ == "__main__":
    root = tk.Tk()
    app = NMRDataProcessor(root)
    root.mainloop()