import originpro as op
import os

template_file_path_1=op.path('e')+'line.otpu'

nt1=op.new_notes()
nt1.append("Graph 1 Information:")
nt1.append(f"Template: {os.path.basename(template_file_path_1)}")
nt1.append(f"Template Path: {os.path.dirname(template_file_path_1)}")

import originpro as op
import tkinter as tk
from tkinter import filedialog
import sys
import os


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)

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

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

    # Load the CSV file into Origin worksheet
    wks = op.new_sheet()
    wks.from_file(csv_file_path, 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()

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

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

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

    # Create a graph page with the user-selected template for graph 2
    gr2 = op.new_graph(template=template_file_path_2)
    gl2 = gr2[0]

    # Prompt user for the range of columns to plot for graph 2
    column_range = input("Enter the range of columns (e.g., 1-12) for graph 2: ")
    start_col, end_col = map(int, column_range.split("-"))
    y_columns = list(range(start_col, end_col + 1))

    for col in y_columns:
        plot = gl2.add_plot(wks2, col, 0)

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

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

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

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

    # Create a graph page with the user-selected template for graph 3
    gr3 = op.new_graph(template=template_file_path_3)
    gl3 = gr3[0]

    # Prompt user for the range of columns to plot for graph 3
    column_range = input("Enter the range of columns (e.g., 1-12) for graph 3: ")
    start_col, end_col = map(int, column_range.split("-"))
    y_columns = list(range(start_col, end_col + 1))

    for col in y_columns:
        plot = gl3.add_plot(wks3, col, 0)

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

    # 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")])

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

    # 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)}")
    nt.append(f"Graph 2 Template: {os.path.basename(template_file_path_2)}")
    nt.append(f"Graph 2 XLSX File: {os.path.basename(xlsx_file_path)}")
    nt.append(f"Graph 3 Template: {os.path.basename(template_file_path_3)}")
    nt.append(f"Graph 3 XLSX File: {os.path.basename(xlsx_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()

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

nt.view()

nt.view=1