import gradio as gr
import pandas as pd
import pandas_ta as ta
import plotly.graph_objects as go
from plotly.subplots import make_subplots

# Sample OHLCV data (replace this with actual data)
data = pd.read_csv('data/brent_futures.csv')
data['Date'] = pd.to_datetime(data['Date'])
data.set_index('Date', inplace=True)
data.sort_index(inplace=True)

def plot_chart(ma_length, draw_mode):
    # Calculate Moving Average and Bollinger Bands
    data['MA'] = ta.sma(data['Close'], length=ma_length)
    bbands = ta.bbands(data['Close'], length=20)
    data['BB_upper'], data['BB_middle'], data['BB_lower'] = bbands.iloc[:, 2], bbands.iloc[:, 1], bbands.iloc[:, 0]
    
    # Create a subplot figure with 2 rows: OHLC and Volume
    fig = make_subplots(
        rows=2, cols=1, shared_xaxes=True, vertical_spacing=0.05,
        row_heights=[0.7, 0.3], subplot_titles=('OHLC with Moving Average and Bollinger Bands', 'Volume')
    )
    
    # Plot OHLC candlestick
    fig.add_trace(go.Candlestick(
        x=data.index, open=data['Open'], high=data['High'], low=data['Low'], close=data['Close'], name="OHLC"), row=1, col=1)
    
    # Add Moving Average and Bollinger Bands
    fig.add_trace(go.Scatter(x=data.index, y=data['MA'], mode='lines', name=f'MA {ma_length}', line=dict(color='blue')), row=1, col=1)
    fig.add_trace(go.Scatter(x=data.index, y=data['BB_upper'], mode='lines', name='BB Upper', line=dict(color='purple')), row=1, col=1)
    fig.add_trace(go.Scatter(x=data.index, y=data['BB_middle'], mode='lines', name='BB Middle', line=dict(color='gray')), row=1, col=1)
    fig.add_trace(go.Scatter(x=data.index, y=data['BB_lower'], mode='lines', name='BB Lower', line=dict(color='purple')), row=1, col=1)
    
    # Plot Volume
    fig.add_trace(go.Bar(x=data.index, y=data['Volume'], name='Volume', marker=dict(color='green')), row=2, col=1)
    
    # Customize layout
    fig.update_layout(
        height=700, showlegend=True,
        legend=dict(orientation='h', yanchor='bottom', y=-0.15, xanchor='center', x=0.5),
        xaxis_rangeslider_visible=False, dragmode='drawline' if draw_mode else 'zoom'
    )
    
    # Enable dynamic y-axis range
    fig.update_yaxes(autorange=True)
    
    return fig

# Gradio UI
def update_chart(ma_length, drawline_button):
    draw_mode = True if drawline_button == 'Draw Line' else False
    return plot_chart(ma_length, draw_mode)

ma_length_input = gr.Number(value=10, label="Moving Average Length", interactive=True, step=1, minimum=1)
drawline_button = gr.Button("Draw Line")
chart_output = gr.Plot()

gr.Interface(
    fn=update_chart,
    inputs=[ma_length_input, drawline_button],
    outputs=[chart_output],
    title="Interactive Technical Analysis Dashboard",
    description="Choose moving average length and toggle line drawing mode."
).launch()