Update app.py

app.py

@@ -1,14 +1,6 @@
 import gradio as gr
-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
+from config import index_options, time_intervals, START_DATE, END_DATE
+from app_utils import update_stock_options, process_inputs
 
 demo = gr.Blocks()
 
@@ -19,7 +11,6 @@ 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 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"))
 
@@ -29,24 +20,6 @@ with demo:
     inputs = [d1, d2, d3, d4, d5, date_start, date_end]
     outputs = [out_graph, out_fundamentals]
 
-    def update_stock_options(index):
-        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(process_inputs, inputs, outputs)
     d3.change(process_inputs, inputs, outputs)