File size: 1,493 Bytes
31ac823
2a73dd7
 
67781cf
31ac823
2e2dfb4
31ac823
 
8b5f94a
31ac823
 
 
7057428
67781cf
 
2e2dfb4
5b5151c
 
2e79a3d
7057428
5b5151c
2e79a3d
8b5f94a
67781cf
 
 
 
 
 
2e2dfb4
5b5151c
 
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
import gradio as gr
from config import index_options, time_intervals, START_DATE, END_DATE
from app_utils import update_stock_options, process_inputs

demo = gr.Blocks()

with demo:
    d1 = gr.Dropdown(index_options, label='Please select Index...', info='Will be adding more indices later on', interactive=True)
    d2 = gr.Dropdown(label='Please Select Stock from your selected index', interactive=True)
    d3 = gr.Dropdown(time_intervals, label='Select Time Interval', value='1d', interactive=True)
    d4 = gr.Radio(['Line Graph', 'Candlestick Graph'], label='Select Graph Type', value='Line Graph', interactive=True)
    d5 = gr.Dropdown(['ARIMA', 'Prophet', 'LSTM'], label='Select Forecasting Method', value='ARIMA', interactive=True)
    
    date_start = gr.Textbox(label="Start Date (YYYY-MM-DD)", value=START_DATE.strftime("%Y-%m-%d"))
    date_end = gr.Textbox(label="End Date (YYYY-MM-DD)", value=END_DATE.strftime("%Y-%m-%d"))

    out_graph = gr.Plot()
    out_fundamentals = gr.DataFrame()

    inputs = [d1, d2, d3, d4, d5, date_start, date_end]
    outputs = [out_graph, out_fundamentals]

    d1.change(update_stock_options, d1, d2)
    d2.change(process_inputs, inputs, outputs)
    d3.change(process_inputs, inputs, outputs)
    d4.change(process_inputs, inputs, outputs)
    d5.change(process_inputs, inputs, outputs)
    date_start.change(process_inputs, inputs, outputs)
    date_end.change(process_inputs, inputs, outputs)

if __name__ == "__main__":
    demo.launch()