rsi sell/buy points && plot price fix

algorithms/price.py

@@ -2,7 +2,7 @@
 import plots.colors as colors
 
 class Algorithm:
-
-	def plot(prices, timestamps):
+	def __init__(self) -> None:
+		pass
+	def plot(self, prices, timestamps):
 		plt.plot(timestamps, prices, color=colors.primary())
-		return plt

algorithms/rsi.py

@@ -2,6 +2,7 @@
 import numpy as np
 import matplotlib.pyplot as plt
 import plots.colors as colors
+from matplotlib.gridspec import GridSpec
 
 class Algorithm:
 
@@ -26,14 +27,24 @@ def signal(self, _, data):
 		return 'no_action', 0
 
 	def plot(self, prices, timestamps, **kwargs):
+		gs = GridSpec(3, 1, figure=plt.gcf())
+
+		plt.subplot(gs[0, :])
 		rsi_line = self.algorithm(prices, **kwargs)
 
 		plt.plot(timestamps, rsi_line, color=colors.primary())
 
 		# Thresholds
 		upper = np.full(rsi_line.shape, self.high)
 		lower = np.full(rsi_line.shape, self.low)
+		upper_condition = rsi_line >= self.high
+		lower_condition = rsi_line <= self.low
 
 		plt.fill_between(timestamps, upper, lower, color='grey', alpha=0.3)
 		plt.plot(timestamps, upper, linestyle='dashed', color=colors.upper())
 		plt.plot(timestamps, lower, linestyle='dashed', color=colors.lower())
+
+		plt.subplot(gs[1, :])
+		plt.plot(timestamps, prices, color=colors.primary())
+		plt.scatter(timestamps[upper_condition], prices[upper_condition], color=colors.upper())
+		plt.scatter(timestamps[lower_condition], prices[lower_condition], color=colors.lower())