Create app.py

app.py

@@ -0,0 +1,249 @@
+# -------- 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>"
+            "K-Wave 50y • Business 9y • Finance 80y • Hegemony 250y"
+            "</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()