File size: 4,074 Bytes
9a42782 46ad9ed 9a42782 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
import gradio as gr
from indicators import SMC
from utils import smc_plot_backtest, smc_ema_plot_backtest, smc_structure_plot_backtest, fetch
import pandas as pd
symbols = pd.read_csv('data/Ticker_List_NSE_India.csv')
limits = pd.read_csv('data/yahoo_limits.csv')
def run(stock, interval, period, strategy, swing_hl, ema1=9, ema2=21, cross_close=False):
# Downloading ticker data.
ticker = symbols[symbols['NAME OF COMPANY'] == stock]['YahooEquiv'].values[0]
data = fetch(ticker, period, interval)
# Plotting signal plot based on strategy.
if strategy == "Order Block" or strategy == "Order Block with EMA":
signal_plot = (SMC(data=data, swing_hl_window_sz=swing_hl).
plot(order_blocks=True, swing_hl=True, show=False).
update_layout(title=dict(text=ticker)))
else:
signal_plot = (SMC(data=data, swing_hl_window_sz=swing_hl).
plot(swing_hl_v2=True, structure=True, show=False).
update_layout(title=dict(text=ticker)))
backtest_plot = gr.Plot()
# Plotting backtest plot based on strategy.
if strategy == "Order Block":
backtest_plot = smc_plot_backtest(data, 'test.html', swing_hl)
if strategy == "Order Block with EMA":
backtest_plot = smc_ema_plot_backtest(data, 'test.html', ema1, ema2, cross_close)
if strategy == "Structure trading":
backtest_plot = smc_structure_plot_backtest(data, 'test.html', swing_hl)
return signal_plot, backtest_plot
with gr.Blocks(fill_width=True) as app:
gr.Markdown(
'# Algorithmic Trading Dashboard'
)
stock = gr.Dropdown(symbols['NAME OF COMPANY'].unique().tolist(), label='Select Company', value=None)
with gr.Row():
interval = gr.Dropdown(limits['interval'].tolist(), label='Select Interval', value=None)
period_list = ['1d', '5d', '1mo', '3mo', '6mo', '1y', '2y', '5y', '10y', 'ytd', 'max']
period = gr.Dropdown(label = 'Select Period', choices=["max"], value="max")
# Updating period based on interval
def update_period(interval):
limit = limits[limits['interval'] == interval]['limit'].values[0]
idx = period_list.index(limit)
return gr.Dropdown(period_list[:idx+1]+['max'], interactive=True, label='Select Period')
interval.change(update_period, [interval], [period])
with gr.Row():
strategy = gr.Dropdown(['Order Block', 'Order Block with EMA', 'Structure trading'], label='Strategy', value=None)
swing_hl = gr.Number(label="Swing High/Low Window Size", value=10, interactive=True)
@gr.render(inputs=[strategy])
def show_extra(strat):
if strat == "Order Block with EMA":
with gr.Row():
ema1 = gr.Number(label='Fast EMA length', value=9)
ema2 = gr.Number(label='Slow EMA length', value=21)
cross_close = gr.Checkbox(label='Close trade on EMA crossover')
input = [stock, interval, period, strategy, swing_hl, ema1, ema2, cross_close]
elif strat == "Order Block" or strat == "Structure trading":
input = [stock, interval, period, strategy, swing_hl]
else:
input = []
btn.click(
run,
inputs=input,
outputs=[signal_plot, backtest_plot]
)
examples = gr.Examples(
examples=[
["Reliance Industries Limited", "15m", "max", "Order Block", 10],
["Reliance Industries Limited", "15m", "max", "Order Block with EMA", 10],
["Reliance Industries Limited", "15m", "max", "Structure trading", 20],
],
example_labels=['Order Block', 'Order Block with EMA', 'Structure trading'],
inputs=[stock, interval, period, strategy, swing_hl]
)
btn = gr.Button("Run")
with gr.Row():
signal_plot = gr.Plot(label='Signal plot')
with gr.Row():
backtest_plot = gr.Plot(label='Backtesting plot')
app.launch(debug=True) |