# -------- app_final.py --------
import os, json, math, pathlib
import numpy as np
import plotly.graph_objects as go
import gradio as gr
import openai

# ──────────────────────────
# 0. OpenAI API key
# ──────────────────────────
if "OPENAI_API_KEY" not in os.environ:
    os.environ["OPENAI_API_KEY"] = input("🔑 Enter your OpenAI API key: ").strip()
openai.api_key = os.environ["OPENAI_API_KEY"]

# ──────────────────────────
# 1. Cycle config
# ──────────────────────────
CENTER = 2025
CYCLES = {
    "K-Wave":   50,   # Kondratiev long wave
    "Business":  9,   # Juglar investment cycle
    "Finance":  80,   # Long credit cycle
    "Hegemony": 250   # Power-shift cycle
}
ORDERED_PERIODS = sorted(CYCLES.values())            # [9, 50, 80, 250]
COLOR = {9:"#66ff66", 50:"#ff3333", 80:"#ffcc00", 250:"#66ccff"}
AMPL  = {9:0.6,        50:1.0,      80:1.6,        250:4.0}
PERIOD_BY_CYCLE = {k:v for k,v in CYCLES.items()}

# ──────────────────────────
# 2. Load events JSON
# ──────────────────────────
EVENTS_PATH = pathlib.Path(__file__).with_name("cycle_events.json")
with open(EVENTS_PATH, encoding="utf-8") as f:
    EVENTS = {int(item["year"]): item["events"] for item in json.load(f)}

# ──────────────────────────
# 3. Helper functions
# ──────────────────────────
def half_sine(xs, period, amp):
    """Half-sine ‘tower’ with zero bottom floor."""
    phase = np.mod(xs - CENTER, period)
    y = amp * np.sin(np.pi * phase / period)
    y[y < 0] = 0
    return y

def build_chart(start: int, end: int):
    """Return (Plotly figure, summary string) for the requested year range."""
    xs = np.linspace(start, end, max(1000, (end - start) * 4))
    fig = go.Figure()

    # ◼ 1) Draw gradient half-sine ‘towers’ (30 layers each, like old matplotlib version)
    for period in ORDERED_PERIODS:
        base_amp = AMPL[period]
        color    = COLOR[period]

        # ── gradient layers (thin, semi-transparent) ──────────────────────
        for frac in np.linspace(base_amp / 30, base_amp, 30):
            ys = half_sine(xs, period, frac)
            fig.add_trace(go.Scatter(
                x = xs,
                y = ys,
                mode = "lines",
                line = dict(color=color, width=0.8),
                opacity = 0.6,
                hoverinfo = "skip",
                showlegend = False
            ))

        # ── outer edge line (slightly thicker) ───────────────────────────
        ys = half_sine(xs, period, base_amp)
        fig.add_trace(go.Scatter(
            x = xs,
            y = ys,
            mode = "lines",
            line = dict(color=color, width=1.6),
            hoverinfo = "skip",
            showlegend = False
        ))

    # ◼ 2) Event markers with bilingual hover text ─────────────────────────
    for year, evs in EVENTS.items():
        if start <= year <= end:
            cycle_name = evs[0]["cycle"]
            period     = PERIOD_BY_CYCLE.get(cycle_name, 50)
            y_val      = float(half_sine(np.array([year]), period, AMPL[period]))
            txt_en     = "<br>".join(e["event_en"] for e in evs)
            txt_ko     = "<br>".join(e["event_ko"] for e in evs)

            fig.add_trace(go.Scatter(
                x = [year], y = [y_val],
                mode = "markers",
                marker = dict(color="white", size=6),
                customdata = [[cycle_name, txt_en, txt_ko]],
                hovertemplate = (
                    "Year %{x} • %{customdata[0]} cycle"
                    "<br>%{customdata[1]}"
                    "<br>%{customdata[2]}<extra></extra>"
                ),
                showlegend = False
            ))

    # ◼ 3) Cosmetic touches: centre line, 250-yr arrow, background, axes ────
    # centre alignment marker
    fig.add_vline(
        x = CENTER,
        line_dash = "dash",
        line_color = "white",
        opacity = 0.6
    )

    # 250-year span arrow (↔) across the top of chart
    arrow_y = AMPL[250] * 1.05
    fig.add_annotation(
        x = CENTER - 125, y = arrow_y,
        ax = CENTER + 125, ay = arrow_y,
        xref = "x", yref = "y", axref = "x", ayref = "y",
        showarrow = True,
        arrowhead = 3,
        arrowsize = 1,
        arrowwidth = 1.2,
        arrowcolor = "white"
    )
    fig.add_annotation(
        x = CENTER, y = arrow_y + 0.15,
        text = "250 yr",
        showarrow = False,
        font = dict(color="white", size=10)
    )

    # global styling
    fig.update_layout(
        template = "plotly_dark",
        paper_bgcolor = "black",
        plot_bgcolor  = "black",
        height = 450,
        margin = dict(t=30, l=40, r=40, b=40),
        hoverlabel = dict(bgcolor="#222", font_size=11)
    )
    fig.update_xaxes(title="Year",  range=[start, end], showgrid=False)
    fig.update_yaxes(title="Relative amplitude", showticklabels=False, showgrid=False)

    summary = f"Range {start}-{end} | Events plotted: {sum(1 for y in EVENTS if start <= y <= end)}"
    return fig, summary

# ──────────────────────────
# 4. GPT chat helper
# ──────────────────────────
BASE_PROMPT = (
    "You are a concise and accurate Korean assistant. "
    "If a chart summary is provided, incorporate it in your answer."
)

def chat_with_gpt(history, user_msg, chart_summary):
    messages = [{"role": "system", "content": BASE_PROMPT}]
    if chart_summary not in ("", "No chart yet."):
        messages.append({"role": "system", "content": f"[Chart summary]\n{chart_summary}"})

    for u, a in history:
        messages.extend([{"role": "user", "content": u},
                         {"role": "assistant", "content": a}])
    messages.append({"role": "user", "content": user_msg})

    res = openai.chat.completions.create(
        model = "gpt-3.5-turbo",
        messages = messages,
        max_tokens = 600,
        temperature = 0.7
    )
    return res.choices[0].message.content.strip()

# ──────────────────────────
# 5. Gradio UI
# ──────────────────────────
def create_app():
    with gr.Blocks(theme=gr.themes.Soft()) as demo:
        # ── Title & subtitle
        gr.Markdown("## 🔭 **CycleNavigator (Interactive)**")
        gr.Markdown("### <sub>Interactive visual service delivering insights at a glance into the four major long cycles—9-year business cycle, 50-year Kondratiev wave, 80-year financial credit cycle, and 250-year hegemony cycle—through dynamic charts and AI chat.</sub>")        
        gr.Markdown(
            "<sub>"
            "<b>Business 9y</b> (credit-investment business cycle) • "
            "<b>K-Wave 50y</b> (long technological-industrial wave) • "
            "<b>Finance 80y</b> (long credit-debt cycle) • "
            "<b>Hegemony 250y</b> (rise & fall of global powers cycle)"
            "</sub>"
        )
    

        chart_summary_state = gr.State(value="No chart yet.")

        with gr.Tabs():
            # ── Tab 1: Timeline Chart ─────────────────────────────────────
            with gr.TabItem("Timeline Chart"):
                with gr.Row():
                    start_year = gr.Number(label="Start Year", value=1500, precision=0)
                    end_year   = gr.Number(label="End Year",   value=2500, precision=0)
                    zoom_in  = gr.Button("🔍 Zoom In")
                    zoom_out = gr.Button("🔎 Zoom Out")

                # initial plot
                fig0, summ0 = build_chart(1500, 2500)
                plot = gr.Plot(value=fig0)
                chart_summary_state.value = summ0

                # refresh on year change
                def refresh(s, e):
                    fig, summ = build_chart(int(s), int(e))
                    return fig, summ
                start_year.change(refresh, [start_year, end_year],
                                             [plot, chart_summary_state])
                end_year.change(refresh,   [start_year, end_year],
                                             [plot, chart_summary_state])

                # zoom helpers
                def zoom(s, e, factor):
                    mid  = (s + e) / 2
                    span = (e - s) * factor / 2
                    ns, ne = int(mid - span), int(mid + span)
                    fig, summ = build_chart(ns, ne)
                    return ns, ne, fig, summ

                zoom_in.click(lambda s, e: zoom(s, e, 0.5),
                              inputs  = [start_year, end_year],
                              outputs = [start_year, end_year, plot, chart_summary_state])
                zoom_out.click(lambda s, e: zoom(s, e, 2.0),
                               inputs  = [start_year, end_year],
                               outputs = [start_year, end_year, plot, chart_summary_state])

                # JSON download
                gr.File(value=str(EVENTS_PATH), label="Download cycle_events.json")

            # ── Tab 2: GPT Chat ───────────────────────────────────────────
            with gr.TabItem("GPT Chat"):
                chatbot    = gr.Chatbot(label="Assistant")
                user_input = gr.Textbox(lines=3, placeholder="메시지를 입력하세요…")
                send_btn   = gr.Button("Send")

                def respond(history, msg, summ):
                    reply = chat_with_gpt(history, msg, summ)
                    history.append((msg, reply))
                    return history, gr.Textbox(value="")

                send_btn.click(respond,
                               [chatbot, user_input, chart_summary_state],
                               [chatbot, user_input])
                user_input.submit(respond,
                                  [chatbot, user_input, chart_summary_state],
                                  [chatbot, user_input])
    return demo


if __name__ == "__main__":
    create_app().launch()