Update app.py

app.py

@@ -1,17 +1,21 @@
 # -------- app_final.py --------
-import os, json, math, pathlib
+import os, json, math, pathlib, re, time, logging, requests
 import numpy as np
 import plotly.graph_objects as go
 import gradio as gr
 import openai
 
 # ──────────────────────────
-# 0. OpenAI API key
+# 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_ENDPOINT = "https://api.search.brave.com/res/v1/web/search"
+logging.basicConfig(level=logging.INFO)
+
 # ──────────────────────────
 # 1. Cycle config
 # ──────────────────────────
@@ -35,7 +39,54 @@ with open(EVENTS_PATH, encoding="utf-8") as f:
     EVENTS = {int(item["year"]): item["events"] for item in json.load(f)}
 
 # ──────────────────────────
-# 3. Helper functions
+# 3. Brave Search helpers
+# ──────────────────────────
+def brave_search(query: str, count: int = 8):
+    """Call Brave Search API; return list of dicts."""
+    if not BRAVE_KEY:
+        logging.warning("⚠️  BRAVE_KEY is empty; web search disabled.")
+        return []
+    try:
+        hdrs = {
+            "Accept": "application/json",
+            "X-Subscription-Token": BRAVE_KEY
+        }
+        data = requests.get(
+            BRAVE_ENDPOINT,
+            headers=hdrs,
+            params={"q": query, "count": str(count)},
+            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]
+        ]
+    except Exception as e:
+        logging.error(f"Brave Search 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:
+        return f"# [Web-Search] No live results for “{query}”.\n"
+    hdr = f"# [Web-Search] Top results for “{query}”\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)
+    )
+    return hdr + body + "\n"
+
+# ──────────────────────────
+# 4. Helper functions (chart)
 # ──────────────────────────
 def half_sine(xs, period, amp):
     """Half-sine ‘tower’ with zero bottom floor."""
@@ -112,7 +163,7 @@ def build_chart(start: int, end: int):
         height=450,
         margin=dict(t=30, l=40, r=40, b=40),
         hoverlabel=dict(bgcolor="#222", font_size=11),
-        hovermode="x"                               # ← unified x-hover
+        hovermode="x"                               # unified x-hover
     )
     fig.update_xaxes(title="Year",  range=[start, end], showgrid=False)
     fig.update_yaxes(title="Relative amplitude",
@@ -122,19 +173,15 @@ def build_chart(start: int, end: int):
               f"{sum(1 for y in EVENTS if start <= y <= end)}"
     return fig, summary
 
-
-# ──────────────────────────
-# 4. GPT chat helper
 # ──────────────────────────
-# ──────────────────────────
-# 4. GPT chat helper
+# 5. GPT chat helper
 # ──────────────────────────
 BASE_PROMPT = (
     "당신은 **CycleNavigator AI**로, 경제사·국제정치·장주기(9y Business, 50y K-Wave, "
     "80y Finance, 250y Hegemony) 분석에 정통한 전문가입니다. "
     "모든 답변은 한국어로 하되 학술적 정확성과 실무적 명료성을 동시에 갖추십시오. "
     "✦ 답변 구조 지침: ① 질문 핵심 요약 → ② 4대 주기와의 관련성 명시 → "
-    "③ 역사·데이터 근거 설명 → ④ 시사점·전망(필요 시) 순으로 서술하며, "
+    "③ 역사·데이터 근거 설명 → ④ 시사점·전망 순으로 서술하며, "
     "번호‧글머리표·짧은 문단을 활용해 논리적으로 배열합니다. "
     "✦ 제공된 [Chart summary]는 반드시 해석·인용하고, "
     "객관적 사실·연도·사건을 근거로 합니다. "
@@ -153,21 +200,26 @@ def chat_with_gpt(history, user_msg, chart_summary):
     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
+        model="gpt-3.5-turbo",
+        messages=messages,
+        max_tokens=600,
+        temperature=0.7
     )
     return res.choices[0].message.content.strip()
 
 # ──────────────────────────
-# 5. Gradio UI
+# 6. Gradio UI
 # ──────────────────────────
 def create_app():
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
-        # ── Title & subtitle
+        # Title & subtitles
         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>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) • "
@@ -176,12 +228,11 @@ def create_app():
             "<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 ─────────────────────────────────────
+            # ── Tab 1: Timeline Chart ──
             with gr.TabItem("Timeline Chart"):
                 with gr.Row():
                     start_year = gr.Number(label="Start Year", value=1500, precision=0)
@@ -189,12 +240,10 @@ def create_app():
                     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
@@ -203,43 +252,58 @@ def create_app():
                 end_year.change(refresh,   [start_year, end_year],
                                              [plot, chart_summary_state])
 
-                # zoom helpers
                 def zoom(s, e, factor):
-                    mid  = (s + e) / 2
+                    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])
+                              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])
+                               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 ───────────────────────────────────────────
+            # ── 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")
 
+                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="")
+
                 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])
+                web_btn.click(respond_with_search,
+                               [chatbot, user_input, chart_summary_state],
+                               [chatbot, user_input])
     return demo
 
-
+# ──────────────────────────
+# 7. main
+# ──────────────────────────
 if __name__ == "__main__":
     create_app().launch()