Update app.py

app.py

@@ -1,181 +1,195 @@
 # -------- app_final.py --------
 import os, json, math, pathlib, re, time, logging, requests
+from datetime import datetime, timedelta
 import numpy as np
+import pandas as pd
 import plotly.graph_objects as go
 import gradio as gr
 import openai
+import torch
+from sentence_transformers import SentenceTransformer, util
 
-# ──────────────────────────
-# 0. API keys & Brave Search
-# ──────────────────────────
+# ────────────────────────── 0. API keys & Brave Search ──────────────────────────
 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"]
 
-BRAVE_KEY      = os.getenv("BRAVE_KEY", "")        # Brave Search API Key
+BRAVE_KEY      = os.getenv("BRAVE_KEY", "")
 BRAVE_ENDPOINT = "https://api.search.brave.com/res/v1/web/search"
 logging.basicConfig(level=logging.INFO)
 
-# ──────────────────────────
-# 1. Cycle config
-# ──────────────────────────
+# ────────────────────────── 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]
+CYCLES = { "K-Wave": 50, "Business": 9, "Finance": 80, "Hegemony": 250 }
+ORDERED_PERIODS = sorted(CYCLES.values())
 COLOR = {9:"#66ff66", 50:"#ff3333", 80:"#ffcc00", 250:"#66ccff"}
-AMPL  = {9:0.6,        50:1.0,      80:1.6,        250:4.0}
+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
-# ──────────────────────────
+# ────────────────────────── 2. Load events JSON & embeddings ───────────────────
 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. Brave Search helpers
-# ──────────────────────────
-def brave_search(query: str, count: int = 8):
-    """Call Brave Search API; return list of dicts."""
+    RAW_EVENTS = json.load(f)
+EVENTS = {int(item["year"]): item["events"] for item in RAW_EVENTS}
+
+logging.info("Embedding historical events…")
+_embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
+_all_sentences = [(yr, ev["event_en"]) for yr, evs in EVENTS.items() for ev in evs]
+_embeddings = _embed_model.encode([s for _, s in _all_sentences], convert_to_tensor=True)
+
+# 유사 사건 top-3 year 사전
+SIMILAR_MAP = {}
+for idx, (yr, _) in enumerate(_all_sentences):
+    scores = util.cos_sim(_embeddings[idx], _embeddings)[0]
+    top_idx = torch.topk(scores, 4).indices.tolist()
+    sims = [_all_sentences[i][0] for i in top_idx if _all_sentences[i][0] != yr][:3]
+    SIMILAR_MAP.setdefault(yr, sims)
+
+# ────────────────────────── 3. Brave Search helpers ──────────────────────────
+def brave_search(query: str, count: int = 8, freshness_days: int | None = None):
     if not BRAVE_KEY:
-        logging.warning("⚠️  BRAVE_KEY is empty; web search disabled.")
         return []
+    params = {"q": query, "count": str(count)}
+    if freshness_days:
+        dt_from = (datetime.utcnow() - timedelta(days=freshness_days)).strftime("%Y-%m-%d")
+        params["freshness"] = dt_from
     try:
-        hdrs = {
-            "Accept": "application/json",
-            "X-Subscription-Token": BRAVE_KEY
-        }
-        data = requests.get(
+        r = requests.get(
             BRAVE_ENDPOINT,
-            headers=hdrs,
-            params={"q": query, "count": str(count)},
+            headers={"Accept": "application/json", "X-Subscription-Token": BRAVE_KEY},
+            params=params,
             timeout=15
-        ).json()
-        raw = data.get("web", {}).get("results") or []
-        return [
-            {
-                "title": r.get("title", ""),
-                "url":   r.get("url", r.get("link", "")),
-                "snippet": r.get("description", r.get("text", "")),
-                "host": re.sub(r"https?://(www\.)?", "", r.get("url", "")).split("/")[0]
-            }
-            for r in raw[:count]
-        ]
+        )
+        raw = r.json().get("web", {}).get("results") or []
+        return [{
+            "title": r.get("title", ""),
+            "url": r.get("url", r.get("link", "")),
+            "snippet": r.get("description", r.get("text", "")),
+            "host": re.sub(r"https?://(www\.)?", "", r.get("url", "")).split("/")[0]
+        } for r in raw[:count]]
     except Exception as e:
-        logging.error(f"Brave Search Error: {e}")
+        logging.error(f"Brave error: {e}")
         return []
 
 def format_search_results(query: str) -> str:
-    """Format search results as markdown for LLM context."""
-    arts = brave_search(query)
-    if not arts:
+    rows = brave_search(query, 6, freshness_days=3)
+    if not rows:
         return f"# [Web-Search] No live results for “{query}”.\n"
-    hdr = f"# [Web-Search] Top results for “{query}”\n\n"
+    hdr = f"# [Web-Search] Top results for “{query}” (last 3 days)\n\n"
     body = "\n".join(
-        f"**{i+1}. {a['title']}** ({a['host']})\n\n"
-        f"{a['snippet']}\n\n"
-        f"[Source]({a['url']})\n"
-        for i, a in enumerate(arts)
+        f"- **{r['title']}** ({r['host']})\n  {r['snippet']}\n  [link]({r['url']})"
+        for r in rows
     )
     return hdr + body + "\n"
 
-# ──────────────────────────
-# 4. Helper functions (chart)
-# ──────────────────────────
+NEWS_KEYWORDS = {
+    "Business": "recession OR GDP slowdown",
+    "K-Wave":   "breakthrough technology innovation",
+    "Finance":  "credit cycle debt crisis",
+    "Hegemony": "great power rivalry geopolitics"
+}
+def fetch_cycle_news():
+    markers = []
+    for cyc, kw in NEWS_KEYWORDS.items():
+        res = brave_search(kw, 1, freshness_days=2)
+        if res:
+            markers.append({
+                "cycle": cyc,
+                "title": res[0]["title"],
+                "year": datetime.utcnow().year,
+                "url": res[0]["url"]
+            })
+    return markers
+NEWS_MARKERS = fetch_cycle_news()
+
+# ────────────────────────── 4. Chart helpers ──────────────────────────
 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,
-       plus a transparent hover-trace so the cursor always shows ‘Year ####’. """
+def build_chart(start: int, end: int, lang: str = "KO"):
     xs = np.linspace(start, end, max(1000, (end - start) * 4))
     fig = go.Figure()
 
-    # 1) Gradient half-sine towers
+    # Gradient towers
     for period in ORDERED_PERIODS:
-        base_amp, color = AMPL[period], COLOR[period]
-        for frac in np.linspace(base_amp / 30, base_amp, 30):
+        base, col = AMPL[period], COLOR[period]
+        for frac in np.linspace(base / 30, base, 30):
             fig.add_trace(go.Scatter(
                 x=xs, y=half_sine(xs, period, frac),
-                mode="lines",
-                line=dict(color=color, width=0.8),
-                opacity=0.6,
-                hoverinfo="skip", showlegend=False))
+                mode="lines", line=dict(color=col, width=0.8),
+                opacity=0.6, hoverinfo="skip", showlegend=False))
         fig.add_trace(go.Scatter(
-            x=xs, y=half_sine(xs, period, base_amp),
-            mode="lines",
-            line=dict(color=color, width=1.6),
+            x=xs, y=half_sine(xs, period, base),
+            mode="lines", line=dict(color=col, width=1.6),
             hoverinfo="skip", showlegend=False))
 
-    # 2) Event markers (bilingual hover)
-    for year, evs in EVENTS.items():
-        if start <= year <= end:
-            cyc = evs[0]["cycle"]; period = PERIOD_BY_CYCLE[cyc]
-            y_val = float(half_sine(np.array([year]), period, AMPL[period]))
+    # Events + similar
+    text_key = "event_ko" if lang == "KO" else "event_en"
+    for yr, evs in EVENTS.items():
+        if start <= yr <= end:
+            cyc = evs[0]["cycle"]
+            period = PERIOD_BY_CYCLE[cyc]
+            yv = float(half_sine(np.array([yr]), period, AMPL[period]))
+            sim = ", ".join(map(str, SIMILAR_MAP.get(yr, []))) or "None"
+            txt = "<br>".join(e[text_key] for e in evs)
             fig.add_trace(go.Scatter(
-                x=[year], y=[y_val], mode="markers",
+                x=[yr], y=[yv], mode="markers",
                 marker=dict(color="white", size=6),
-                customdata=[[cyc,
-                             "<br>".join(e["event_en"] for e in evs),
-                             "<br>".join(e["event_ko"] for e in evs)]],
+                customdata=[[cyc, txt, sim]],
                 hovertemplate=(
-                    "Year %{x} • %{customdata[0]} cycle"
-                    "<br>%{customdata[1]}"
-                    "<br>%{customdata[2]}<extra></extra>"
+                    "Year %{x} • %{customdata[0]}<br>"
+                    "%{customdata[1]}<br>"
+                    "Similar: %{customdata[2]}<extra></extra>"
                 ),
                 showlegend=False))
 
-    # 3) Transparent hover-trace so any x shows ‘Year ####’
+    # Live-news markers
+    for m in NEWS_MARKERS:
+        if start <= m["year"] <= end:
+            p = PERIOD_BY_CYCLE[m["cycle"]]
+            yv = float(half_sine(np.array([m["year"]]), p, AMPL[p])) * 1.05
+            fig.add_trace(go.Scatter(
+                x=[m["year"]], y=[yv], mode="markers+text",
+                marker=dict(color="gold", size=8, symbol="star"),
+                text=["📰"], textposition="top center",
+                customdata=[[m["cycle"], m["title"], m["url"]]],
+                hovertemplate=("Live news • %{customdata[0]}<br>"
+                               "%{customdata[1]}<extra></extra>"),
+                showlegend=False))
+
+    # Hover Year trace
     fig.add_trace(go.Scatter(
-        x=xs,
-        y=np.full_like(xs, -0.05),                 # just below baseline
-        mode="lines",
-        line=dict(color="rgba(0,0,0,0)", width=1), # invisible
-        hovertemplate="Year %{x:.0f}<extra></extra>",
-        showlegend=False))
-
-    # 4) Cosmetic touches: centre line & 250-yr arrow
+        x=xs, y=np.full_like(xs, -0.05),
+        mode="lines", line=dict(color="rgba(0,0,0,0)", width=1),
+        hovertemplate="Year %{x:.0f}<extra></extra>", showlegend=False))
+
+    # Cosmetics
     fig.add_vline(x=CENTER, line_dash="dash", line_color="white", opacity=0.6)
     arrow_y = AMPL[250] * 1.05
     fig.add_annotation(
-        x=CENTER-125, y=arrow_y, ax=CENTER+125, ay=arrow_y,
+        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",
+        x=CENTER, y=arrow_y + 0.15, text="250 yr",
         showarrow=False, font=dict(color="white", size=10))
-
-    # 5) Layout
     fig.update_layout(
         template="plotly_dark",
         paper_bgcolor="black", plot_bgcolor="black",
-        height=450,
-        margin=dict(t=30, l=40, r=40, b=40),
+        height=500, margin=dict(t=30, l=40, r=40, b=40),
         hoverlabel=dict(bgcolor="#222", font_size=11),
-        hovermode="x"                               # unified x-hover
-    )
-    fig.update_xaxes(title="Year",  range=[start, end], showgrid=False)
-    fig.update_yaxes(title="Relative amplitude",
-                     showticklabels=False, showgrid=False)
+        hovermode="x")
+    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: " \
-              f"{sum(1 for y in EVENTS if start <= y <= end)}"
+    summary = f"Range {start}-{end} | Events: {sum(1 for y in EVENTS if start <= y <= end)}"
     return fig, summary
 
-# ──────────────────────────
-# 5. GPT chat helper
-# ──────────────────────────
+# ────────────────────────── 5. GPT helper ──────────────────────────
 BASE_PROMPT = (
     "당신은 **CycleNavigator AI**로, 경제사·국제정치·장주기(9y Business, 50y K-Wave, "
     "80y Finance, 250y Hegemony) 분석에 정통한 전문가입니다. "
@@ -189,106 +203,111 @@ BASE_PROMPT = (
     "✦ 불필요한 장황함은 삼가고 3개 단락 또는 7개 이하 bullets 내로 요약하십시오."
 )
 
-def chat_with_gpt(history, user_msg, chart_summary):
-    messages = [{"role": "system", "content": BASE_PROMPT}]
+def chat_with_gpt(hist, msg, chart_summary):
+    msgs = [{"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(
+        msgs.append({"role": "system", "content": f"[Chart summary]\n{chart_summary}"})
+    for u, a in hist:
+        msgs.extend([{"role": "user", "content": u}, {"role": "assistant", "content": a}])
+    msgs.append({"role": "user", "content": msg})
+    return openai.chat.completions.create(
         model="gpt-3.5-turbo",
-        messages=messages,
+        messages=msgs,
         max_tokens=600,
         temperature=0.7
-    )
-    return res.choices[0].message.content.strip()
+    ).choices[0].message.content.strip()
 
-# ──────────────────────────
-# 6. Gradio UI
-# ──────────────────────────
+# ────────────────────────── 6. Gradio UI ──────────────────────────
 def create_app():
-    with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        # Title & subtitles
+    with gr.Blocks(
+        theme=gr.themes.Soft(),
+        css="""
+           #discord-badge{position:fixed; bottom:10px; left:50%;
+                          transform:translateX(-50%);}
+        """
+    ) as demo:
         gr.Markdown("## 🔭 **CycleNavigator (Interactive)**")
-        gr.Markdown(
-            "<sub>"
-            "<b>Interactive visual service delivering insights at a glance into the four major long cycles— </b>"
-            "<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)"
-            "<b> —through dynamic charts and AI chat.</b>"
-            "</sub>"
+# ── 언어 선택 ───────────────────────────────────────────
+        lang_state = gr.State(value="KO")
+
+        lang_radio = gr.Radio(
+            ["English", "한국어"],
+            value="English",
+            label="Language / 언어",
+            interactive=True        # ← 이것만 두고
         )
 
+
+        # 초기 차트·상태
         chart_summary_state = gr.State(value="No chart yet.")
+        fig0, summ0 = build_chart(1500, 2500, "KO")
+        plot = gr.Plot(value=fig0)
+        chart_summary_state.value = summ0
+
+        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")
+
+        # ── functions ──
+        def refresh(s, e, lang_code):
+            fig, summ = build_chart(int(s), int(e), lang_code)
+            return fig, summ
+
+        def zoom(s, e, f, lang_code):
+            mid = (s + e) / 2
+            span = (e - s) * f / 2
+            ns, ne = int(mid - span), int(mid + span)
+            fig, summ = build_chart(ns, ne, lang_code)
+            return ns, ne, fig, summ
+
+        def change_lang(lang_label, s, e):
+            code = "KO" if lang_label == "한국어" else "EN"
+            fig, summ = build_chart(int(s), int(e), code)
+            return code, fig, summ
+
+        # ── event wiring ──
+        start_year.change(refresh, [start_year, end_year, lang_state], [plot, chart_summary_state])
+        end_year.change(refresh,   [start_year, end_year, lang_state], [plot, chart_summary_state])
+
+        zoom_in.click(
+            lambda s, e, lc: zoom(s, e, 0.5, lc),
+            [start_year, end_year, lang_state],
+            [start_year, end_year, plot, chart_summary_state]
+        )
+        zoom_out.click(
+            lambda s, e, lc: zoom(s, e, 2.0, lc),
+            [start_year, end_year, lang_state],
+            [start_year, end_year, plot, chart_summary_state]
+        )
 
-        with gr.Tabs():
-            with gr.TabItem("Timeline Chart"):
-    # 1️⃣ 먼저 차트를 만들고 화면에 배치
-                fig0, summ0 = build_chart(1500, 2500)
-                plot = gr.Plot(value=fig0)                    # ⬆️ 차트가 맨 위에 위치
-                chart_summary_state.value = summ0
+        lang_radio.change(
+            change_lang,
+            [lang_radio, start_year, end_year],
+            [lang_state, plot, chart_summary_state]
+        )
 
-    # 2️⃣ 차트 아래에 컨트롤(연도 입력 + 줌 버튼) 배치
-                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")
-
-
-
-                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])
-
-                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])
-
-                gr.File(value=str(EVENTS_PATH), label="Download cycle_events.json")
-
-            # ── Tab 2: GPT Chat ──
+        gr.File(value=str(EVENTS_PATH), label="Download cycle_events.json")
+
+        with gr.Tabs():
             with gr.TabItem("Deep Research Chat"):
                 chatbot    = gr.Chatbot(label="Assistant")
                 user_input = gr.Textbox(lines=3, placeholder="메시지를 입력하세요…")
-
                 with gr.Row():
                     send_btn = gr.Button("Send", variant="primary")
-                    web_btn  = gr.Button("🔎 Web Search")  # NEW
-
-                # 기본 답변
-                def respond(history, msg, summ):
-                    reply = chat_with_gpt(history, msg, summ)
-                    history.append((msg, reply))
-                    return history, gr.Textbox(value="")
-
-                # 웹 검색 후 답변
-                def respond_with_search(history, msg, summ):
-                    search_md = format_search_results(msg)
-                    combined  = f"{msg}\n\n{search_md}"
-                    reply = chat_with_gpt(history, combined, summ)
-                    history.append((f"{msg}\n\n(웹검색 실행)", reply))
-                    return history, gr.Textbox(value="")
+                    web_btn  = gr.Button("🔎 Web Search")
+
+                def respond(hist, msg, summ):
+                    reply = chat_with_gpt(hist, msg, summ)
+                    hist.append((msg, reply))
+                    return hist, gr.Textbox(value="")
+
+                def respond_ws(hist, msg, summ):
+                    md = format_search_results(msg)
+                    reply = chat_with_gpt(hist, f"{msg}\n\n{md}", summ)
+                    hist.append((f"{msg}\n\n(웹검색)", reply))
+                    return hist, gr.Textbox(value="")
 
                 send_btn.click(respond,
                                [chatbot, user_input, chart_summary_state],
@@ -296,20 +315,20 @@ def create_app():
                 user_input.submit(respond,
                                   [chatbot, user_input, chart_summary_state],
                                   [chatbot, user_input])
-                web_btn.click(respond_with_search,
-                               [chatbot, user_input, chart_summary_state],
-                               [chatbot, user_input])
+                web_btn.click(respond_ws,
+                              [chatbot, user_input, chart_summary_state],
+                              [chatbot, user_input])
 
+        # ── fixed Discord badge ──
         gr.HTML(
             '<a href="https://discord.gg/openfreeai" target="_blank" id="discord-badge">'
             '<img src="https://img.shields.io/static/v1?label=Discord&message=Openfree%20AI&'
             'color=%230000ff&labelColor=%23800080&logo=discord&logoColor=white&style=for-the-badge"'
             ' alt="badge"></a>'
-        )                
+        )
+
     return demo
 
-# ──────────────────────────
-# 7. main
-# ──────────────────────────
+# ────────────────────────── main ──────────────────────────
 if __name__ == "__main__":
     create_app().launch()