Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from datetime import date, timedelta
 from config import index_options, time_intervals, START_DATE, END_DATE, FORECAST_PERIOD
 from data_fetcher import get_stocks_from_index
 from stock_analysis import get_stock_graph_and_info
@@ -11,11 +12,15 @@ with demo:
     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)
+    
+    # New date inputs
+    date_start = gr.Date(label="Start Date", value=START_DATE)
+    date_end = gr.Date(label="End Date", value=END_DATE)
 
     out_graph = gr.Plot()
     out_fundamentals = gr.DataFrame()
 
-    inputs = [d1, d2, d3, d4, d5]
+    inputs = [d1, d2, d3, d4, d5, date_start, date_end]
     outputs = [out_graph, out_fundamentals]
 
     def update_stock_options(index):
@@ -27,6 +32,8 @@ with demo:
     d3.change(get_stock_graph_and_info, inputs, outputs)
     d4.change(get_stock_graph_and_info, inputs, outputs)
     d5.change(get_stock_graph_and_info, inputs, outputs)
+    date_start.change(get_stock_graph_and_info, inputs, outputs)
+    date_end.change(get_stock_graph_and_info, inputs, outputs)
 
 if __name__ == "__main__":
     demo.launch()