mg-pso-gui 0.1.40__py3-none-any.whl → 0.2.75__py3-none-any.whl

mgpsogui/gui/VisualizeTab/SideBar.py

@@ -9,6 +9,8 @@ from customtkinter import CTkOptionMenu
 from PIL import Image
 import os
 
+from . import MatrixEditor as me
+
 import pandas as pd
 
 class SideBar(CTkScrollableFrame):
@@ -37,27 +39,168 @@ class SideBar(CTkScrollableFrame):
             0, 0), pady=(0, 0), sticky="ew")
         self.containerFrame.grid_columnconfigure(0, weight=1)
         
-        selected_graph = self.home_page.graph_selector_value.get()
+        selected_graph = self.option_manager.get("selected_graph").get()
         
         if (selected_graph == "Best Cost Stacked"):
-            self.graph_label = CTkLabel(self.containerFrame, text="Best Cost Stacked")
-            self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+            #self.graph_label = CTkLabel(self.containerFrame, text="Best Cost Stacked")
+            #self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
             pass
         elif (selected_graph == "Best Cost by Round"):
-            self.graph_label = CTkLabel(self.containerFrame, text="Best Cost by Round")
-            self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+            #self.graph_label = CTkLabel(self.containerFrame, text="Best Cost by Round")
+            #self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
             pass
         elif (selected_graph == "Iteration Table"):
-            self.graph_label = CTkLabel(self.containerFrame, text="Iteration Table")
-            self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+            #self.graph_label = CTkLabel(self.containerFrame, text="Iteration Table")
+            #self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
             pass
         elif (selected_graph == "Calibrated Parameters"):
-            self.graph_label = CTkLabel(self.containerFrame, text="Calibrated Parameters")
-            self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+            #self.graph_label = CTkLabel(self.containerFrame, text="Calibrated Parameters")
+            #self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+            pass
         elif (selected_graph == "Custom CSV"):
             
-            info = self.option_manager.get_project_data()
-            folder = os.path.join(info['path'], info['name'])
+            folder = self.option_manager.get_project_folder()
+            if not os.path.exists(folder):
+                os.makedirs(folder)
+                
+            # Get all CSV files in the folder and add their paths to a list
+            path_map = {}
+            name_list = []
+            for root, dirs, files in os.walk(folder):
+                for file in files:
+                    if file.endswith(".csv"):
+                        name = file.replace(".csv", "")
+                        name_list.append(name)
+                        path_map[name] = os.path.join(root, file)
+            
+            if (len(name_list) == 0):
+                name_list.append("No files found...")
+            else:
+                if (self.option_manager.get("selected_csv").get() not in name_list):
+                    self.option_manager.get("selected_csv").set(name_list[0])
+            
+            file_label = CTkLabel(self.containerFrame, text="CSV File:")
+            file_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+            
+            self.home_page.csv_file_selector = CTkOptionMenu(self.containerFrame, values=name_list, variable=self.option_manager.get("selected_csv"), command=self.home_page.update_graph)
+            self.home_page.csv_file_selector.grid(row=1, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+            
+            selected_file = self.option_manager.get("selected_csv").get()
+            if (selected_file in path_map and selected_file != self.home_page.open_file):
+                self.home_page.csv_data = self.load_special_csv(path_map[selected_file])
+                print(self.home_page.csv_data)
+                self.home_page.open_file = selected_file
+                
+            if (self.home_page.csv_data is not None):
+                # Get all column names of CSV
+                columns = self.home_page.csv_data["data"].columns
+                
+                x_axis_label = CTkLabel(self.containerFrame, text="X Axis:")
+                x_axis_label.grid(row=2, column=0, padx=(20, 20), pady=(40, 5), sticky="w")
+                
+                self.home_page.csv_x_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_x"), command=self.home_page.update_graph)
+                self.home_page.csv_x_selector.grid(row=3, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                if (self.home_page.csv_x_selector.get() not in columns):
+                    self.home_page.csv_x_selector.set(columns[1])
+                
+                y1_axis_label = CTkLabel(self.containerFrame, text="Y Axis:")
+                y1_axis_label.grid(row=4, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.csv_y1_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_y1"), command=self.home_page.update_graph)
+                self.home_page.csv_y1_selector.grid(row=5, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                if (self.home_page.csv_y1_selector.get() not in columns):
+                    self.home_page.csv_y1_selector.set(columns[2])
+                
+                y2_axis_label = CTkLabel(self.containerFrame, text="Secondary Y Axis:")
+                y2_axis_label.grid(row=6, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.csv_y2_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_y2"), command=self.home_page.update_graph)
+                self.home_page.csv_y2_selector.grid(row=7, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                if (self.home_page.csv_y2_selector.get() not in columns):
+                    self.home_page.csv_y2_selector.set(columns[3])
+                    
+        elif (selected_graph == "Compare CSV"):
+            folder = self.option_manager.get_project_folder()
+            if not os.path.exists(folder):
+                os.makedirs(folder)
+                
+            # Get all CSV files in the folder and add their paths to a list
+            path_map = {}
+            name_list = []
+            for root, dirs, files in os.walk(folder):
+                for file in files:
+                    if file.endswith(".csv"):
+                        name = file.replace(".csv", "")
+                        name_list.append(name)
+                        path_map[name] = os.path.join(root, file)
+            
+            if (len(name_list) == 0):
+                name_list.append("No files found...")
+            else:
+                if (self.option_manager.get("selected_csv").get() not in name_list):
+                    self.option_manager.get("selected_csv").set(name_list[0])
+                if (self.option_manager.get("selected_csv2").get() not in name_list):
+                    self.option_manager.get("selected_csv2").set(name_list[0])
+            
+            file_label = CTkLabel(self.containerFrame, text="CSV Files:")
+            file_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+            
+            self.home_page.csv_file_selector = CTkOptionMenu(self.containerFrame, values=name_list, variable=self.option_manager.get("selected_csv"), command=self.home_page.update_graph)
+            self.home_page.csv_file_selector.grid(row=1, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+            
+            self.home_page.csv_file_selector2 = CTkOptionMenu(self.containerFrame, values=name_list, variable=self.option_manager.get("selected_csv2"), command=self.home_page.update_graph)
+            self.home_page.csv_file_selector2.grid(row=2, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+            
+            selected_file = self.option_manager.get("selected_csv").get()
+            if (selected_file in path_map and selected_file != self.home_page.open_file):
+                self.home_page.csv_data = self.load_special_csv(path_map[selected_file])
+                print(self.home_page.csv_data)
+                self.home_page.open_file = selected_file
+                
+            selected_file2 = self.option_manager.get("selected_csv2").get()
+            if (selected_file2 in path_map and selected_file2 != self.home_page.open_file2):
+                self.home_page.csv_data2 = self.load_special_csv(path_map[selected_file2])
+                print(self.home_page.csv_data2)
+                self.home_page.open_file2 = selected_file2
+                
+            if (self.home_page.csv_data is not None and self.home_page.csv_data2 is not None):
+                # Get all column names of CSV
+                columns = self.home_page.csv_data["data"].columns
+                columns2 = self.home_page.csv_data2["data"].columns
+                
+                x_axis_label = CTkLabel(self.containerFrame, text="X Axis:")
+                x_axis_label.grid(row=3, column=0, padx=(20, 20), pady=(40, 5), sticky="w")
+                
+                self.home_page.csv_x_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_x"), command=self.home_page.update_graph)
+                self.home_page.csv_x_selector.grid(row=4, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                if (self.home_page.csv_x_selector.get() not in columns):
+                    self.home_page.csv_x_selector.set(columns[1])
+                
+                y1_axis_label = CTkLabel(self.containerFrame, text="Y Axis:")
+                y1_axis_label.grid(row=5, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.csv_y1_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_y1"), command=self.home_page.update_graph)
+                self.home_page.csv_y1_selector.grid(row=6, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                if (self.home_page.csv_y1_selector.get() not in columns):
+                    self.home_page.csv_y1_selector.set(columns[2])
+                
+                y2_axis_label = CTkLabel(self.containerFrame, text="Secondary Y Axis:")
+                y2_axis_label.grid(row=7, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.csv_y2_selector = CTkOptionMenu(self.containerFrame, values=columns2, variable=self.option_manager.get("selected_y2"), command=self.home_page.update_graph)
+                self.home_page.csv_y2_selector.grid(row=8, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                if (self.home_page.csv_y2_selector.get() not in columns2):
+                    self.home_page.csv_y2_selector.set(columns2[2])
+                    
+        elif (selected_graph == "Sampling CSV"):
+            
+            folder = self.option_manager.get_project_folder()
             if not os.path.exists(folder):
                 os.makedirs(folder)
                 
@@ -74,35 +217,183 @@ class SideBar(CTkScrollableFrame):
             if (len(name_list) == 0):
                 name_list.append("No files found...")
             else:
-                if (self.home_page.selected_csv.get() not in name_list):
-                    self.home_page.selected_csv.set(name_list[0])
+                if (self.option_manager.get("selected_csv").get() not in name_list):
+                    self.option_manager.get("selected_csv").set(name_list[0])
+            
+            file_label = CTkLabel(self.containerFrame, text="CSV File:")
+            file_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
             
-            self.home_page.csv_file_selector = CTkOptionMenu(self.containerFrame, values=name_list, variable=self.home_page.selected_csv, command=self.home_page.update_graph)
-            self.home_page.csv_file_selector.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+            self.home_page.csv_file_selector = CTkOptionMenu(self.containerFrame, values=name_list, variable=self.option_manager.get("selected_csv"), command=self.home_page.update_graph)
+            self.home_page.csv_file_selector.grid(row=1, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
             
-            selected_file = self.home_page.selected_csv.get()
+            selected_file = self.option_manager.get("selected_csv").get()
             if (selected_file in path_map and selected_file != self.home_page.open_file):
-                self.home_page.csv_data = pd.read_csv(path_map[selected_file], skiprows=3)
+                self.home_page.csv_data = pd.read_csv(path_map[selected_file])
+                print(self.home_page.csv_data)
                 self.home_page.open_file = selected_file
                 
-            if (self.home_page.csv_data != None):
+            if (self.home_page.csv_data is not None):
                 # Get all column names of CSV
                 columns = self.home_page.csv_data.columns
                 
-                self.home_page.csv_x_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.home_page.selected_x, command=self.home_page.update_graph)
-                self.home_page.csv_x_selector.grid(row=2, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+                x_axis_label = CTkLabel(self.containerFrame, text="X Axis:")
+                x_axis_label.grid(row=2, column=0, padx=(20, 20), pady=(40, 5), sticky="w")
+                
+                self.home_page.csv_x_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_x"), command=self.home_page.update_graph)
+                self.home_page.csv_x_selector.grid(row=3, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
                 
                 if (self.home_page.csv_x_selector.get() not in columns):
                     self.home_page.csv_x_selector.set(columns[1])
                 
-                self.home_page.csv_y1_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.home_page.selected_y1, command=self.home_page.update_graph)
-                self.home_page.csv_y1_selector.grid(row=2, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+                y1_axis_label = CTkLabel(self.containerFrame, text="Y Axis:")
+                y1_axis_label.grid(row=4, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.csv_y1_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_y1"), command=self.home_page.update_graph)
+                self.home_page.csv_y1_selector.grid(row=5, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
                 
                 if (self.home_page.csv_y1_selector.get() not in columns):
                     self.home_page.csv_y1_selector.set(columns[2])
                 
-                self.home_page.csv_y2_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.home_page.selected_y2, command=self.home_page.update_graph)
-                self.home_page.csv_y2_selector.grid(row=2, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+                y2_axis_label = CTkLabel(self.containerFrame, text="Secondary Y Axis:")
+                y2_axis_label.grid(row=6, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.csv_y2_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_y2"), command=self.home_page.update_graph)
+                self.home_page.csv_y2_selector.grid(row=7, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
                 
                 if (self.home_page.csv_y2_selector.get() not in columns):
-                    self.home_page.csv_y2_selector.set(columns[3])
+                    self.home_page.csv_y2_selector.set(columns[3])
+
+                style_label = CTkLabel(self.containerFrame, text="Figure Style:")
+                style_label.grid(row=8, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.figure_style_selector = CTkOptionMenu(self.containerFrame, values=["Scatter", "Bars", "Lines", "Area", "Box"], variable=self.option_manager.get("figure_style"), command=self.home_page.update_graph)
+                self.home_page.figure_style_selector.grid(row=9, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+
+        elif (selected_graph == "Matrix Editor"):
+            
+            folder = self.option_manager.get_project_folder()
+            if not os.path.exists(folder):
+                os.makedirs(folder)
+                
+            # Get all CSV files in the folder and add their paths to a list
+            path_map = {}
+            name_list = []
+            for root, dirs, files in os.walk(folder):
+                for file in files:
+                    if file.endswith(".csv"):
+                        name = file.replace(".csv", "")
+                        name_list.append(name)
+                        path_map[name] = os.path.join(root, file)
+            
+            if (len(name_list) == 0):
+                name_list.append("No files found...")
+            else:
+                if (self.option_manager.get("selected_csv").get() not in name_list):
+                    self.option_manager.get("selected_csv").set(name_list[0])
+            
+            file_label = CTkLabel(self.containerFrame, text="CSV File:")
+            file_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+            
+            self.home_page.csv_file_selector = CTkOptionMenu(self.containerFrame, values=name_list, variable=self.option_manager.get("selected_csv"), command=self.home_page.update_graph)
+            self.home_page.csv_file_selector.grid(row=1, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+            
+            selected_file = self.option_manager.get("selected_csv").get()
+            if (selected_file in path_map and selected_file != self.home_page.open_file):
+                self.home_page.csv_data = pd.read_csv(path_map[selected_file])
+                print(self.home_page.csv_data)
+                self.home_page.open_file = selected_file
+                
+            if (self.home_page.csv_data is not None):
+                # Get all column names of CSV
+                columns = self.home_page.csv_data.columns
+                
+                style_label = CTkLabel(self.containerFrame, text="Figure Style:")
+                style_label.grid(row=2, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+                
+                self.home_page.figure_style_selector = CTkOptionMenu(self.containerFrame, values=["Scatter", "Bars", "Lines", "Area", "Box"], variable=self.option_manager.get("figure_style"), command=self.home_page.update_graph)
+                self.home_page.figure_style_selector.grid(row=3, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+
+                x_axis_label = CTkLabel(self.containerFrame, text="Y Axis:")
+                x_axis_label.grid(row=4, column=0, padx=(20, 20), pady=(40, 5), sticky="w")
+                
+                self.home_page.csv_x_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_x"), command=self.home_page.update_graph)
+                self.home_page.csv_x_selector.grid(row=5, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                if (self.home_page.csv_x_selector.get() not in columns):
+                    self.home_page.csv_x_selector.set(columns[1])
+                
+                self.matrix_editor = me.MatrixEditor(self.containerFrame, width=280, option_manager=self.option_manager, home_page=self.home_page, columns=columns)
+                self.matrix_editor.grid(row=6, column=0, padx=(10, 10), pady=(10, 0), sticky="nsew")
+                self.matrix_editor.grid_columnconfigure(0, weight=1)
+                self.matrix_editor.grid_rowconfigure(0, weight=1)
+
+
+                #y1_axis_label = CTkLabel(self.containerFrame, text="Y Axis:")
+                #y1_axis_label.grid(row=6, column=0, padx=(20, 20), pady=(20, 5), sticky="w")
+
+                #self.home_page.csv_y1_selector = CTkOptionMenu(self.containerFrame, values=columns, variable=self.option_manager.get("selected_y1"), command=self.home_page.update_graph)
+                #self.home_page.csv_y1_selector.grid(row=7, column=0, padx=(20, 20), pady=(5, 5), sticky="ew")
+                
+                #if (self.home_page.csv_y1_selector.get() not in columns):
+                #    self.home_page.csv_y1_selector.set(columns[2])
+
+    def load_special_csv(self, file_path):
+        file_metadata = {}
+        data_metadata = {}
+        mode = "file_metadata"
+        columns = []
+        data_lines = []
+        
+        with open(file_path, "r") as file:
+            lines = file.readlines()
+            
+            if (not lines[0].startswith("@")):
+                return {
+                    "file_attributes": {},
+                    "data_attributes": {},
+                    "data": pd.read_csv(file_path)
+                }
+            
+            for line in lines:
+                if (line.startswith("@H,")):
+                    mode = "data_metadata"
+                    columns = line.strip().rstrip(",").split(",")[1:]
+                    continue
+                elif (line.startswith("@T,")):
+                    mode = "file_metadata"
+                    continue
+                elif (line.startswith(",") and mode == "data_metadata"):
+                    mode = "file_data"
+                    
+                if (mode == "file_metadata"):
+                    try:
+                        key, value = line.strip().rstrip(",").split(",")
+                        file_metadata[key] = value
+                    except:
+                        pass
+                    
+                elif (mode == "data_metadata"):
+                    try:
+                        values = line.strip().rstrip(",").split(",")
+                        key = values[0]
+                        values = values[1:]
+                        if len(values) == len(columns):
+                            data_metadata[key] = {}
+                            for i in range(len(columns)):
+                                data_metadata[key][columns[i]] = values[i]
+                    except:
+                        pass
+                    
+                elif (mode == "file_data"):
+                    try:
+                        values = line.strip().rstrip(",").split(",")[1:]
+                        if len(values) == len(columns):
+                            data_lines.append(values)
+                    except:
+                        pass
+                        
+        return {
+            "file_attributes": file_metadata,
+            "data_attributes": data_metadata,
+            "data": pd.DataFrame(data_lines, columns=columns)
+        }

mgpsogui/gui/VisualizeTab/VisualizeTab.py

@@ -4,6 +4,7 @@ from PIL import Image, ImageTk
 import os
 import platform
 import subprocess
+import shutil
 
 from . import SideBar
 
@@ -11,8 +12,7 @@ def create_tab(self, tab):
     
     def open_graph_in_browser():
         # Open the file in the default program
-        info = self.option_manager.get_project_data()
-        folder = os.path.join(info['path'], info['name'])
+        folder = self.option_manager.get_project_folder()
         
         if not os.path.exists(folder):
             os.makedirs(folder)
@@ -25,7 +25,62 @@ def create_tab(self, tab):
             subprocess.Popen(["open", file_path])
         else:
             subprocess.Popen(["xdg-open", file_path])
-    
+
+    def export_graph():
+        
+        folder = self.option_manager.get_project_folder()
+        
+        if not os.path.exists(folder):
+            os.makedirs(folder)
+
+        html_file = os.path.join(folder, self.selected_graph_name + ".html")
+        json_file = os.path.join(folder, self.selected_graph_name + ".json")
+        png_file = os.path.join(folder, self.selected_graph_name + ".png")
+        pdf_file = os.path.join(folder, self.selected_graph_name + ".pdf")
+
+        # Make a directory call package
+        if not os.path.exists(os.path.join(folder, self.selected_graph_name + "_package")):
+            os.makedirs(os.path.join(folder, self.selected_graph_name + "_package"))
+
+        # Check if html_file exists and copy it to the package directory
+        if os.path.exists(html_file):
+            with open(html_file, 'r') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".html"), 'w') as new_file:
+                    new_file.write(data)
+
+        # Check if json_file exists and copy it to the package directory
+        if os.path.exists(json_file):
+            with open(json_file, 'r') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".json"), 'w') as new_file:
+                    new_file.write(data)
+        
+        # Check if png_file exists and copy it to the package directory
+        if os.path.exists(png_file):
+            with open(png_file, 'rb') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".png"), 'wb') as new_file:
+                    new_file.write(data)
+
+        # Check if pdf_file exists and copy it to the package directory
+        if os.path.exists(pdf_file):
+            with open(pdf_file, 'rb') as file:
+                data = file.read()
+                with open(os.path.join(folder, self.selected_graph_name + "_package", self.selected_graph_name + ".pdf"), 'wb') as new_file:
+                    new_file.write(data)
+        
+        # Zip the package directory
+        shutil.make_archive(os.path.join(folder, self.selected_graph_name + "_package"), 'zip', os.path.join(folder, self.selected_graph_name + "_package"))
+
+        # Open the directory containing the package
+        if platform.system() == "Windows":
+            os.startfile(os.path.join(folder, self.selected_graph_name + "_package"))
+        elif platform.system() == "Darwin":
+            subprocess.Popen(["open", os.path.join(folder, self.selected_graph_name + "_package")])
+        else:
+            subprocess.Popen(["xdg-open", os.path.join(folder, self.selected_graph_name + "_package")])
+
     def _resize_image(event):
         self.graph_label.update_idletasks()
         new_width = self.graph_label.winfo_width()
@@ -61,20 +116,26 @@ def create_tab(self, tab):
     self.graph_container.grid_columnconfigure(0, weight=1)
     self.graph_container.grid_rowconfigure(0, weight=1)
     
-    self.graph_selector = customtkinter.CTkOptionMenu(self.graph_sidebar, values=["Best Cost Stacked", "Best Cost by Round", "Calibrated Parameters", "Iteration Table", "Custom CSV"], variable=self.graph_selector_value, command=self.update_graph)
+    graph_types = []
+    self.graph_selector = customtkinter.CTkOptionMenu(self.graph_sidebar, values=graph_types, variable=self.option_manager.get("selected_graph"), command=self.update_graph)
     self.graph_selector.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
     
     # Create SideBar
     self.vis_sidebar = SideBar.SideBar(self.graph_sidebar, option_manager=self.option_manager, home_page=self, fg_color="transparent")
-    self.vis_sidebar.grid(row=1, column=0, rowspan=8, padx=(0, 0), pady=(0, 0), sticky="nsew")
+    self.vis_sidebar.grid(row=1, column=0, rowspan=6, padx=(0, 0), pady=(0, 0), sticky="nsew")
     
+
+    self.graph_theme = customtkinter.CTkOptionMenu(self.graph_sidebar, values=["Dark", "Light", "Publication"], variable=self.option_manager.get("graph_theme"), command=self.update_graph)
+    self.graph_theme.grid(row=7, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+
+    self.graph_export = customtkinter.CTkButton(self.graph_sidebar, text="Export", command=export_graph)
+    self.graph_export.grid(row=8, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
+
     # Add a button to call open_graph_in_browser
-    self.graph_button = customtkinter.CTkButton(self.graph_sidebar, text="Open in Browser", command=open_graph_in_browser)
+    self.graph_button = customtkinter.CTkButton(self.graph_sidebar, text="Preview", command=open_graph_in_browser)
     self.graph_button.grid(row=9, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
     
     self.graph_image_obj = Image.open(os.path.join("./images", "refresh_hd.png"))
     self.graph_image = customtkinter.CTkImage(self.graph_image_obj, size=(1280, 720))
     self.graph_label = customtkinter.CTkLabel(self.graph_container, text=None, image=self.graph_image)
     self.graph_label.grid(row=0, column=0, padx=(20, 20), pady=(20, 20), sticky="nsew")
-    #window = customtkinter.CTk()
-    #self.graph_label.bind('<Configure>', _resize_image)