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

mgpsogui/gui/VisualizeTab/SideBar.py

@@ -0,0 +1,399 @@
+from customtkinter import CTkScrollableFrame
+from customtkinter import CTkFrame
+from customtkinter import CTkLabel
+from customtkinter import CTkButton
+from customtkinter import CTkEntry
+from customtkinter import CTkTextbox
+from customtkinter import CTkImage
+from customtkinter import CTkOptionMenu
+from PIL import Image
+import os
+
+from . import MatrixEditor as me
+
+import pandas as pd
+
+class SideBar(CTkScrollableFrame):
+    def __init__(self, *args,
+                 option_manager: None,
+                 home_page: None,
+                 **kwargs):
+        super().__init__(*args, **kwargs)
+
+        self.option_manager = option_manager
+        self.home_page = home_page
+        
+        self.render()
+
+    def clear(self):
+        self.containerFrame.destroy()
+
+    def refresh(self):
+        self.clear()
+        self.render()
+
+    def render(self):
+
+        self.containerFrame = CTkFrame(self, width=300, fg_color="transparent")
+        self.containerFrame.grid(row=0, column=0, padx=(
+            0, 0), pady=(0, 0), sticky="ew")
+        self.containerFrame.grid_columnconfigure(0, weight=1)
+        
+        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")
+            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")
+            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")
+            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")
+            pass
+        elif (selected_graph == "Custom 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])
+            
+            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)
+                
+            # 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
+                
+                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])
+
+                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,13 +4,15 @@ from PIL import Image, ImageTk
 import os
 import platform
 import subprocess
+import shutil
+
+from . import SideBar
 
 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)
@@ -23,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()
@@ -45,7 +102,7 @@ def create_tab(self, tab):
         self.graph_label.update_idletasks()
     
     
-    tab.grid_columnconfigure(0, weight=1)
+    tab.grid_columnconfigure(0, weight=2)
     tab.grid_columnconfigure(1, weight=8)
     tab.grid_rowconfigure(0, weight=1)
     
@@ -55,22 +112,30 @@ def create_tab(self, tab):
     self.graph_container.grid(row=0, column=1, padx=(20, 20), pady=(20, 20), sticky="nsew")
     
     self.graph_sidebar.grid_columnconfigure(0, weight=1)
-    self.graph_sidebar.grid_rowconfigure(8, weight=1)
+    self.graph_sidebar.grid_rowconfigure(1, weight=1)
     self.graph_container.grid_columnconfigure(0, weight=1)
     self.graph_container.grid_rowconfigure(0, weight=1)
     
-    self.graph_selector_value = tk.StringVar()
-    self.graph_selector_value.set("Best Cost Stacked")
-    self.graph_selector = customtkinter.CTkOptionMenu(self.graph_sidebar, values=["Best Cost Stacked", "Best Cost by Round", "Calibrated Parameters", "Iteration Table"], 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=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)