ManualPlotAnalysis.py

import tkinter as tk
from tkinter import messagebox
from BaseAnalysis import BaseAnalysis

class ManualPlotAnalysis(BaseAnalysis):
	def __init__(self, main_app):
		super().__init__()
		self.main_app = main_app
		self.plot_title = tk.StringVar(value = "Manual Plot")

		self.code_entry_text = (
			"# Write your code here. \n"
			"# Ensure that the resulting Figure object is stored in a variable named 'fig'.\n"
			"# For example:\n"
			"#import seaborn as sns\n"
			"#sns.set(style='darkgrid')\n"
			"#g=sns.jointplot(x=df['Age'],y=df['BMI'],data=df,kind='kde')\n"
			"#fig = g.fig\n"
			"#\n"
			"#You can also manipluate the data manually, for example:\n"
			"#import pandas as pd\n"
			"#date = pd.to_datetime(df['Date'], format='%d.%m.%Y')\n"
			"#df['Day_of_Week'] = date.dt.day_name()\n"
		) 
 
	def init_ui(self):
		# Remove previously created widgets in the parent_frame
		for widget in self.main_app.manual_plot_tab_frame.winfo_children():
			widget.destroy()
   
		# Frame for Buttons and Code Entry
		button_frame = tk.Frame(self.main_app.manual_plot_tab_frame)
		button_frame.pack(padx=5, pady=5, anchor='w')
		show_plot_button = tk.Button(button_frame, text="Show Plot", command=lambda: self.execute_and_show_plot(False))
		show_plot_button.grid(row=0, column=0, padx=5, sticky='w')
		show_plot_button = tk.Button(button_frame, text="Refresh Plot", command=lambda: self.execute_and_show_plot(True))
		show_plot_button.grid(row=0, column=0, padx=110, sticky='w')

		self.code_entry = tk.Text(button_frame, height=20)
		self.code_entry.grid(row=1, column=0, padx=5, pady=5, sticky='w')
		self.code_entry.insert(tk.END, self.code_entry_text)

	def execute_and_show_plot(self, refresh_plot):
		user_env = {"df": self.main_app.df}
		user_code = self.code_entry.get("1.0", tk.END)
		exec(user_code, user_env)
  
		if "fig" in user_env:
			fig = user_env["fig"]
			if refresh_plot:
				self.main_app.open_windows[-1].refresh_plot(fig)
			elif self.main_app.use_plotly.get():
				self.main_app.open_windows.append(self.display_plotly_plot(fig))
			else:
				self.main_app.open_windows.append(self.display_plot(fig))

		else:
			messagebox.showinfo("Error", "The code did not produce a 'fig' variable.")
			print("Error: The code did not produce a 'fig' variable.")  

	def save_code_text(self):
		self.code_entry_text = self.code_entry.get("1.0", tk.END)