import gradio as gr
import yfinance as yf
import numpy as np
import pandas as pd
import plotly.graph_objects as go
from datetime import datetime

def download_stock_data(ticker, start_date, end_date):
    stock = yf.Ticker(ticker)
    df = stock.history(start=start_date, end=end_date)
    return df

def plot_interactive_logarithmic_stock_chart(ticker, start_date, end_date):
    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."

        x = (data.index - data.index[0]).days
        y = np.log(data['Close'])
        slope, intercept = np.polyfit(x, y, 1)

        future_days = 365 * 10
        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

        extended_dates = pd.date_range(start=data.index[0], periods=len(all_days), freq='D')

        fig = go.Figure()

        fig.add_trace(go.Scatter(x=data.index, y=data['Close'], mode='lines', name='Close Price', line=dict(color='#3366CC')))
        fig.add_trace(go.Scatter(x=extended_dates, y=log_trend, mode='lines', name='Log Trend', line=dict(color='#DC3912')))
        fig.add_trace(go.Scatter(x=extended_dates, y=inner_upper_band, mode='lines', name='Inner Upper Band', line=dict(color='#FF9900')))
        fig.add_trace(go.Scatter(x=extended_dates, y=inner_lower_band, mode='lines', name='Inner Lower Band', line=dict(color='#FF9900')))
        fig.add_trace(go.Scatter(x=extended_dates, y=outer_upper_band, mode='lines', name='Outer Upper Band', line=dict(color='#109618')))
        fig.add_trace(go.Scatter(x=extended_dates, y=outer_lower_band, mode='lines', name='Outer Lower Band', line=dict(color='#109618')))

        fig.update_layout(
            title={
                'text': 'Stock Log Charts',
                'y': 0.95,
                'x': 0.5,
                'xanchor': 'center',
                'yanchor': 'top'
            },
            xaxis_title='Date',
            yaxis_title='Price (Log Scale)',
            yaxis_type="log",
            height=800,
            legend=dict(x=0.01, y=0.99, bgcolor='rgba(255, 255, 255, 0.8)'),
            hovermode='x unified',
            plot_bgcolor='#F5F5F5',
            paper_bgcolor='#F5F5F5',
            font=dict(family="Arial", size=12, color="#333333")
        )

        fig.update_xaxes(
            rangeslider_visible=True,
            rangeselector=dict(
                buttons=list([
                    dict(count=1, label="1m", step="month", stepmode="backward"),
                    dict(count=6, label="6m", step="month", stepmode="backward"),
                    dict(count=1, label="YTD", step="year", stepmode="todate"),
                    dict(count=1, label="1y", step="year", stepmode="backward"),
                    dict(step="all")
                ])
            )
        )

        return fig
    except Exception as e:
        return f"An error occurred: {str(e)}"

# Get the current date
current_date = datetime.now().strftime("%Y-%m-%d")

# Custom CSS to make charts full-width and larger, and apply a theme
custom_css = """
.container {max-width: 100% !important; padding: 0 !important;}
.plot-container {height: 800px !important; width: 100% !important;}
.react-plotly-container {height: 100% !important; width: 100% !important;}
body {background-color: #F5F5F5; color: #333333; font-family: Arial, sans-serif;}
.gradio-container {background-color: #FFFFFF; border-radius: 10px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);}
.gr-button {background-color: #3366CC; color: white;}
.gr-button:hover {background-color: #254EDB;}
.gr-input {border: 1px solid #CCCCCC; border-radius: 4px;}
.gr-input:focus {border-color: #3366CC; box-shadow: 0 0 0 2px rgba(51, 102, 204, 0.2);}
"""

# Create Gradio interface
with gr.Blocks(css=custom_css, title="Stock Log Charts") as iface:
    gr.Markdown("# Stock Log Charts", elem_id="main-title")
    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)
    
    submit_button = gr.Button("Generate Chart")
    
    with gr.Row():
        log_plot = gr.Plot(label="Logarithmic Stock Chart")
    
    submit_button.click(
        plot_interactive_logarithmic_stock_chart,
        inputs=[ticker, start_date, end_date],
        outputs=[log_plot]
    )

# Launch the app
iface.launch()