Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from datetime import datetime, date, timedelta
+import plotly.graph_objects as go
+import pandas as pd
 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
@@ -10,6 +12,12 @@ def validate_date(date_string):
     except ValueError:
         return None
 
+def create_error_plot(error_message):
+    fig = go.Figure()
+    fig.add_annotation(x=0.5, y=0.5, text=error_message, showarrow=False, font_size=20)
+    fig.update_layout(title="Error", xaxis_title="", yaxis_title="")
+    return fig
+
 demo = gr.Blocks()
 
 with demo:
@@ -19,7 +27,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"))
 
@@ -40,17 +47,21 @@ with demo:
         end = validate_date(end_date)
         
         if start is None or end is None:
-            return "Invalid date format. Please use YYYY-MM-DD.", None
+            return create_error_plot("Invalid date format. Please use YYYY-MM-DD."), None
         
         if start > end:
-            return "Start date must be before end date.", None
+            return create_error_plot("Start date must be before end date."), None
         
         try:
             fig, fundamentals = get_stock_graph_and_info(idx, stock, interval, graph_type, forecast_method, start, end)
+            if not isinstance(fig, go.Figure):
+                raise ValueError("Expected a plotly Figure object")
+            if not isinstance(fundamentals, pd.DataFrame):
+                raise ValueError("Expected a pandas DataFrame for fundamentals")
             return fig, fundamentals
         except Exception as e:
             error_message = f"An error occurred: {str(e)}"
-            return error_message, None
+            return create_error_plot(error_message), pd.DataFrame()
 
     d1.change(update_stock_options, d1, d2)
     d2.change(process_inputs, inputs, outputs)