Update app.py

app.py

@@ -63,13 +63,13 @@ def download_stock_data(ticker, start_date, end_date):
     df = stock.history(start=start_date, end=end_date)
     return df
 
-def plot_chart(ticker, start_date, end_date, chart_type):
+def plot_interactive_stock_chart(ticker, start_date, end_date, chart_type):
     try:
         stock = yf.Ticker(ticker)
         data = stock.history(start=start_date, end=end_date)
 
         if data.empty:
-            return "No data available for the specified date range.", None, None, None
+            return "No data available for the specified date range."
 
         if chart_type == "Log":
             return plot_logarithmic_chart(data, ticker)
@@ -77,7 +77,7 @@ def plot_chart(ticker, start_date, end_date, chart_type):
             return plot_candlestick_chart(data, ticker)
 
     except Exception as e:
-        return f"An error occurred: {str(e)}", None, None, None
+        return f"An error occurred: {str(e)}"
 
 def plot_logarithmic_chart(data, ticker):
     x = (data.index - data.index[0]).days
@@ -136,11 +136,7 @@ def plot_logarithmic_chart(data, ticker):
         )
     )
 
-    current_price = data['Close'].iloc[-1]
-    log_price = log_trend[-len(data):]
-    percent_diff = ((current_price - log_price.iloc[-1]) / log_price.iloc[-1]) * 100
-
-    return fig, current_price, log_price.iloc[-1], percent_diff
+    return fig
 
 def plot_candlestick_chart(data, ticker):
     fig = go.Figure(data=[go.Candlestick(x=data.index,
@@ -180,33 +176,7 @@ def plot_candlestick_chart(data, ticker):
         )
     )
 
-    current_price = data['Close'].iloc[-1]
-    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}"
-    
-    color = "green" if percent_diff > 0 else "red"
-    intensity = min(abs(percent_diff) / 100, 1)  # Normalize to 0-1
-    bg_color = f"rgba({255 if color == 'red' else 0}, {255 if color == 'green' else 0}, 0, {intensity})"
-    
-    return f"""
-    <div style="display: flex; justify-content: space-between; padding: 10px; background-color: {bg_color}; border-radius: 10px;">
-        <div>
-            <p>Current Price: ${current_price:.2f}</p>
-            <p>Log Price: ${log_price:.2f}</p>
-        </div>
-        <div>
-            <p style="font-size: 1.2em; font-weight: bold; color: {color};">
-                {percent_diff:.2f}% {'above' if percent_diff > 0 else 'below'} log price
-            </p>
-        </div>
-    </div>
-    """
+    return fig
 
 # Get the current date
 current_date = datetime.now().strftime("%Y-%m-%d")
@@ -222,7 +192,7 @@ custom_css = """
 }
 """
 
-# Update the Gradio interface section
+# Create Gradio interface with custom Seafoam theme
 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.")
@@ -238,19 +208,12 @@ with gr.Blocks(theme=seafoam, title="Stock Charts", css=custom_css) as iface:
     submit_button = gr.Button("Generate Chart", elem_id="generate-button")
     
     with gr.Row():
-        chart = gr.Plot(label="Stock Chart")
+        stock_chart = gr.Plot(label="Stock Chart")
     
-    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(
-        update_chart_and_info,
+        plot_interactive_stock_chart,
         inputs=[ticker, start_date, end_date, chart_type],
-        outputs=[chart, price_info]
+        outputs=[stock_chart]
     )
 
 # Launch the app