import os
import tkinter as tk
from tkinter import ttk, filedialog, simpledialog, messagebox
import originpro as op
import pandas as pd
import sys


# Code snippet 1: Rename columns
def rename_columns():
    def rename_columns_action():
        # Select the CSV file
        file_path = filedialog.askopenfilename(filetypes=[("CSV Files", "*.csv")])

        if file_path:
            try:
                # Prompt for start and end voltage values
                start_voltage = float(simpledialog.askstring("Start Voltage", "Enter the starting voltage:"))
                mid_voltage = float(simpledialog.askstring("Mid Voltage", "Enter the ending voltage:"))

                # Read the CSV file
                df = pd.read_csv(file_path)

                # Get the headers
                headers = df.columns.tolist()

                # Calculate the voltage step
                step_up = (mid_voltage - start_voltage) / (len(headers) // 2 - 1)
                step_down = (start_voltage - mid_voltage) / (len(headers) // 2 - 1)

                # Rename the columns
                for i in range(1, len(headers)):
                    if i <= len(headers) // 2:
                        voltage = start_voltage + (i - 1) * step_up
                    else:
                        voltage = mid_voltage + (i - len(headers) // 2 - 1) * step_down
                    new_header = "{:.2f}".format(voltage)  # Format the voltage with 2 decimal places
                    df.rename(columns={headers[i]: new_header}, inplace=True)

                # Save the renamed data to XLSX file
                save_path = os.path.splitext(file_path)[0] + "_renamed.xlsx"
                df.to_excel(save_path, index=False)

                # Open the folder where the output files are saved
                os.startfile(save_path)

                messagebox.showinfo("Success", "Columns renamed successfully. Saved as " + save_path)
            except Exception as e:
                messagebox.showerror("Error", "An error occurred: " + str(e))

    rename_columns_action()


root = tk.Tk()
root.withdraw()


# End of Code snippet 1: Rename columns

# code 2 start
def bg_processing():
    # Select the CSV file
    xlsx_file_path = filedialog.askopenfilename(title="Select Input XLSX File", filetypes=[("XLSX Files", "*.xlsx")])

    if xlsx_file_path:
        try:
            # Read the input XLSX file and rename the sheets
            xlsx = pd.ExcelFile(xlsx_file_path)
            sheets = xlsx.sheet_names
            df = None  # Initialize df variable
            wavenumber_column = None  # Initialize wavenumber_column variable

            for i, sheet in enumerate(sheets):
                df_sheet = pd.read_excel(xlsx_file_path, sheet_name=sheet)
                df_sheet.to_excel(xlsx_file_path, sheet_name=f"Sheet{i + 1}", index=False)

                if i == 0:
                    df = df_sheet
                    wavenumber_column = df_sheet["Wavenumber"]

            # Create a Tkinter window for column selection
            column_window = tk.Tk()
            column_window.title("Select Column")
            column_window.geometry("300x100")

            # Create a label for column selection
            column_label = ttk.Label(column_window, text="Choose a column:")
            column_label.pack()

            # Create a combobox for column selection
            column_combobox = ttk.Combobox(column_window, values=df.columns.tolist())
            column_combobox.pack()

            # Create a button to confirm column selection
            confirm_button = ttk.Button(column_window, text="Confirm", command=column_window.quit)
            confirm_button.pack()

            # Run the column selection window
            column_window.mainloop()
            # Get the chosen column
            chosen_column = column_combobox.get()

            # Create a new sheet for processing
            processed_sheet = pd.DataFrame()
            processed_sheet["Wavenumber"] = wavenumber_column

            for column in df.columns[1:]:
                if column == chosen_column:
                    processed_sheet[column] = 0
                else:
                    processed_sheet[column] = df[column] - df[chosen_column]

            # Prompt for the output directory
            xlsx_output_dir = filedialog.askdirectory(title="Select Output Directory")

            # Get the input filename without extension
            input_xlsx_file_name = os.path.splitext(os.path.basename(xlsx_file_path))[0]

            # Construct the output file path
            output_xlsx_file_name = f"{input_xlsx_file_name}_{chosen_column}.xlsx"
            output_xlsx_file_path = os.path.join(xlsx_output_dir, output_xlsx_file_name)

            # Save the processed sheet to a new workbook
            with pd.ExcelWriter(output_xlsx_file_path, engine="openpyxl") as writer:
                processed_sheet.to_excel(writer, sheet_name="Sheet1", index=False)

            # Open the folder where the output files are saved
            os.startfile(xlsx_output_dir)

            # Show completion message
            message = f"Processing completed! Output saved as:\n{output_xlsx_file_path}"
            messagebox.showinfo("Processing Complete", message)
            column_window.destroy()
        except Exception as e:
            messagebox.showerror("Error", "An error occurred: " + str(e))


root = tk.Tk()
root.withdraw()


# code 2 end

# Code snippet 3: Create Origin graphs
def create_origin_graphs():
    def origin_shutdown_exception_hook(exctype, value, traceback):
        """Ensures Origin gets shut down if an uncaught exception"""
        op.exit()
        sys.__excepthook__(exctype, value, traceback)

    sys.excepthook = origin_shutdown_exception_hook

    # Only run if external Python
    if op.oext:
        # Create a new Origin project
        op.new()
        op.set_show(True)

    try:
        # Prompt user to select an Origin template file for graph 1
        template_file_path_1 = filedialog.askopenfilename(title="Select Origin Template File for Graph 1",
                                                          filetypes=[("Origin Template Files", "*.otpu")])

        # Prompt user to select a data file (CSV or XLSX) for the selected template
        data_file_types = [("CSV Files", "*.csv"), ("XLSX Files", "*.xlsx")]
        csv_file_path_1 = filedialog.askopenfilename(
            title=f"Select Data File for {os.path.basename(template_file_path_1)}",
            filetypes=data_file_types)

        # Load the CSV file into Origin worksheet
        wks = op.new_sheet()
        wks.from_file(csv_file_path_1, False)

        # Create a new graph using the selected template for graph 1
        gr1 = op.new_graph(template=template_file_path_1)
        gr1[0].add_plot(wks, 1, 0)
        gr1[0].rescale()

        # Add more graphs
        more_graphs = True
        graph_num = 2

        while more_graphs:
            # Prompt user to select an Origin template file for the graph
            template_file_path = filedialog.askopenfilename(title=f"Select Origin Template File for Graph {graph_num}",
                                                            filetypes=[("Origin Template Files", "*.otpu")])

            # Prompt user to select an XLSX file for the graph
            xlsx_file_path = filedialog.askopenfilename(
                title=f"Select XLSX File for {os.path.basename(template_file_path)}",
                filetypes=[("XLSX Files", "*.xlsx")])

            # Load the XLSX file into a new worksheet
            wks = op.new_sheet()
            wks.from_file(xlsx_file_path, False)

            # Create a graph page with the user-selected template for the graph
            gr = op.new_graph(template=template_file_path)
            gl = gr[0]

            # Prompt user for the range of columns to plot for the graph
            column_range = simpledialog.askstring("Range of Columns",
                                                  f"Enter the range of columns (e.g., 1-12) for Graph {graph_num}:")

            start_col, end_col = map(int, column_range.split("-"))
            y_columns = list(range(start_col, end_col + 1))

            for col in y_columns:
                plot = gl.add_plot(wks, col, 0)

            # Group and Rescale the graph
            gl.group()
            gl.rescale()

            # Prompt user if they want to add more graphs
            response = tk.messagebox.askyesno("Add More Graphs", "Do you want to add more graphs?")

            if response:
                graph_num += 1
            else:
                more_graphs = False

        # Tile all windows
        op.lt_exec('win-s T')

        # Prompt user for the output Origin file name
        output_file_path = filedialog.asksaveasfilename(title="Save Output Origin File",
                                                        filetypes=[("Origin Project Files", "*.opju")])

        # Creating a notes window
        nt = op.new_notes()

        # Appending input information to the notes
        nt.append("Input Information:")
        nt.append(f"Graph 1 Template: {os.path.basename(template_file_path_1)}")
        nt.append(f"Graph 1 CSV File: {os.path.basename(csv_file_path_1)}")

        for i in range(2, graph_num + 1):
            template_var = locals().get(f"template_file_path_{i}")
            csv_var = locals().get(f"csv_file_path_{i}")

            if template_var and csv_var:
                nt.append(f"\nGraph {i} Template: {os.path.basename(template_var)}")
                nt.append(f"Graph {i} CSV File: {os.path.basename(csv_var)}")

        # Appending folder path information to the notes
        nt.append("\nFolder Paths:")
        nt.append(f"Graph 1 Folder: {os.path.dirname(template_file_path_1)}")

        for i in range(2, graph_num + 1):
            template_var = locals().get(f"template_file_path_{i}")

            if template_var:
                nt.append(f"Graph {i} Folder: {os.path.dirname(template_var)}")

        nt.append(f"Output Folder: {os.path.dirname(output_file_path)}")

        # Appending output file information to the notes
        nt.append("\nOutput Information:")
        nt.append(f"Output File: {os.path.basename(output_file_path)}")

        # Displaying the note
        nt.view = 1

        # Tile all windows
        op.lt_exec('win-s T')

        # Save the project to the specified output file path
        if op.oext:
            output_file_path = os.path.abspath(output_file_path)
            op.save(output_file_path)

    except Exception as e:
        print(f"An error occurred: {str(e)}")
        op.exit()


# End of Code snippet 3: Create Origin graphs


# Code snippet 4: Add graphs to existing Origin project
def add_graphs_to_project():
    def origin_shutdown_exception_hook(exctype, value, traceback):
        """Ensures Origin gets shut down if an uncaught exception"""
        op.exit()
        sys.__excepthook__(exctype, value, traceback)

    sys.excepthook = origin_shutdown_exception_hook

    # Only run if external Python
    if op.oext:
        op.set_show(True)

    def save_origin_project(output_path):
        try:
            op.save(output_path)
            return True  # Save operation successful
        except PermissionError:
            return False  # Save operation failed due to read-only
        except Exception as e:
            print(f"An error occurred while saving: {str(e)}")
            return False  # Save operation failed for other reasons

    def add_graphs_to_project_action():
        try:
            # Prompt user to select an existing Origin project file
            origin_file_path = filedialog.askopenfilename(title="Select Existing Origin Project File",
                                                          filetypes=[("Origin Project Files", "*.opju")])

            # Load the existing Origin project file
            op.open(origin_file_path)

            # Add more graphs
            more_graphs = True
            graph_num = 1

            while more_graphs:
                # Prompt user to select an Origin template file for the graph
                template_file_path = filedialog.askopenfilename(
                    title=f"Select Origin Template File for Graph {graph_num}",
                    filetypes=[("Origin Template Files", "*.otpu")])

                # Prompt user to select an XLSX file for the graph
                data_file_types = [("CSV Files", "*.csv"), ("XLSX Files", "*.xlsx")]
                xlsx_file_path = filedialog.askopenfilename(
                    title=f"Select XLSX File for {os.path.basename(template_file_path)}",
                    filetypes=data_file_types)

                # Load the XLSX file into a new worksheet
                wks = op.new_sheet()
                wks.from_file(xlsx_file_path, False)

                # Create a graph page with the user-selected template for the graph
                gr = op.new_graph(template=template_file_path)
                gl = gr[0]

                # Prompt user for the range of columns to plot for the graph
                column_range = simpledialog.askstring("Range of Columns",
                                                      f"Enter the range of columns (e.g., 1-12) for Graph {graph_num}:")

                start_col, end_col = map(int, column_range.split("-"))
                y_columns = list(range(start_col, end_col + 1))

                for col in y_columns:
                    gl.add_plot(wks, col, 0)

                # Group and Rescale the graph
                gl.group()
                gl.rescale()

                # Prompt user if they want to add more graphs
                response = tk.messagebox.askyesno("Add More Graphs", "Do you want to add more graphs?")

                if response:
                    graph_num += 1
                else:
                    more_graphs = False

            # Tile all windows
            op.lt_exec('win-s T')

            # Prompt user for the output Origin project file name and location
            output_file_path = filedialog.asksaveasfilename(title="Save Output Origin File As",
                                                            filetypes=[("Origin Project Files", "*.opju")])

            if not output_file_path:
                print("Save operation canceled by user.")
                op.exit()
                return  # Exit the function if the user cancels the save operation

            # Try to save the project file
            output_file_path = os.path.abspath(output_file_path)
            success = save_origin_project(output_file_path)

            # If the save operation failed due to read-only, prompt user for a new file path
            while not success:
                new_output_path = filedialog.asksaveasfilename(title="Save Output Origin File As",
                                                               filetypes=[("Origin Project Files", "*.opju")])
                if not new_output_path:
                    print("Save operation canceled by user.")
                    op.exit()
                    break  # Exit the loop if the user cancels the save operation

                new_output_path = os.path.abspath(new_output_path)
                success = save_origin_project(new_output_path)

                if success:
                    message = f"Processing completed! Output saved as:\n{new_output_path}"
                    messagebox.showinfo("Processing Complete", message)
                    op.exit()
                    break  # Exit the loop if the new save is successful
                else:
                    # Display an error message if the new save location is also read-only
                    messagebox.showerror("Error",
                                         "Selected save location is read-only. Please choose a different location.")
                    continue  # Continue the loop and prompt the user again

        except Exception as e:
            print(f"An error occurred: {str(e)}")
            op.exit()

    add_graphs_to_project_action()


root = tk.Tk()
root.withdraw()


# End of Code snippet 4: Add graphs to existing Origin project


# Code snippet 5: Function to exit the application
def exit_application():
    try:
        # Close the active Origin project
        op.exit()
        window.destroy()  # Close the main GUI window, which ends the tkinter event loop
    except Exception as e:
        print(f"An error occurred while closing Origin: {str(e)}")


# End of Code snippet 5: Function to exit the application


# Create the main GUI window
window = tk.Tk()
window.title("FTIR Data Processing")
window.geometry("300x230")

# Create a frame to contain the buttons and align it to the left
button_frame = tk.Frame(window, padx=20, pady=20)
button_frame.pack(anchor="w")  # Use "w" to anchor (justify) the frame to the left

# Create buttons with some styling
button_style = {
    "font": ("Helvetica", 14),
    "width": 30,
    "height": 2,
    "bd": 2,  # Border thickness
    "relief": "solid",  # Border style
}

# Create buttons for each functionality
rename_columns_button = tk.Button(window, text="Step 1: Rename Columns", command=rename_columns, bg="light blue")
process_background_data_button = tk.Button(window, text="Step 2: Reprocess Background", command=bg_processing,
                                           bg="light blue")
create_origin_graphs_button = tk.Button(window, text="Step 3: Create Origin Project to add Graphs",
                                        command=create_origin_graphs, bg="light blue")
add_to_project_button = tk.Button(window, text="Step 4: Add Graphs to Existing Origin Project",
                                  command=add_graphs_to_project, bg="light blue")
exit_button = tk.Button(window, text="Exit Application", command=exit_application, bg="red")

# Pack the buttons with some spacing and justify to the left
rename_columns_button.pack(pady=10, anchor="w")
process_background_data_button.pack(pady=10, anchor="w")
create_origin_graphs_button.pack(pady=10, anchor="w")
add_to_project_button.pack(pady=10, anchor="w")
exit_button.pack(pady=10, anchor="w", )

# Start the tkinter event loop
window.mainloop()

def exit_application():
    try:
        # Close the active Origin project
        op.exit()
        #window.destroy()  # Close the main GUI window, which ends the tkinter event loop
        window.quit()

import os
import tkinter as tk
from tkinter import ttk, filedialog, simpledialog, messagebox
import originpro as op
import pandas as pd
import sys


# Code snippet 1: Rename columns
def rename_columns():
    def rename_columns_action():
        # Select the CSV file
        file_path = filedialog.askopenfilename(filetypes=[("CSV Files", "*.csv")])

        if file_path:
            try:
                # Prompt for start and end voltage values
                start_voltage = float(simpledialog.askstring("Start Voltage", "Enter the starting voltage:"))
                mid_voltage = float(simpledialog.askstring("Mid Voltage", "Enter the ending voltage:"))

                # Read the CSV file
                df = pd.read_csv(file_path)

                # Get the headers
                headers = df.columns.tolist()

                # Calculate the voltage step
                step_up = (mid_voltage - start_voltage) / (len(headers) // 2 - 1)
                step_down = (start_voltage - mid_voltage) / (len(headers) // 2 - 1)

                # Rename the columns
                for i in range(1, len(headers)):
                    if i <= len(headers) // 2:
                        voltage = start_voltage + (i - 1) * step_up
                    else:
                        voltage = mid_voltage + (i - len(headers) // 2 - 1) * step_down
                    new_header = "{:.2f}".format(voltage)  # Format the voltage with 2 decimal places
                    df.rename(columns={headers[i]: new_header}, inplace=True)

                # Save the renamed data to XLSX file
                save_path = os.path.splitext(file_path)[0] + "_renamed.xlsx"
                df.to_excel(save_path, index=False)

                # Open the folder where the output files are saved
                os.startfile(save_path)

                messagebox.showinfo("Success", "Columns renamed successfully. Saved as " + save_path)
            except Exception as e:
                messagebox.showerror("Error", "An error occurred: " + str(e))

    rename_columns_action()


root = tk.Tk()
root.withdraw()


# End of Code snippet 1: Rename columns

# code 2 start
def bg_processing():
    # Select the CSV file
    xlsx_file_path = filedialog.askopenfilename(title="Select Input XLSX File", filetypes=[("XLSX Files", "*.xlsx")])

    if xlsx_file_path:
        try:
            # Read the input XLSX file and rename the sheets
            xlsx = pd.ExcelFile(xlsx_file_path)
            sheets = xlsx.sheet_names
            df = None  # Initialize df variable
            wavenumber_column = None  # Initialize wavenumber_column variable

            for i, sheet in enumerate(sheets):
                df_sheet = pd.read_excel(xlsx_file_path, sheet_name=sheet)
                df_sheet.to_excel(xlsx_file_path, sheet_name=f"Sheet{i + 1}", index=False)

                if i == 0:
                    df = df_sheet
                    wavenumber_column = df_sheet["Wavenumber"]

            # Create a Tkinter window for column selection
            column_window = tk.Tk()
            column_window.title("Select Column")
            column_window.geometry("300x100")

            # Create a label for column selection
            column_label = ttk.Label(column_window, text="Choose a column:")
            column_label.pack()

            # Create a combobox for column selection
            column_combobox = ttk.Combobox(column_window, values=df.columns.tolist())
            column_combobox.pack()

            # Create a button to confirm column selection
            confirm_button = ttk.Button(column_window, text="Confirm", command=column_window.quit)
            confirm_button.pack()

            # Run the column selection window
            column_window.mainloop()
            # Get the chosen column
            chosen_column = column_combobox.get()

            # Create a new sheet for processing
            processed_sheet = pd.DataFrame()
            processed_sheet["Wavenumber"] = wavenumber_column

            for column in df.columns[1:]:
                if column == chosen_column:
                    processed_sheet[column] = 0
                else:
                    processed_sheet[column] = df[column] - df[chosen_column]

            # Prompt for the output directory
            xlsx_output_dir = filedialog.askdirectory(title="Select Output Directory")

            # Get the input filename without extension
            input_xlsx_file_name = os.path.splitext(os.path.basename(xlsx_file_path))[0]

            # Construct the output file path
            output_xlsx_file_name = f"{input_xlsx_file_name}_{chosen_column}.xlsx"
            output_xlsx_file_path = os.path.join(xlsx_output_dir, output_xlsx_file_name)

            # Save the processed sheet to a new workbook
            with pd.ExcelWriter(output_xlsx_file_path, engine="openpyxl") as writer:
                processed_sheet.to_excel(writer, sheet_name="Sheet1", index=False)

            # Open the folder where the output files are saved
            os.startfile(xlsx_output_dir)

            # Show completion message
            message = f"Processing completed! Output saved as:\n{output_xlsx_file_path}"
            messagebox.showinfo("Processing Complete", message)
            column_window.destroy()
        except Exception as e:
            messagebox.showerror("Error", "An error occurred: " + str(e))


root = tk.Tk()
root.withdraw()


# Code snippet 2 end

# Code snippet 3: Create Origin graphs
def create_origin_graphs():
    def origin_shutdown_exception_hook(exctype, value, traceback):
        """Ensures Origin gets shut down if an uncaught exception"""
        op.exit()
        sys.__excepthook__(exctype, value, traceback)

    sys.excepthook = origin_shutdown_exception_hook

    # Only run if external Python
    if op.oext:
        # Create a new Origin project
        op.new()
        op.set_show(True)

    try:
        # Prompt user to select an Origin template file for graph 1
        template_file_path_1 = filedialog.askopenfilename(title="Select Origin Template File for Graph 1",
                                                          filetypes=[("Origin Template Files", "*.otpu")])

        # Prompt user to select a data file (CSV or XLSX) for the selected template
        data_file_types = [("CSV Files", "*.csv"), ("XLSX Files", "*.xlsx")]
        csv_file_path_1 = filedialog.askopenfilename(
            title=f"Select Data File for {os.path.basename(template_file_path_1)}",
            filetypes=data_file_types)

        # Load the CSV file into Origin worksheet
        wks = op.new_sheet()
        wks.from_file(csv_file_path_1, False)

        # Create a new graph using the selected template for graph 1
        gr1 = op.new_graph(template=template_file_path_1)
        gr1[0].add_plot(wks, 1, 0)
        gr1[0].rescale()

        # Add more graphs
        more_graphs = True
        graph_num = 2

        while more_graphs:
            # Prompt user to select an Origin template file for the graph
            template_file_path = filedialog.askopenfilename(title=f"Select Origin Template File for Graph {graph_num}",
                                                            filetypes=[("Origin Template Files", "*.otpu")])

            # Prompt user to select an XLSX file for the graph
            xlsx_file_path = filedialog.askopenfilename(
                title=f"Select XLSX File for {os.path.basename(template_file_path)}",
                filetypes=[("XLSX Files", "*.xlsx")])

            # Load the XLSX file into a new worksheet
            wks = op.new_sheet()
            wks.from_file(xlsx_file_path, False)

            # Create a graph page with the user-selected template for the graph
            gr = op.new_graph(template=template_file_path)
            gl = gr[0]

            # Prompt user for the range of columns to plot for the graph
            column_range = simpledialog.askstring("Range of Columns",
                                                  f"Enter the range of columns (e.g., 1-12) for Graph {graph_num}:")

            start_col, end_col = map(int, column_range.split("-"))
            y_columns = list(range(start_col, end_col + 1))

            for col in y_columns:
                plot = gl.add_plot(wks, col, 0)

            # Group and Rescale the graph
            gl.group()
            gl.rescale()

            # Prompt user if they want to add more graphs
            response = tk.messagebox.askyesno("Add More Graphs", "Do you want to add more graphs?")

            if response:
                graph_num += 1
            else:
                more_graphs = False

        # Tile all windows
        op.lt_exec('win-s T')

        # Prompt user for the output Origin file name
        output_file_path = filedialog.asksaveasfilename(title="Save Output Origin File",
                                                        filetypes=[("Origin Project Files", "*.opju")])

        # Creating a notes window
        nt = op.new_notes()

        # Appending input information to the notes
        nt.append("Input Information:")
        nt.append(f"Graph 1 Template: {os.path.basename(template_file_path_1)}")
        nt.append(f"Graph 1 CSV File: {os.path.basename(csv_file_path_1)}")

        for i in range(2, graph_num + 1):
            template_var = locals().get(f"template_file_path_{i}")
            csv_var = locals().get(f"csv_file_path_{i}")

            if template_var and csv_var:
                nt.append(f"\nGraph {i} Template: {os.path.basename(template_var)}")
                nt.append(f"Graph {i} CSV File: {os.path.basename(csv_var)}")

        # Appending folder path information to the notes
        nt.append("\nFolder Paths:")
        nt.append(f"Graph 1 Folder: {os.path.dirname(template_file_path_1)}")

        for i in range(2, graph_num + 1):
            template_var = locals().get(f"template_file_path_{i}")

            if template_var:
                nt.append(f"Graph {i} Folder: {os.path.dirname(template_var)}")

        nt.append(f"Output Folder: {os.path.dirname(output_file_path)}")

        # Appending output file information to the notes
        nt.append("\nOutput Information:")
        nt.append(f"Output File: {os.path.basename(output_file_path)}")

        # Displaying the note
        nt.view = 1

        # Tile all windows
        op.lt_exec('win-s T')

        # Save the project to the specified output file path
        if op.oext:
            output_file_path = os.path.abspath(output_file_path)
            op.save(output_file_path)

    except Exception as e:
        print(f"An error occurred: {str(e)}")
        op.exit()


# End of Code snippet 3: Create Origin graphs


# Code snippet 4: Add graphs to existing Origin project
def add_graphs_to_project():
    def origin_shutdown_exception_hook(exctype, value, traceback):
        """Ensures Origin gets shut down if an uncaught exception"""
        op.exit()
        sys.__excepthook__(exctype, value, traceback)

    sys.excepthook = origin_shutdown_exception_hook

    # Only run if external Python
    if op.oext:
        op.set_show(True)

    def save_origin_project(output_path):
        try:
            op.save(output_path)
            return True  # Save operation successful
        except PermissionError:
            return False  # Save operation failed due to read-only
        except Exception as e:
            print(f"An error occurred while saving: {str(e)}")
            return False  # Save operation failed for other reasons

    def add_graphs_to_project_action():
        try:
            # Prompt user to select an existing Origin project file
            origin_file_path = filedialog.askopenfilename(title="Select Existing Origin Project File",
                                                          filetypes=[("Origin Project Files", "*.opju")])

            # Load the existing Origin project file
            op.open(origin_file_path)

            # Add more graphs
            more_graphs = True
            graph_num = 1

            while more_graphs:
                # Prompt user to select an Origin template file for the graph
                template_file_path = filedialog.askopenfilename(
                    title=f"Select Origin Template File for Graph {graph_num}",
                    filetypes=[("Origin Template Files", "*.otpu")])

                # Prompt user to select an XLSX file for the graph
                data_file_types = [("CSV Files", "*.csv"), ("XLSX Files", "*.xlsx")]
                xlsx_file_path = filedialog.askopenfilename(
                    title=f"Select XLSX File for {os.path.basename(template_file_path)}",
                    filetypes=data_file_types)

                # Load the XLSX file into a new worksheet
                wks = op.new_sheet()
                wks.from_file(xlsx_file_path, False)

                # Create a graph page with the user-selected template for the graph
                gr = op.new_graph(template=template_file_path)
                gl = gr[0]

                # Prompt user for the range of columns to plot for the graph
                column_range = simpledialog.askstring("Range of Columns",
                                                      f"Enter the range of columns (e.g., 1-12) for Graph {graph_num}:")

                start_col, end_col = map(int, column_range.split("-"))
                y_columns = list(range(start_col, end_col + 1))

                for col in y_columns:
                    gl.add_plot(wks, col, 0)

                # Group and Rescale the graph
                gl.group()
                gl.rescale()

                # Prompt user if they want to add more graphs
                response = tk.messagebox.askyesno("Add More Graphs", "Do you want to add more graphs?")

                if response:
                    graph_num += 1
                else:
                    more_graphs = False

            # Tile all windows
            op.lt_exec('win-s T')

            # Prompt user for the output Origin project file name and location
            output_file_path = filedialog.asksaveasfilename(title="Save Output Origin File As",
                                                            filetypes=[("Origin Project Files", "*.opju")])

            if not output_file_path:
                print("Save operation canceled by user.")
                op.exit()
                return  # Exit the function if the user cancels the save operation

            # Try to save the project file
            output_file_path = os.path.abspath(output_file_path)
            success = save_origin_project(output_file_path)

            # If the save operation failed due to read-only, prompt user for a new file path
            while not success:
                new_output_path = filedialog.asksaveasfilename(title="Save Output Origin File As",
                                                               filetypes=[("Origin Project Files", "*.opju")])
                if not new_output_path:
                    print("Save operation canceled by user.")
                    op.exit()
                    break  # Exit the loop if the user cancels the save operation

                new_output_path = os.path.abspath(new_output_path)
                success = save_origin_project(new_output_path)

                if success:
                    message = f"Processing completed! Output saved as:\n{new_output_path}"
                    messagebox.showinfo("Processing Complete", message)
                    op.exit()
                    break  # Exit the loop if the new save is successful
                else:
                    # Display an error message if the new save location is also read-only
                    messagebox.showerror("Error",
                                         "Selected save location is read-only. Please choose a different location.")
                    continue  # Continue the loop and prompt the user again

        except Exception as e:
            print(f"An error occurred: {str(e)}")
            op.exit()

    add_graphs_to_project_action()


root = tk.Tk()
root.withdraw()


# End of Code snippet 4: Add graphs to existing Origin project

# Code snippet 5: Function to exit the application
def exit_application():
    try:
        window.quit()  # Close the main GUI window, which ends the tkinter event loop
    except Exception as e:
        print(f"An error occurred while closing Origin: {str(e)}")


# End of Code snippet 5: Function to exit the application

# Create the main GUI window
window = tk.Tk()
window.title("FTIR Data Processing")
window.geometry("400x400")

# Create a frame for the header
header_frame = tk.Frame(window, padx=20, pady=20)
header_frame.pack()

# Create a label for the header
header_label = tk.Label(header_frame, text="FTIR Data Processing", font=("Helvetica", 16, "bold"))
header_label.pack()

# Create a frame for the content
content_frame = tk.Frame(window, padx=20, pady=20)
content_frame.pack()

# Step 1 Section
label_step1 = tk.Label(content_frame, text="Step 1: Rename the header with a CV voltage range.")
label_step1.pack(anchor="w")

rename_columns_button = tk.Button(content_frame, text="Rename Columns", command=rename_columns, bg="light blue")
rename_columns_button.pack(pady=5, anchor="w")

# Step 2 Section
label_step2 = tk.Label(content_frame, text="Step 2: Change the background spectrum with a chosen column.")
label_step2.pack(anchor="w")

process_background_data_button = tk.Button(content_frame, text="Reprocess Background", command=bg_processing,
                                           bg="light blue")
process_background_data_button.pack(pady=5, anchor="w")

# Create labels for step instructions
step3_label = tk.Label(content_frame, text="Step 3: Create Origin Project to add Graphs")
step3_label.pack(anchor="w")  # Align label to the left

# Create a button for Step 3 functionality
create_origin_graphs_button = tk.Button(content_frame, text="Create Origin Project",
                                        command=create_origin_graphs, bg="light blue")
create_origin_graphs_button.pack(pady=5, anchor="w")

# Create a label for Step 4
step4_label = tk.Label(content_frame, text="Step 4: Add Graphs to Existing Origin Project")
step4_label.pack(anchor="w")

# Create a button for Step 4 functionality
add_to_project_button = tk.Button(content_frame, text="Existing Origin Project",
                                  command=add_graphs_to_project, bg="light blue")
add_to_project_button.pack(pady=5, anchor="w")

# Exit Section
exit_label = tk.Label(content_frame, text="To quit, click below")
exit_label.pack(anchor="w")

exit_button = tk.Button(content_frame, text="Exit Application", command=exit_application, bg="red")
exit_button.pack(pady=10, anchor="w")

# Start the tkinter event loop
window.mainloop()