Update app.py

app.py

@@ -184,6 +184,9 @@ def plot_candlestick_chart(data, ticker):
     return fig, current_price, None, None
 
 def format_price_info(current_price, log_price, percent_diff):
+    if current_price is None:
+        return "Unable to retrieve price information."
+    
     if log_price is None or percent_diff is None:
         return f"Current Price: ${current_price:.2f}"
     
@@ -219,7 +222,7 @@ custom_css = """
 }
 """
 
-# Create Gradio interface with custom Seafoam theme
+# Update the Gradio interface section
 with gr.Blocks(theme=seafoam, title="Stock Charts", css=custom_css) as iface:
     gr.Markdown("# Stock Charts")
     gr.Markdown("Enter a stock ticker and date range to generate a chart.")
@@ -239,14 +242,15 @@ with gr.Blocks(theme=seafoam, title="Stock Charts", css=custom_css) as iface:
     
     price_info = gr.HTML(label="Price Information")
 
+    def update_chart_and_info(ticker, start_date, end_date, chart_type):
+        chart_data, current_price, log_price, percent_diff = plot_chart(ticker, start_date, end_date, chart_type)
+        price_info_html = format_price_info(current_price, log_price, percent_diff)
+        return chart_data, price_info_html
+
     submit_button.click(
-        plot_chart,
+        update_chart_and_info,
         inputs=[ticker, start_date, end_date, chart_type],
         outputs=[chart, price_info]
-    ).then(
-        lambda current_price, log_price, percent_diff: format_price_info(current_price, log_price, percent_diff),
-        inputs=[gr.State(None), gr.State(None), gr.State(None)],
-        outputs=[price_info]
     )
 
 # Launch the app