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='#3A9184')))
        fig.add_trace(go.Scatter(x=extended_dates, y=log_trend, mode='lines', name='Log Trend', line=dict(color='#FF8C61')))
        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.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='#F5F9F8',
            paper_bgcolor='#F5F9F8',
            font=dict(family="Inter, sans-serif", size=12, color="#313D38")
        )

        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")

# Create Gradio interface with seafoam theme
with gr.Blocks(theme=gr.themes.Seafoam(), title="Stock Log Charts") as iface:
    gr.Markdown("# Stock Log Charts")
    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()