import gradio as gr
import seaborn as sns
import matplotlib.pyplot as plt
import plotly.express as px
import pandas as pd

# Load dataset
df = sns.load_dataset('tips')

# Plot functions
def basic_plot(plot_type):
    plt.figure(figsize=(6,4))
    if plot_type == "Scatter":
        sns.scatterplot(x='total_bill', y='tip', data=df)
    elif plot_type == "Line":
        sns.lineplot(x='size', y='tip', data=df)
    elif plot_type == "Bar":
        sns.barplot(x='day', y='total_bill', data=df)
    elif plot_type == "Histogram":
        sns.histplot(df['total_bill'], bins=10, kde=True)
    plt.title(f"{plot_type} Plot")
    return plt.gcf()

def advanced_plot(plot_type):
    if plot_type == "Heatmap":
        plt.figure(figsize=(6,4))
        sns.heatmap(df.corr(), annot=True, cmap='coolwarm')
        plt.title("Heatmap")
        return plt.gcf()
    elif plot_type == "Pair Plot":
        g = sns.pairplot(df[['total_bill', 'tip', 'size']])
        return g.fig
    elif plot_type == "Facet Grid":
        g = sns.FacetGrid(df, col='sex', row='time')
        g.map(sns.scatterplot, 'total_bill', 'tip')
        return g.fig

def interactive_plot(_):
    fig = px.scatter(df, x='total_bill', y='tip', color='sex',
                     size='size', hover_data=['day', 'time'],
                     title='Interactive Scatter Plot')
    return fig

def storytelling_plot(_):
    avg_tip = df.groupby('day')['tip'].mean().reset_index()
    plt.figure(figsize=(6,4))
    sns.barplot(x='day', y='tip', data=avg_tip, palette='viridis')
    plt.axhline(y=avg_tip['tip'].mean(), color='red', linestyle='--', label='Overall Avg')
    plt.title('Average Tip by Day')
    plt.legend()
    return plt.gcf()

# Gradio UI
with gr.Blocks() as demo:
    gr.Markdown("## 📊 Data Visualization Explorer")

    with gr.Tab("Basic Plots"):
        with gr.Row():
            plot_type = gr.Dropdown(["Scatter", "Line", "Bar", "Histogram"], label="Choose Plot Type")
            submit_btn = gr.Button("Submit")
        plot_output = gr.Plot()
        submit_btn.click(fn=basic_plot, inputs=plot_type, outputs=plot_output)

    with gr.Tab("Advanced Plots"):
        with gr.Row():
            adv_type = gr.Dropdown(["Heatmap", "Pair Plot", "Facet Grid"], label="Choose Advanced Plot")
            adv_btn = gr.Button("Submit")
        adv_output = gr.Plot()
        adv_btn.click(fn=advanced_plot, inputs=adv_type, outputs=adv_output)

    with gr.Tab("Interactive Plot"):
        inter_btn = gr.Button("Submit")
        inter_output = gr.Plot()
        inter_btn.click(fn=interactive_plot, inputs=[], outputs=inter_output)

    with gr.Tab("Storytelling"):
        story_btn = gr.Button("Submit")
        story_output = gr.Plot()
        story_btn.click(fn=storytelling_plot, inputs=[], outputs=story_output)

demo.launch()