dibend commited on
Commit
0852535
1 Parent(s): 25e5bc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -11
app.py CHANGED
@@ -48,21 +48,25 @@ def plot_data(data, buy_signals_rsi, sell_signals_rsi, buy_signals_macd, sell_si
48
  # Add the candlestick chart
49
  fig.add_trace(go.Candlestick(x=data.index, open=data['Open'], high=data['High'], low=data['Low'], close=data['Close'], name='Market Data'), row=1, col=1)
50
 
51
- # Plot RSI signals with labels
52
- for date, price in buy_signals_rsi:
53
- fig.add_trace(go.Scatter(x=[date], y=[price], mode='markers', marker_symbol='triangle-up', marker_color='green', name='Buy Signal RSI', hovertext="RSI Buy", hoverinfo='text'), row=1, col=1)
54
- for date, price in sell_signals_rsi:
55
- fig.add_trace(go.Scatter(x=[date], y=[price], mode='markers', marker_symbol='triangle-down', marker_color='red', name='Sell Signal RSI', hovertext="RSI Sell", hoverinfo='text'), row=1, col=1)
56
 
57
- # Plot MACD signals with labels
58
- for date, price in buy_signals_macd:
59
- fig.add_trace(go.Scatter(x=[date], y=[price], mode='markers', marker=dict(symbol='triangle-up', color='blue'), name='Buy Signal MACD', hovertext="MACD Buy", hoverinfo='text'), row=1, col=1)
60
- for date, price in sell_signals_macd:
61
- fig.add_trace(go.Scatter(x=[date], y=[price], mode='markers', marker=dict(symbol='triangle-down', color='purple'), name='Sell Signal MACD', hovertext="MACD Sell", hoverinfo='text'), row=1, col=1)
 
 
 
 
 
62
 
63
- fig.update_layout(title='Stock Analysis with RSI and MACD Signals')
64
  return fig
65
 
 
66
  def main_interface(ticker, period):
67
  data = fetch_data(ticker, period)
68
  rsi = compute_rsi(data)
 
48
  # Add the candlestick chart
49
  fig.add_trace(go.Candlestick(x=data.index, open=data['Open'], high=data['High'], low=data['Low'], close=data['Close'], name='Market Data'), row=1, col=1)
50
 
51
+ # Combine all buy signals into one list and create a custom hover text list
52
+ buy_dates = [date for date, _ in buy_signals_rsi + buy_signals_macd]
53
+ buy_prices = [price for _, price in buy_signals_rsi + buy_signals_macd]
54
+ buy_text = ["RSI Buy" for _ in buy_signals_rsi] + ["MACD Buy" for _ in buy_signals_macd]
 
55
 
56
+ fig.add_trace(go.Scatter(x=buy_dates, y=buy_prices, mode='markers', marker_symbol='triangle-up', marker_color='green', name='Buy Signals', text=buy_text, hoverinfo='text'))
57
+
58
+ # Combine all sell signals into one list and create a custom hover text list
59
+ sell_dates = [date for date, _ in sell_signals_rsi + sell_signals_macd]
60
+ sell_prices = [price for _, price in sell_signals_rsi + sell_signals_macd]
61
+ sell_text = ["RSI Sell" for _ in sell_signals_rsi] + ["MACD Sell" for _ in sell_signals_macd]
62
+
63
+ fig.add_trace(go.Scatter(x=sell_dates, y=sell_prices, mode='markers', marker_symbol='triangle-down', marker_color='red', name='Sell Signals', text=sell_text, hoverinfo='text'))
64
+
65
+ fig.update_layout(title='Stock Analysis with RSI and MACD Signals', xaxis_title='Date', yaxis_title='Price')
66
 
 
67
  return fig
68
 
69
+
70
  def main_interface(ticker, period):
71
  data = fetch_data(ticker, period)
72
  rsi = compute_rsi(data)