Update app.py

app.py

@@ -71,6 +71,9 @@ def plot_interactive_logarithmic_stock_chart(ticker, start_date, end_date):
         if data.empty:
             return "No data available for the specified date range."
 
+        log_returns = np.log(data['Close'] / data['Close'].shift(1))
+        log_returns_std = log_returns.std()
+
         x = (data.index - data.index[0]).days
         y = np.log(data['Close'])
         slope, intercept = np.polyfit(x, y, 1)
@@ -79,10 +82,10 @@ def plot_interactive_logarithmic_stock_chart(ticker, start_date, end_date):
         all_days = np.arange(len(x) + future_days)
         log_trend = np.exp(intercept + slope * all_days)
 
-        inner_upper_band = log_trend * 2
-        inner_lower_band = log_trend / 2
-        outer_upper_band = log_trend * 4
-        outer_lower_band = log_trend / 4
+        inner_upper_band = log_trend * np.exp(log_returns_std)
+        inner_lower_band = log_trend * np.exp(-log_returns_std)
+        outer_upper_band = log_trend * np.exp(2 * log_returns_std)
+        outer_lower_band = log_trend * np.exp(-2 * log_returns_std)
 
         extended_dates = pd.date_range(start=data.index[0], periods=len(all_days), freq='D')
 
@@ -90,10 +93,10 @@ def plot_interactive_logarithmic_stock_chart(ticker, start_date, end_date):
 
         fig.add_trace(go.Scatter(x=data.index, y=data['Close'], mode='lines', name='Close Price', line=dict(color='blue')))
         fig.add_trace(go.Scatter(x=extended_dates, y=log_trend, mode='lines', name='Log Trend', line=dict(color='red')))
-        fig.add_trace(go.Scatter(x=extended_dates, y=inner_upper_band, mode='lines', name='Inner Upper Band', line=dict(color='#6FB1A7')))
-        fig.add_trace(go.Scatter(x=extended_dates, y=inner_lower_band, mode='lines', name='Inner Lower Band', line=dict(color='#6FB1A7')))
-        fig.add_trace(go.Scatter(x=extended_dates, y=outer_upper_band, mode='lines', name='Outer Upper Band', line=dict(color='#FFC2A5')))
-        fig.add_trace(go.Scatter(x=extended_dates, y=outer_lower_band, mode='lines', name='Outer Lower Band', line=dict(color='#FFC2A5')))
+        fig.add_trace(go.Scatter(x=extended_dates, y=inner_upper_band, mode='lines', name='1 SD Upper Band', line=dict(color='#6FB1A7')))
+        fig.add_trace(go.Scatter(x=extended_dates, y=inner_lower_band, mode='lines', name='1 SD Lower Band', line=dict(color='#6FB1A7')))
+        fig.add_trace(go.Scatter(x=extended_dates, y=outer_upper_band, mode='lines', name='2 SD Upper Band', line=dict(color='#FFC2A5')))
+        fig.add_trace(go.Scatter(x=extended_dates, y=outer_lower_band, mode='lines', name='2 SD Lower Band', line=dict(color='#FFC2A5')))
 
         fig.update_layout(
             title={
@@ -134,9 +137,9 @@ def plot_interactive_logarithmic_stock_chart(ticker, start_date, end_date):
 # Get the current date
 current_date = datetime.now().strftime("%Y-%m-%d")
 
-# Custom CSS for button hover effect
+# Custom CSS for button and input hover effects
 custom_css = """
-#generate-button:hover {
+#generate-button:hover, #ticker-input:hover, #start-date-input:hover, #end-date-input:hover {
     background-color: #FFB3BA !important;  /* Pastel red */
 }
 """
@@ -147,17 +150,15 @@ with gr.Blocks(theme=seafoam, title="Stock Log Charts", css=custom_css) as iface
     gr.Markdown("Enter a stock ticker and date range to generate a logarithmic chart.")
     
     with gr.Row():
-        ticker = gr.Textbox(label="Stock Ticker", value="MSFT")
-        start_date = gr.Textbox(label="Start Date", value="2015-01-01")
-        end_date = gr.Textbox(label="End Date", value=current_date)
+        ticker = gr.Textbox(label="Stock Ticker", value="MSFT", elem_id="ticker-input")
+        start_date = gr.Textbox(label="Start Date", value="2015-01-01", elem_id="start-date-input")
+        end_date = gr.Textbox(label="End Date", value=current_date, elem_id="end-date-input")
     
     submit_button = gr.Button("Generate Chart", elem_id="generate-button")
     
     with gr.Row():
         log_plot = gr.Plot(label="Logarithmic Stock Chart")
     
-    share_button = gr.Button("Share Chart")
-
     submit_button.click(
         plot_interactive_logarithmic_stock_chart,
         inputs=[ticker, start_date, end_date],