Update app.py

app.py

@@ -1,9 +1,15 @@
 import gradio as gr
-from datetime import date, timedelta
+from datetime import datetime, 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
 
+def validate_date(date_string):
+    try:
+        return datetime.strptime(date_string, "%Y-%m-%d").date()
+    except ValueError:
+        return None
+
 demo = gr.Blocks()
 
 with demo:
@@ -13,9 +19,9 @@ with demo:
     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)
+    # New date inputs using Textbox
+    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()
@@ -27,13 +33,27 @@ with demo:
         stocks = get_stocks_from_index(index)
         return gr.Dropdown(choices=stocks)
 
+    def process_inputs(*args):
+        idx, stock, interval, graph_type, forecast_method, start_date, end_date = args
+        
+        start = validate_date(start_date)
+        end = validate_date(end_date)
+        
+        if start is None or end is None:
+            return gr.Textbox(value="Invalid date format. Please use YYYY-MM-DD."), None
+        
+        if start > end:
+            return gr.Textbox(value="Start date must be before end date."), None
+        
+        return get_stock_graph_and_info(idx, stock, interval, graph_type, forecast_method, start, end)
+
     d1.change(update_stock_options, d1, d2)
-    d2.change(get_stock_graph_and_info, inputs, outputs)
-    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)
+    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()