<commit_before>#!/usr/bin/env python2

from functools import partial

import wx

from boltzmannizer.gui.plot import PlotFrame2DByTemperature, PlotFrame3DPopulation
from boltzmannizer.gui.utils import DataPanel
from boltzmannizer.science.boltzmann_distribution import BoltzmannDistribution
from boltzmannizer.tools.misc import Reserver


class BoltzmannDistributionGUI(BoltzmannDistribution):
	"""
	BoltzmannDistribution with some GUI-specific features.
	"""

	def __init__(self, color, *args, **kwargs):
		BoltzmannDistribution.__init__(self, *args, **kwargs)

		self.color = color

	@classmethod
	def from_file(cls, color, *args, **kwargs):
		bd = BoltzmannDistribution.from_file(*args, **kwargs)

		bd.color = color

		return bd


class MainFrame(wx.Frame):
	COLORS = ['blue', 'green', 'red', 'magenta', 'black', 'cyan', 'brown']
	# Yellow is awful, but we've run out of colors at this point!
	OVERFLOW_COLOR = 'yellow'

	def __init__(self):
		wx.Frame.__init__(self, None, title='Boltzmannizer', size=(600, 400))
<commit_msg>Add more colors<commit_after>#!/usr/bin/env python2

from functools import partial

import wx

from boltzmannizer.gui.plot import PlotFrame2DByTemperature, PlotFrame3DPopulation
from boltzmannizer.gui.utils import DataPanel
from boltzmannizer.science.boltzmann_distribution import BoltzmannDistribution
from boltzmannizer.tools.misc import Reserver


class BoltzmannDistributionGUI(BoltzmannDistribution):
	"""
	BoltzmannDistribution with some GUI-specific features.
	"""

	def __init__(self, color, *args, **kwargs):
		BoltzmannDistribution.__init__(self, *args, **kwargs)

		self.color = color

	@classmethod
	def from_file(cls, color, *args, **kwargs):
		bd = BoltzmannDistribution.from_file(*args, **kwargs)

		bd.color = color

		return bd


class MainFrame(wx.Frame):
	COLORS = ['blue', 'green', 'red', 'magenta', 'black', 'cyan', 'brown', 'orange', 'grey']
	# Yellow is awful, but we've run out of colors at this point!
	OVERFLOW_COLOR = 'yellow'

	def __init__(self):
		wx.Frame.__init__(self, None, title='Boltzmannizer', size=(600, 400))

{'fraction_non_alphanumeric': 0.11657032755298652, 'fraction_numerical': 0.008670520231213872, 'mean_word_length': 5.854785478547854, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '2609079', 'n_tokens_mistral': 745, 'n_tokens_neox': 640, 'n_words': 176}